Parcourir la source

feat: 保存重定向信息到 task.Properties

creamlike1024 il y a 1 mois
Parent
commit
347ad047f9
2 fichiers modifiés avec 9 ajouts et 1 suppressions
  1. 4 1
      model/task.go
  2. 5 0
      relay/relay_task.go

+ 4 - 1
model/task.go

@@ -72,7 +72,10 @@ func (t *Task) GetData(v any) error {
 }
 
 type Properties struct {
-	Input string `json:"input"`
+	Input             string `json:"input"`
+	IsModelMapped     bool   `json:"is_model_mapped,omitempty"`
+	OriginModelName   string `json:"origin_model_name,omitempty"`
+	UpstreamModelName string `json:"upstream_model_name,omitempty"`
 }
 
 func (m *Properties) Scan(val interface{}) error {

+ 5 - 0
relay/relay_task.go

@@ -214,6 +214,11 @@ func RelayTaskSubmit(c *gin.Context, info *relaycommon.RelayInfo) (taskErr *dto.
 	task.Quota = quota
 	task.Data = taskData
 	task.Action = info.Action
+	task.Properties = model.Properties{
+		IsModelMapped:     info.IsModelMapped,
+		UpstreamModelName: info.UpstreamModelName,
+		OriginModelName:   info.OriginModelName,
+	}
 	err = task.Insert()
 	if err != nil {
 		taskErr = service.TaskErrorWrapper(err, "insert_task_failed", http.StatusInternalServerError)