106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
|
// Code scaffolded by goctl. Safe to edit.
|
||
|
|
// goctl 1.10.1
|
||
|
|
|
||
|
|
package brand
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
kgusecase "haixun-backend/internal/model/knowledge_graph/domain/usecase"
|
||
|
|
"haixun-backend/internal/svc"
|
||
|
|
"haixun-backend/internal/types"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type GetKnowledgeGraphLogic struct {
|
||
|
|
logx.Logger
|
||
|
|
ctx context.Context
|
||
|
|
svcCtx *svc.ServiceContext
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGetKnowledgeGraphLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetKnowledgeGraphLogic {
|
||
|
|
return &GetKnowledgeGraphLogic{
|
||
|
|
Logger: logx.WithContext(ctx),
|
||
|
|
ctx: ctx,
|
||
|
|
svcCtx: svcCtx,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *GetKnowledgeGraphLogic) GetKnowledgeGraph(req *types.BrandPath) (resp *types.KnowledgeGraphData, err error) {
|
||
|
|
tenantID, uid, err := actorFrom(l.ctx)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if _, err := l.svcCtx.Brand.Get(l.ctx, tenantID, uid, req.ID); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
graph, err := l.svcCtx.KnowledgeGraph.Get(l.ctx, tenantID, uid, req.ID)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
data := toKnowledgeGraphData(graph)
|
||
|
|
return &data, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func toKnowledgeGraphData(graph *kgusecase.GraphSummary) types.KnowledgeGraphData {
|
||
|
|
if graph == nil {
|
||
|
|
return types.KnowledgeGraphData{}
|
||
|
|
}
|
||
|
|
nodes := make([]types.KnowledgeGraphNodeData, 0, len(graph.Nodes))
|
||
|
|
for _, node := range graph.Nodes {
|
||
|
|
evidence := make([]types.KnowledgeGraphEvidenceData, 0, len(node.Evidence))
|
||
|
|
for _, item := range node.Evidence {
|
||
|
|
evidence = append(evidence, types.KnowledgeGraphEvidenceData{
|
||
|
|
URL: item.URL,
|
||
|
|
Snippet: item.Snippet,
|
||
|
|
Query: item.Query,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
nodes = append(nodes, types.KnowledgeGraphNodeData{
|
||
|
|
ID: node.ID,
|
||
|
|
Label: node.Label,
|
||
|
|
NodeKind: node.NodeKind,
|
||
|
|
Type: node.Type,
|
||
|
|
Layer: node.Layer,
|
||
|
|
Relation: node.Relation,
|
||
|
|
PlacementValue: node.PlacementValue,
|
||
|
|
ProductFitScore: node.ProductFitScore,
|
||
|
|
SelectedForScan: node.SelectedForScan,
|
||
|
|
RelevanceTags: append([]string{}, node.DerivedTags.Relevance...),
|
||
|
|
RecencyTags: append([]string{}, node.DerivedTags.Recency...),
|
||
|
|
Evidence: evidence,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
edges := make([]types.KnowledgeGraphEdgeData, 0, len(graph.Edges))
|
||
|
|
for _, edge := range graph.Edges {
|
||
|
|
edges = append(edges, types.KnowledgeGraphEdgeData{
|
||
|
|
From: edge.From,
|
||
|
|
To: edge.To,
|
||
|
|
Relation: edge.Relation,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
sources := make([]types.BraveSourceData, 0, len(graph.BraveSources))
|
||
|
|
for _, src := range graph.BraveSources {
|
||
|
|
sources = append(sources, types.BraveSourceData{
|
||
|
|
Query: src.Query,
|
||
|
|
Title: src.Title,
|
||
|
|
URL: src.URL,
|
||
|
|
Snippet: src.Snippet,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
return types.KnowledgeGraphData{
|
||
|
|
ID: graph.ID,
|
||
|
|
BrandID: graph.BrandID,
|
||
|
|
Seed: graph.Seed,
|
||
|
|
Nodes: nodes,
|
||
|
|
Edges: edges,
|
||
|
|
BraveSources: sources,
|
||
|
|
ExpandStrategy: graph.ExpandStrategy,
|
||
|
|
PainTagCount: graph.PainTagCount,
|
||
|
|
GeneratedAt: graph.GeneratedAt,
|
||
|
|
CreateAt: graph.CreateAt,
|
||
|
|
UpdateAt: graph.UpdateAt,
|
||
|
|
}
|
||
|
|
}
|