Files
ai-customer-service/internal/store/postgres/audit_store.go
Your Name cf46b27610 fix: P0-1 RateLimiter并发写安全 + P0-2工单操作错误码区分 + P1 rows.Close修复
P0-1 (limits.go): Allow()方法改为全程使用写锁保护counters map读写,避免RLock写入时的data race
P0-2 (ticket_workflow.go+ticket_handler.go): Assign/Resolve/Close操作先查询ticket存在性和状态,返回明确的CS_TICKET_4001/CS_TKT_4002/CS_TICKET_4092/CS_TICKET_4093错误码,handler根据错误前缀路由HTTP状态码
P1-1 (ticket_store.go): 移除GetStats中3处手动rows.Close(),只保留defer Close()
2026-05-01 20:56:25 +08:00

87 lines
2.0 KiB
Go

package postgres
import (
"context"
"database/sql"
"encoding/json"
"fmt"
"strings"
"time"
"github.com/bridge/ai-customer-service/internal/domain/audit"
)
type AuditStore struct {
db *sql.DB
}
func NewAuditStore(db *sql.DB) *AuditStore {
return &AuditStore{db: db}
}
func (s *AuditStore) Add(ctx context.Context, event audit.Event) error {
if s.db == nil {
return fmt.Errorf("db is nil")
}
if event.CreatedAt.IsZero() {
event.CreatedAt = time.Now()
}
beforeState, err := marshalJSON(event.BeforeState)
if err != nil {
return err
}
afterState, err := marshalJSON(resolveAfterState(event))
if err != nil {
return err
}
objectType, objectID := resolveAuditObject(event)
action := strings.TrimSpace(event.Action)
if action == "" {
action = "update"
}
actorID := strings.TrimSpace(event.ActorID)
if actorID == "" {
actorID = coalesceActor(event.OpenID)
}
_, err = s.db.ExecContext(ctx, `INSERT INTO cs_audit_logs(id, tenant_id, object_type, object_id, action, before_state, after_state, actor_id, source_ip, created_at) VALUES ($1::uuid, $2, $3, $4, $5, $6::jsonb, $7::jsonb, $8, NULLIF($9,''), $10)`, event.ID, "default", objectType, objectID, action, beforeState, afterState, actorID, event.SourceIP, event.CreatedAt)
return err
}
func marshalJSON(value map[string]any) (string, error) {
if len(value) == 0 {
return "{}", nil
}
payload, err := json.Marshal(value)
if err != nil {
return "", err
}
return string(payload), nil
}
func resolveAfterState(event audit.Event) map[string]any {
if len(event.AfterState) > 0 {
return event.AfterState
}
if len(event.Payload) > 0 {
return event.Payload
}
return map[string]any{}
}
func resolveAuditObject(event audit.Event) (string, string) {
if strings.TrimSpace(event.TicketID) != "" {
return "ticket", event.TicketID
}
if strings.TrimSpace(event.SessionID) != "" {
return event.Type, event.SessionID
}
return event.Type, "system"
}
func coalesceActor(actor string) string {
if actor == "" {
return "system"
}
return actor
}