add whitelist

controllers/task.go

@@ -12,6 +12,7 @@ import (
 	"fmt"
 	"github.com/beego/beego/orm"
 	"github.com/beego/beego/v2/core/logs"
+	"github.com/odysseus/cache/model"
 	"net/http"
 	"sort"
 	"strconv"
@@ -1492,7 +1493,6 @@ func (server *TaskController) UpdateTaskType() {
 		Username:        appRequest.TaskTypeIn.Username,
 		Password:        appRequest.TaskTypeIn.Password,
 		Category:        appRequest.TaskTypeIn.Category,
-		CreatedTime:     timestamp,
 		UpdatedTime:     timestamp,
 	}
 
@@ -1545,6 +1545,32 @@ func (server *TaskController) UpdateTaskType() {
 	//	server.respond(models.BusinessFailed, "填加 task type 失败")
 	//	return
 	//}
+
+	task := model.TaskType{
+		ID:              int64(dbType.Id),
+		BaseModel:       dbType.BaseModel,
+		Model:           dbType.Model,
+		Desc:            dbType.Desc,
+		Price:           price,
+		PublicKey:       dbType.PublicKey,
+		Complexity:      int64(dbType.Complexity),
+		HardwareRequire: string(hardwareRequire),
+		ImageId:         dbType.ImageId,
+		ImageUrl:        dbType.ImageUrl,
+		Type:            int(dbType.Type),
+		Kind:            dbType.Kind,
+		Cmd:             string(cmd),
+		Workload:        int64(dbType.Workload),
+		ApiPath:         dbType.ApiPath,
+		ImageName:       dbType.ImageName,
+		SignUrl:         dbType.SignUrl,
+		Username:        dbType.Username,
+		Password:        dbType.Password,
+		CreatedTime:     appRequest.TaskTypeIn.CreatedTime,
+		UpdatedTime:     dbType.UpdatedTime,
+	}
+
+	odysseus.SetTaskDataToRedis(&task)
 	odysseus.PublichTaskUpdate(dbType.ApiPath)
 	server.respond(http.StatusOK, "修改 task type 成功")
 }

lastupdate.tmp

-{"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709619166614368556}
+{"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709621001721511850}
\ No newline at end of file

libs/cronjob/cronjob.go

@@ -29,7 +29,7 @@ func Start() {
 	//defer loopCronTask.Stop()
 	startHeatKey()
 	startDebit()
-	startRegistBackend()
+	//startRegistBackend()
 }
 
 func startHeatKey() {
@@ -187,7 +187,7 @@ func startDebit() {
 }
 
 func startRegistBackend() {
-	spec := "@every 1h"
+	spec := "@every 1m"
 	registryTask.AddFunc(spec, func() {
 		logs.Debug("startRegistBackend")
 		registry.RegistryBackend()

libs/odysseus/odysseus.go

@@ -58,6 +58,10 @@ func GetUserBalance(id int64) (int64, error) {
 	return user.Balance, err
 }
 
+func SetTaskDataToRedis(task *model.TaskType) error {
+	return syncinfo.SetTaskDataToRedis(task)
+}
+
 func GetTaskType(id int64) (*model.TaskType, error) {
 	return syncinfo.GetTaskWithId(id)
 }

libs/registry/registry.go

@@ -7,7 +7,6 @@ import (
 	"github.com/odysseus/service-registry/query"
 	registor "github.com/odysseus/service-registry/registry"
 	"os"
-	"time"
 )
 
 var querier *query.ServiceQuerier
@@ -75,6 +74,6 @@ func (d backendService) DetailInfo() (json.RawMessage, error) {
 func RegistryBackend() {
 	r := registor.NewRegistry(redisParam, backendService{})
 	r.Start()
-	time.Sleep(time.Second * 5)
-	r.Stop()
+	//time.Sleep(time.Second * 5)
+	//r.Stop()
 }

main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"ai_developer_admin/libs/cronjob"
 	"ai_developer_admin/libs/mysql"
+	"ai_developer_admin/libs/registry"
 	_ "ai_developer_admin/routers"
 	"fmt"
 	beego "github.com/beego/beego/v2/server/web"
@@ -63,6 +64,8 @@ func main() {
 	//postgres.Query("")
 	beego.Run()
 
+	registry.RegistryBackend()
+
 	//id, err := snowflake.NextId()
 	//if err != nil {
 	//	logs.Debug("dates = ", err.Error())