212 lines
9.2 KiB
Go
212 lines
9.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"git.sundynix.cn/Blizzard/Veyra/internal/apperror"
|
|
"git.sundynix.cn/Blizzard/Veyra/internal/model"
|
|
"github.com/google/uuid"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
type CandidateAnalysisRepository struct{ pool *pgxpool.Pool }
|
|
|
|
func NewCandidateAnalysisRepository(pool *pgxpool.Pool) *CandidateAnalysisRepository {
|
|
return &CandidateAnalysisRepository{pool: pool}
|
|
}
|
|
|
|
const policyID = "a945c2bf-9c77-4d4a-9f4f-a986b729776c"
|
|
|
|
func scanPolicy(row pgx.Row) (model.AIReviewPolicy, error) {
|
|
var p model.AIReviewPolicy
|
|
err := row.Scan(&p.ID, &p.Enabled, &p.AutoPublish, &p.Version, &p.UpdatedBy)
|
|
return p, mapError(err)
|
|
}
|
|
func (r *CandidateAnalysisRepository) Policy(ctx context.Context) (model.AIReviewPolicy, error) {
|
|
return scanPolicy(r.pool.QueryRow(ctx, `SELECT id,enabled,auto_publish,version,updated_by FROM veyra_ai_review_policy WHERE id=$1`, policyID))
|
|
}
|
|
func (r *CandidateAnalysisRepository) SavePolicy(ctx context.Context, p model.AIReviewPolicy, actor, requestID string) (model.AIReviewPolicy, error) {
|
|
tx, err := r.pool.Begin(ctx)
|
|
if err != nil {
|
|
return p, mapError(err)
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
row := tx.QueryRow(ctx, `UPDATE veyra_ai_review_policy SET enabled=$2,auto_publish=$3,version=version+1,updated_by=$4,updated_at=now() WHERE id=$1 AND version=$5 RETURNING id,enabled,auto_publish,version,updated_by`, policyID, p.Enabled, p.AutoPublish, actor, p.Version)
|
|
p, err = scanPolicy(row)
|
|
if err == apperror.ErrNotFound {
|
|
return p, apperror.ErrConflict
|
|
}
|
|
if err != nil {
|
|
return p, err
|
|
}
|
|
if err = appendAudit(ctx, tx, actor, "AI_REVIEW_POLICY_SAVED", "ai_review_policy", policyID, requestID, map[string]any{"enabled": p.Enabled, "auto_publish": p.AutoPublish, "version": p.Version}); err != nil {
|
|
return p, err
|
|
}
|
|
return p, mapError(tx.Commit(ctx))
|
|
}
|
|
func (r *CandidateAnalysisRepository) Claim(ctx context.Context, p model.AIReviewPolicy) (model.CandidateRevision, string, error) {
|
|
tx, err := r.pool.Begin(ctx)
|
|
if err != nil {
|
|
return model.CandidateRevision{}, "", mapError(err)
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
if _, err = tx.Exec(ctx, `SELECT pg_advisory_xact_lock(hashtext('veyra:ai-claim')::bigint)`); err != nil {
|
|
return model.CandidateRevision{}, "", mapError(err)
|
|
}
|
|
c, err := scanCandidate(tx.QueryRow(ctx, `SELECT `+candidateColumns+candidateJoin+` WHERE r.status='PENDING_REVIEW' AND r.revision=c.latest_revision AND r.valid_until>now() AND `+screeningEligibility+` AND NOT EXISTS(SELECT 1 FROM veyra_candidate_analysis a WHERE a.candidate_revision_id=r.id AND a.input_version=r.version AND a.policy_version=$1 AND (a.status NOT IN ('FAILED','RUNNING') OR a.attempts>=3 OR a.lease_until>now())) ORDER BY r.created_at LIMIT 1`, p.Version))
|
|
if err != nil {
|
|
return c, "", err
|
|
}
|
|
if c.Screening != nil {
|
|
if err = tx.QueryRow(ctx, `SELECT input FROM veyra_strategy_screening WHERE id=$1`, c.Screening.ID).Scan(&c.Screening.Input); err != nil {
|
|
return c, "", mapError(err)
|
|
}
|
|
}
|
|
token := uuid.NewString()
|
|
_, err = tx.Exec(ctx, `INSERT INTO veyra_candidate_analysis(id,candidate_revision_id,input_version,policy_version,status,claim_token,lease_until) VALUES($1,$2,$3,$4,'RUNNING',$5,now()+interval '5 minutes') ON CONFLICT(candidate_revision_id,input_version,policy_version) DO UPDATE SET status='RUNNING',attempts=veyra_candidate_analysis.attempts+1,claim_token=excluded.claim_token,lease_until=excluded.lease_until,updated_at=now(),message=''`, uuid.NewString(), c.ID, c.Version, p.Version, token)
|
|
if err != nil {
|
|
return c, "", mapError(err)
|
|
}
|
|
return c, token, mapError(tx.Commit(ctx))
|
|
}
|
|
func (r *CandidateAnalysisRepository) Complete(ctx context.Context, p model.AIReviewPolicy, c model.CandidateRevision, token string, primary, review model.ProviderConfig, a, b *model.AIConclusion, message string) error {
|
|
tx, err := r.pool.Begin(ctx)
|
|
if err != nil {
|
|
return mapError(err)
|
|
}
|
|
defer tx.Rollback(context.Background())
|
|
current, err := scanPolicy(tx.QueryRow(ctx, `SELECT id,enabled,auto_publish,version,updated_by FROM veyra_ai_review_policy WHERE id=$1 FOR SHARE`, policyID))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
status := "FAILED"
|
|
if message == "" {
|
|
status = "REJECTED"
|
|
if a != nil && b != nil && a.Decision == "APPROVE" && b.Decision == "APPROVE" {
|
|
status = "APPROVED"
|
|
}
|
|
}
|
|
if !current.Enabled || current.Version != p.Version {
|
|
status = "STALE"
|
|
message = "审核策略已变更"
|
|
}
|
|
// Lock provider rows in stable key order; a changed route cannot publish old output.
|
|
rows, err := tx.Query(ctx, `SELECT config_key,version,enabled FROM veyra_provider_config WHERE config_key IN ('market_primary','model_primary','model_review') ORDER BY config_key FOR SHARE`)
|
|
if err != nil {
|
|
return mapError(err)
|
|
}
|
|
valid := 0
|
|
sourceValid := c.Screening == nil
|
|
for rows.Next() {
|
|
var key string
|
|
var version int64
|
|
var enabled bool
|
|
if err = rows.Scan(&key, &version, &enabled); err != nil {
|
|
rows.Close()
|
|
return mapError(err)
|
|
}
|
|
if key == "market_primary" && c.Screening != nil && enabled && version == c.Screening.SourceVersion {
|
|
sourceValid = true
|
|
}
|
|
if enabled && ((key == primary.ConfigKey && version == primary.Version) || (key == review.ConfigKey && version == review.Version)) {
|
|
valid++
|
|
}
|
|
}
|
|
err = rows.Err()
|
|
rows.Close()
|
|
if err != nil {
|
|
return mapError(err)
|
|
}
|
|
if valid != 2 {
|
|
status = "STALE"
|
|
message = "模型配置已变更或未启用"
|
|
}
|
|
if !sourceValid {
|
|
status = "STALE"
|
|
message = "行情来源已停用或版本已变化"
|
|
}
|
|
var storedToken, storedStatus string
|
|
if err = tx.QueryRow(ctx, `SELECT claim_token,status FROM veyra_candidate_analysis WHERE candidate_revision_id=$1 AND input_version=$2 AND policy_version=$3 FOR UPDATE`, c.ID, c.Version, p.Version).Scan(&storedToken, &storedStatus); err != nil {
|
|
return mapError(err)
|
|
}
|
|
if token != storedToken || storedStatus != "RUNNING" {
|
|
return apperror.ErrConflict
|
|
}
|
|
// Existing transition owns candidate/strategy locks and validates source/version/time.
|
|
if status == "APPROVED" && current.AutoPublish && current.UpdatedBy != nil {
|
|
var actorRole, actorStatus string
|
|
err = tx.QueryRow(ctx, `SELECT role,status FROM veyra_user WHERE id=$1 FOR SHARE`, *current.UpdatedBy).Scan(&actorRole, &actorStatus)
|
|
if err != nil {
|
|
return mapError(err)
|
|
}
|
|
if actorRole != "ADMIN" || actorStatus != "ACTIVE" {
|
|
status = "STALE"
|
|
message = "自动发布授权身份已停用"
|
|
} else {
|
|
// Savepoint allows a rejected rule check to remain visible without publishing.
|
|
if _, err = tx.Exec(ctx, "SAVEPOINT ai_publish"); err != nil {
|
|
return mapError(err)
|
|
}
|
|
_, err = transitionCandidate(ctx, tx, c.ID, "PUBLISH", model.CandidateActionInput{ExpectedVersion: c.Version, Reason: "AI 自动审核通过；主分析与复核已保存，价格由策略规则生成"}, *current.UpdatedBy, uuid.NewString())
|
|
if err != nil {
|
|
if _, rollbackErr := tx.Exec(ctx, "ROLLBACK TO SAVEPOINT ai_publish"); rollbackErr != nil {
|
|
return mapError(rollbackErr)
|
|
}
|
|
status = "STALE"
|
|
message = "发布前规则校验未通过，请检查候选时效与策略版本"
|
|
} else {
|
|
status = "PUBLISHED"
|
|
message = "AI 分析与复核通过，已自动发布"
|
|
}
|
|
}
|
|
}
|
|
if message == "" {
|
|
if status == "REJECTED" {
|
|
message = "AI 未通过，保留人工复核"
|
|
} else {
|
|
message = "AI 审核通过，当前为人工发布模式"
|
|
}
|
|
}
|
|
rawA, _ := json.Marshal(a)
|
|
rawB, _ := json.Marshal(b)
|
|
_, err = tx.Exec(ctx, `UPDATE veyra_candidate_analysis SET status=$4,primary_model=$5,review_model=$6,primary_version=$7,review_version=$8,analysis=$9,review=$10,message=$11,lease_until=now()+interval '5 minutes',updated_at=now() WHERE candidate_revision_id=$1 AND input_version=$2 AND policy_version=$3`, c.ID, c.Version, p.Version, status, primary.ModelName, review.ModelName, primary.Version, review.Version, rawA, rawB, message)
|
|
if err != nil {
|
|
return mapError(err)
|
|
}
|
|
if current.UpdatedBy != nil {
|
|
if err = appendAudit(ctx, tx, *current.UpdatedBy, "AI_CANDIDATE_COMPLETED", "candidate_revision", c.ID, uuid.NewString(), map[string]any{"status": status, "policy_version": p.Version, "input_version": c.Version}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return mapError(tx.Commit(ctx))
|
|
}
|
|
func (r *CandidateAnalysisRepository) List(ctx context.Context) ([]model.CandidateAnalysis, error) {
|
|
rows, err := r.pool.Query(ctx, `SELECT id,candidate_revision_id,input_version,policy_version,status,attempts,primary_model,review_model,primary_version,review_version,analysis,review,message,created_at,updated_at FROM veyra_candidate_analysis ORDER BY updated_at DESC LIMIT 100`)
|
|
if err != nil {
|
|
return nil, mapError(err)
|
|
}
|
|
defer rows.Close()
|
|
out := make([]model.CandidateAnalysis, 0)
|
|
for rows.Next() {
|
|
var v model.CandidateAnalysis
|
|
var a, b []byte
|
|
if err = rows.Scan(&v.ID, &v.CandidateRevisionID, &v.InputVersion, &v.PolicyVersion, &v.Status, &v.Attempts, &v.PrimaryModel, &v.ReviewModel, &v.PrimaryVersion, &v.ReviewVersion, &a, &b, &v.Message, &v.CreatedAt, &v.UpdatedAt); err != nil {
|
|
return nil, mapError(err)
|
|
}
|
|
if len(a) > 0 {
|
|
if err = json.Unmarshal(a, &v.Analysis); err != nil {
|
|
return nil, apperror.ErrUnavailable
|
|
}
|
|
}
|
|
if len(b) > 0 {
|
|
if err = json.Unmarshal(b, &v.Review); err != nil {
|
|
return nil, apperror.ErrUnavailable
|
|
}
|
|
}
|
|
out = append(out, v)
|
|
}
|
|
return out, mapError(rows.Err())
|
|
}
|