59 lines
2.2 KiB
Go
59 lines
2.2 KiB
Go
package sop
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"git.iwork-ai.com/xdc/iqudo-top1/internal/model"
|
|
"gorm.io/datatypes"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type knowledgeNodeConfig struct {
|
|
KnowledgeCardID uint64 `json:"knowledge_card_id"`
|
|
KnowledgeCardVersionID uint64 `json:"knowledge_card_version_id"`
|
|
}
|
|
|
|
// BindKnowledgeVersions freezes the current published knowledge-card version
|
|
// into every knowledge node before the SOP version becomes immutable.
|
|
func BindKnowledgeVersions(tx *gorm.DB, tenantID, sopVersionID uint64) error {
|
|
nodes := make([]model.SOPNode, 0)
|
|
if err := tx.Where("tenant_id = ? AND sop_version_id = ? AND type = ?", tenantID, sopVersionID, "knowledge").Find(&nodes).Error; err != nil {
|
|
return err
|
|
}
|
|
for _, node := range nodes {
|
|
var config knowledgeNodeConfig
|
|
if err := json.Unmarshal(node.Config, &config); err != nil || config.KnowledgeCardID == 0 {
|
|
return fmt.Errorf("knowledge node %s has invalid configuration", node.NodeKey)
|
|
}
|
|
var version model.KnowledgeCardVersion
|
|
err := tx.Table("knowledge_card_versions kv").Select("kv.*").
|
|
Joins("JOIN knowledge_cards kc ON kc.id = kv.knowledge_card_id").
|
|
Where("kv.tenant_id = ? AND kv.knowledge_card_id = ? AND kv.status = ? AND kc.tenant_id = ? AND kc.status = ?", tenantID, config.KnowledgeCardID, "published", tenantID, "published").
|
|
Order("kv.version DESC").First(&version).Error
|
|
if err != nil {
|
|
return fmt.Errorf("knowledge node %s has no published card version: %w", node.NodeKey, err)
|
|
}
|
|
updated, err := withKnowledgeVersion(node.Config, version.ID)
|
|
if err != nil {
|
|
return fmt.Errorf("update knowledge node %s: %w", node.NodeKey, err)
|
|
}
|
|
if err := tx.Model(&model.SOPNode{}).Where("id = ? AND tenant_id = ? AND sop_version_id = ?", node.ID, tenantID, sopVersionID).Update("config", updated).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func withKnowledgeVersion(config datatypes.JSON, versionID uint64) (datatypes.JSON, error) {
|
|
value := map[string]interface{}{}
|
|
if len(config) > 0 {
|
|
if err := json.Unmarshal(config, &value); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
value["knowledge_card_version_id"] = versionID
|
|
encoded, err := json.Marshal(value)
|
|
return datatypes.JSON(encoded), err
|
|
}
|