cf46b27610
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()
31 lines
688 B
Go
31 lines
688 B
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
type DedupStore struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewDedupStore(db *sql.DB) *DedupStore {
|
|
return &DedupStore{db: db}
|
|
}
|
|
|
|
func (s *DedupStore) TryRecord(ctx context.Context, channel, messageID, sessionID string) (bool, error) {
|
|
if s.db == nil {
|
|
return false, fmt.Errorf("db is nil")
|
|
}
|
|
result, err := s.db.ExecContext(ctx, `INSERT INTO cs_message_dedup(channel, message_id, session_id) VALUES ($1,$2,NULLIF($3,'')::uuid) ON CONFLICT DO NOTHING`, channel, messageID, sessionID)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
affected, err := result.RowsAffected()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return affected == 1, nil
|
|
}
|