zheromiku
dcdc8d1ddd
全栈知识治理平台首次入库（P1-P10 + 发布链路模拟验证闭环）： - 治理核心：源目录/变更集/人工审核/发布批次/发布执行（fail-closed 闸口） - 测试通道：WeKnora 测试 KB 上传/轮询/验收；poll 完成时写入 parser_generation/index_generation，修复 manual/seed 路径发布闸口永久拦截 - 发布适配器 v071（本地 source=测试角色 / 云端 target=生产角色 模拟验证通过）： * ES 导出/导入 mapping 自适应（.keyword 子字段、可选 tenant_id） * OSS locator 支持嵌套 scheme；chunks.seq_id 由目标序列分配 * bundle 归档同身份重试原子替换；verifier 按 bundle chunk_type 计数 * 回滚 job payload 支持 prior_execution_id（strict 解码） - 三通道发布+回滚演练实测：publish 4/4 verified，检索命中， rollback 4/4 rolled_back，目标端 PG is_enabled=false、ES is_enabled=false - 前端 A·Data-console 风格全站统一，120 用例通过 - 门禁：go vet / go test 全包通过，gofmt 干净
66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"knowledge-governance/internal/governance"
|
|
)
|
|
|
|
type PublicationForRemoval struct {
|
|
ID string
|
|
TestKBID string
|
|
KnowledgeID string
|
|
RetrievalState governance.RetrievalState
|
|
}
|
|
|
|
func (s *Store) LoadPublicationForRemoval(ctx context.Context, publicationID string) (PublicationForRemoval, error) {
|
|
var candidate PublicationForRemoval
|
|
var retrievalState string
|
|
err := s.db.QueryRow(ctx, `SELECT id,test_kb_id,knowledge_id,retrieval_state
|
|
FROM test_publications
|
|
WHERE id=$1 AND knowledge_id<>'' AND retrieval_state='disabling'`, publicationID).Scan(
|
|
&candidate.ID, &candidate.TestKBID, &candidate.KnowledgeID, &retrievalState,
|
|
)
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return PublicationForRemoval{}, ErrPublicationStateConflict
|
|
}
|
|
if err != nil {
|
|
return PublicationForRemoval{}, fmt.Errorf("load publication for removal: %w", err)
|
|
}
|
|
candidate.RetrievalState = governance.RetrievalState(retrievalState)
|
|
return candidate, nil
|
|
}
|
|
|
|
func (s *Store) RecordRemovalCompleted(ctx context.Context, publicationID string, now time.Time) error {
|
|
tag, err := s.db.Exec(ctx, `UPDATE test_publications
|
|
SET retrieval_state='disabled',last_error='',updated_at=$2
|
|
WHERE id=$1 AND retrieval_state='disabling'`, publicationID, now)
|
|
if err != nil {
|
|
return fmt.Errorf("record publication removal: %w", err)
|
|
}
|
|
if tag.RowsAffected() != 1 {
|
|
return ErrPublicationStateConflict
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Store) RecordRemovalFailure(ctx context.Context, publicationID, code string, now time.Time) error {
|
|
if code != "removal_identity_invalid" && code != "removal_remote_rejected" {
|
|
return ErrInvalidErrorCode
|
|
}
|
|
tag, err := s.db.Exec(ctx, `UPDATE test_publications
|
|
SET retrieval_state='failed',last_error=$2,updated_at=$3
|
|
WHERE id=$1 AND retrieval_state='disabling'`, publicationID, code, now)
|
|
if err != nil {
|
|
return fmt.Errorf("record publication removal failure: %w", err)
|
|
}
|
|
if tag.RowsAffected() != 1 {
|
|
return ErrPublicationStateConflict
|
|
}
|
|
return nil
|
|
}
|