mirror of https://github.com/usual2970/certimate
346 lines
8.2 KiB
Go
346 lines
8.2 KiB
Go
package migrations
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pocketbase/pocketbase/core"
|
|
m "github.com/pocketbase/pocketbase/migrations"
|
|
)
|
|
|
|
func init() {
|
|
m.Register(func(app core.App) error {
|
|
// create collection `workflow_logs`
|
|
{
|
|
jsonData := `{
|
|
"createRule": null,
|
|
"deleteRule": null,
|
|
"fields": [
|
|
{
|
|
"autogeneratePattern": "[a-z0-9]{15}",
|
|
"hidden": false,
|
|
"id": "text3208210256",
|
|
"max": 15,
|
|
"min": 15,
|
|
"name": "id",
|
|
"pattern": "^[a-z0-9]+$",
|
|
"presentable": false,
|
|
"primaryKey": true,
|
|
"required": true,
|
|
"system": true,
|
|
"type": "text"
|
|
},
|
|
{
|
|
"cascadeDelete": true,
|
|
"collectionId": "tovyif5ax6j62ur",
|
|
"hidden": false,
|
|
"id": "relation3371272342",
|
|
"maxSelect": 1,
|
|
"minSelect": 0,
|
|
"name": "workflowId",
|
|
"presentable": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "relation"
|
|
},
|
|
{
|
|
"cascadeDelete": true,
|
|
"collectionId": "qjp8lygssgwyqyz",
|
|
"hidden": false,
|
|
"id": "relation821863227",
|
|
"maxSelect": 1,
|
|
"minSelect": 0,
|
|
"name": "runId",
|
|
"presentable": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "relation"
|
|
},
|
|
{
|
|
"autogeneratePattern": "",
|
|
"hidden": false,
|
|
"id": "text157423495",
|
|
"max": 0,
|
|
"min": 0,
|
|
"name": "nodeId",
|
|
"pattern": "",
|
|
"presentable": false,
|
|
"primaryKey": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "text"
|
|
},
|
|
{
|
|
"autogeneratePattern": "",
|
|
"hidden": false,
|
|
"id": "text3227511481",
|
|
"max": 0,
|
|
"min": 0,
|
|
"name": "nodeName",
|
|
"pattern": "",
|
|
"presentable": false,
|
|
"primaryKey": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "text"
|
|
},
|
|
{
|
|
"hidden": false,
|
|
"id": "number2782324286",
|
|
"max": null,
|
|
"min": null,
|
|
"name": "timestamp",
|
|
"onlyInt": false,
|
|
"presentable": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "number"
|
|
},
|
|
{
|
|
"autogeneratePattern": "",
|
|
"hidden": false,
|
|
"id": "text2599078931",
|
|
"max": 0,
|
|
"min": 0,
|
|
"name": "level",
|
|
"pattern": "",
|
|
"presentable": false,
|
|
"primaryKey": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "text"
|
|
},
|
|
{
|
|
"autogeneratePattern": "",
|
|
"hidden": false,
|
|
"id": "text3065852031",
|
|
"max": 0,
|
|
"min": 0,
|
|
"name": "message",
|
|
"pattern": "",
|
|
"presentable": false,
|
|
"primaryKey": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "text"
|
|
},
|
|
{
|
|
"hidden": false,
|
|
"id": "json2918445923",
|
|
"maxSize": 0,
|
|
"name": "data",
|
|
"presentable": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "json"
|
|
},
|
|
{
|
|
"hidden": false,
|
|
"id": "autodate2990389176",
|
|
"name": "created",
|
|
"onCreate": true,
|
|
"onUpdate": false,
|
|
"presentable": false,
|
|
"system": false,
|
|
"type": "autodate"
|
|
}
|
|
],
|
|
"id": "pbc_1682296116",
|
|
"indexes": [
|
|
"CREATE INDEX ` + "`" + `idx_IOlpy6XuJ2` + "`" + ` ON ` + "`" + `workflow_logs` + "`" + ` (` + "`" + `workflowId` + "`" + `)",
|
|
"CREATE INDEX ` + "`" + `idx_qVlTb2yl7v` + "`" + ` ON ` + "`" + `workflow_logs` + "`" + ` (` + "`" + `runId` + "`" + `)"
|
|
],
|
|
"listRule": null,
|
|
"name": "workflow_logs",
|
|
"system": false,
|
|
"type": "base",
|
|
"updateRule": null,
|
|
"viewRule": null
|
|
}`
|
|
|
|
collection := &core.Collection{}
|
|
if err := json.Unmarshal([]byte(jsonData), &collection); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := app.Save(collection); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// migrate data
|
|
{
|
|
workflowRuns, err := app.FindAllRecords("workflow_run")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, workflowRun := range workflowRuns {
|
|
type dWorkflowRunLogRecord struct {
|
|
Time string `json:"time"`
|
|
Level string `json:"level"`
|
|
Content string `json:"content"`
|
|
Error string `json:"error"`
|
|
}
|
|
type dWorkflowRunLog struct {
|
|
NodeId string `json:"nodeId"`
|
|
NodeName string `json:"nodeName"`
|
|
Records []dWorkflowRunLogRecord `json:"records"`
|
|
Error string `json:"error"`
|
|
}
|
|
|
|
logs := make([]dWorkflowRunLog, 0)
|
|
if err := workflowRun.UnmarshalJSONField("logs", &logs); err != nil {
|
|
continue
|
|
}
|
|
|
|
collection, err := app.FindCollectionByNameOrId("workflow_logs")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, log := range logs {
|
|
for _, logRecord := range log.Records {
|
|
record := core.NewRecord(collection)
|
|
createdAt, _ := time.Parse(time.RFC3339, logRecord.Time)
|
|
record.Set("workflowId", workflowRun.Get("workflowId"))
|
|
record.Set("runId", workflowRun.Get("id"))
|
|
record.Set("nodeId", log.NodeId)
|
|
record.Set("nodeName", log.NodeName)
|
|
record.Set("timestamp", createdAt.UnixMilli())
|
|
record.Set("level", logRecord.Level)
|
|
record.Set("message", strings.TrimSpace(logRecord.Content+" "+logRecord.Error))
|
|
record.Set("created", createdAt)
|
|
if err := app.Save(record); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// update collection `workflow_run`
|
|
{
|
|
collection, err := app.FindCollectionByNameOrId("workflow_run")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := collection.Fields.AddMarshaledJSONAt(6, []byte(`{
|
|
"hidden": false,
|
|
"id": "json772177811",
|
|
"maxSize": 5000000,
|
|
"name": "detail",
|
|
"presentable": false,
|
|
"required": false,
|
|
"system": false,
|
|
"type": "json"
|
|
}`)); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := app.Save(collection); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// migrate data
|
|
{
|
|
workflowRuns, err := app.FindAllRecords("workflow_run")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
workflowOutputs, err := app.FindAllRecords("workflow_output")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
type dWorkflowNode struct {
|
|
Id string `json:"id"`
|
|
Type string `json:"type"`
|
|
Name string `json:"name"`
|
|
Config map[string]any `json:"config"`
|
|
Inputs []map[string]any `json:"inputs"`
|
|
Outputs []map[string]any `json:"outputs"`
|
|
Next *dWorkflowNode `json:"next,omitempty"`
|
|
Branches []dWorkflowNode `json:"branches,omitempty"`
|
|
Validated bool `json:"validated"`
|
|
}
|
|
|
|
for _, workflowRun := range workflowRuns {
|
|
node := &dWorkflowNode{}
|
|
for _, workflowOutput := range workflowOutputs {
|
|
if workflowOutput.GetString("runId") != workflowRun.Get("id") {
|
|
continue
|
|
}
|
|
|
|
if err := workflowOutput.UnmarshalJSONField("node", node); err != nil {
|
|
continue
|
|
}
|
|
|
|
if node.Type != "apply" {
|
|
node = &dWorkflowNode{}
|
|
continue
|
|
}
|
|
}
|
|
|
|
if node.Id == "" {
|
|
workflow, _ := app.FindRecordById("workflow", workflowRun.GetString("workflowId"))
|
|
if workflow != nil {
|
|
workflowRun.Set("detail", workflow.Get("content"))
|
|
} else {
|
|
workflowRun.Set("detail", make(map[string]any))
|
|
}
|
|
} else {
|
|
workflow, _ := app.FindRecordById("workflow", workflowRun.GetString("workflowId"))
|
|
if workflow != nil {
|
|
rootNode := &dWorkflowNode{}
|
|
if err := workflow.UnmarshalJSONField("content", rootNode); err != nil {
|
|
return err
|
|
}
|
|
|
|
rootNode.Next = node
|
|
workflowRun.Set("detail", rootNode)
|
|
} else {
|
|
rootNode := &dWorkflowNode{
|
|
Id: core.GenerateDefaultRandomId(),
|
|
Type: "start",
|
|
Name: "开始",
|
|
Config: map[string]any{
|
|
"trigger": "manual",
|
|
},
|
|
Next: node,
|
|
Validated: true,
|
|
}
|
|
workflowRun.Set("detail", rootNode)
|
|
}
|
|
}
|
|
|
|
if err := app.Save(workflowRun); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
// update collection `workflow_run`
|
|
{
|
|
collection, err := app.FindCollectionByNameOrId("workflow_run")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
collection.Fields.RemoveByName("logs")
|
|
|
|
if err := app.Save(collection); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}, func(app core.App) error {
|
|
return nil
|
|
})
|
|
}
|