package repository import ( "context" "database/sql" "errors" "fmt" "github.com/pocketbase/dbx" "github.com/pocketbase/pocketbase/core" "github.com/usual2970/certimate/internal/app" "github.com/usual2970/certimate/internal/domain" ) type WorkflowLogRepository struct{} func NewWorkflowLogRepository() *WorkflowLogRepository { return &WorkflowLogRepository{} } func (r *WorkflowLogRepository) ListByWorkflowRunId(ctx context.Context, workflowRunId string) ([]*domain.WorkflowLog, error) { records, err := app.GetApp().FindRecordsByFilter( domain.CollectionNameWorkflowLog, "runId={:runId}", "timestamp", 0, 0, dbx.Params{"runId": workflowRunId}, ) if err != nil { return nil, err } workflowLogs := make([]*domain.WorkflowLog, 0) for _, record := range records { workflowLog, err := r.castRecordToModel(record) if err != nil { return nil, err } workflowLogs = append(workflowLogs, workflowLog) } return workflowLogs, nil } func (r *WorkflowLogRepository) Save(ctx context.Context, workflowLog *domain.WorkflowLog) (*domain.WorkflowLog, error) { collection, err := app.GetApp().FindCollectionByNameOrId(domain.CollectionNameWorkflowLog) if err != nil { return workflowLog, err } var record *core.Record if workflowLog.Id == "" { record = core.NewRecord(collection) } else { record, err = app.GetApp().FindRecordById(collection, workflowLog.Id) if err != nil { if errors.Is(err, sql.ErrNoRows) { return workflowLog, err } record = core.NewRecord(collection) } } record.Set("workflowId", workflowLog.WorkflowId) record.Set("runId", workflowLog.RunId) record.Set("nodeId", workflowLog.NodeId) record.Set("nodeName", workflowLog.NodeName) record.Set("timestamp", workflowLog.Timestamp) record.Set("level", workflowLog.Level) record.Set("message", workflowLog.Message) record.Set("data", workflowLog.Data) record.Set("created", workflowLog.CreatedAt) err = app.GetApp().Save(record) if err != nil { return workflowLog, err } workflowLog.Id = record.Id workflowLog.CreatedAt = record.GetDateTime("created").Time() workflowLog.UpdatedAt = record.GetDateTime("updated").Time() return workflowLog, nil } func (r *WorkflowLogRepository) castRecordToModel(record *core.Record) (*domain.WorkflowLog, error) { if record == nil { return nil, fmt.Errorf("record is nil") } logdata := make(map[string]any) if err := record.UnmarshalJSONField("data", &logdata); err != nil { return nil, err } workflowLog := &domain.WorkflowLog{ Meta: domain.Meta{ Id: record.Id, CreatedAt: record.GetDateTime("created").Time(), UpdatedAt: record.GetDateTime("updated").Time(), }, WorkflowId: record.GetString("workflowId"), RunId: record.GetString("runId"), NodeId: record.GetString("nodeId"), NodeName: record.GetString("nodeName"), Timestamp: int64(record.GetInt("timestamp")), Level: record.GetString("level"), Message: record.GetString("message"), Data: logdata, } return workflowLog, nil }