add task type manager

conf/kong.conf

 [dev]
 kongadminurl = "http://192.168.1.10:8001"
 
+[test]
+kongadminurl = "http://43.198.252.255:8001"
+
 [prod]
 kongadminurl = "http://172.31.12.187:8001"
\ No newline at end of file

conf/mysql.conf

@@ -5,11 +5,12 @@ mysqlhost = "192.168.1.211"
 mysqlport = 3306
 mysqldb = "liuxuzhong"
 
-; mysqluser = "ai"
-; mysqlpass = "RFnnKHRar5xk7TEF"
-; mysqlhost = "43.198.252.255"
-; mysqlport = 3306
-; mysqldb = "ai"
+[test]
+mysqluser = "ai"
+mysqlpass = "RFnnKHRar5xk7TEF"
+mysqlhost = "43.198.252.255"
+mysqlport = 3306
+mysqldb = "ai"
 
 [prod]
 mysqluser = "ai"

conf/postgres.conf

@@ -5,6 +5,13 @@ postgreshost = "192.168.1.10"
 postgresport = 8812
 postgresdb = "qdb"
 
+[test]
+postgresuser = "admin"
+postgrespass = "quest"
+postgreshost = "43.198.252.255"
+postgresport = 8812
+postgresdb = "qdb"
+
 [prod]
 postgresuser = "admin"
 postgrespass = "quest"

conf/redis.conf

@@ -7,6 +7,15 @@ balancehost = "192.168.1.10:6379"
 balancedb = 0
 balancepass = ""
 
+[test]
+tokenhost = "43.198.252.255:6379"
+tokendb = 0
+tokenpass = "iH0g2CqzjI6SfercGwsT"
+
+balancehost = "43.198.252.255:6379"
+balancedb = 0
+balancepass = "iH0g2CqzjI6SfercGwsT"
+
 
 [prod]
 tokenhost = "172.31.12.187:6379"

controllers/userlevel.go

@@ -19,7 +19,7 @@ func (server *UserLevelController) Levels() {
 		server.respond(http.StatusUnauthorized, err.Error())
 		return
 	}
-	qs := mysql.GetMysqlInstace().Ormer.QueryTable("user_level")
+	qs := mysql.GetMysqlInstace().Ormer.QueryTable("user_level").Filter("deleted", 0)
 	count, err := qs.Count()
 	logs.Debug("Levels = ", count)
 	var levels []*models.UserLevel
@@ -101,6 +101,8 @@ func (server *UserLevelController) UpdateLevel() {
 		return
 	}
 
+	request.CreatedTime = checkLevel.CreatedTime
+	request.UpdatedTime = time.Now()
 	_, err = mysql.GetMysqlInstace().Ormer.Update(&request)
 	if err != nil {
 		server.respond(models.BusinessFailed, "更新失败")
@@ -135,7 +137,8 @@ func (server *UserLevelController) DelLevel() {
 		server.respond(models.MissingParameter, "默认等级不能删除")
 		return
 	}
+	request.Deleted = 1
 
-	mysql.GetMysqlInstace().Ormer.Delete(&request)
+	mysql.GetMysqlInstace().Ormer.Update(&request)
 	server.respond(http.StatusOK, "删除成功")
 }

lastupdate.tmp

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

libs/cronjob/cronjob.go

 package cronjob
 
 import (
-	"ai_developer_admin/libs/odysseus"
 	"ai_developer_admin/libs/postgres"
 	"ai_developer_admin/libs/redis"
 	"ai_developer_admin/models"
@@ -13,10 +12,14 @@ import (
 
 var loopCronTask = cron.New(cron.WithSeconds())
 
+var HeatKey = "task:heat"
+
 func Start() {
 	//defer loopCronTask.Stop()
 
-	spec := "0 0 * * * ?" //"@every 1h"
+	//spec := "0 0 * * * ?" //"@every 1h"
+	spec := "@every 1h" //"@every 1h"
+	//spec := "*/1 * * * * ?" //"@every 1h"
 	loopCronTask.AddFunc(spec, func() {
 		logs.Debug("loopCronTask")
 		sql := "SELECT type,count(type) FROM bills GROUP BY type ORDER BY count DESC;"
@@ -24,39 +27,107 @@ func Start() {
 		if err != nil {
 			return
 		}
+		dataString, err := redis.GetDataToString(HeatKey)
 		var response []models.TaskHeat
+		if err != nil {
+			return
+		}
+		err = json.Unmarshal([]byte(dataString), &response)
+		if err != nil {
+			return
+		}
 		for _, value := range tasks {
 			taskId, err := strconv.Atoi(value.Type)
 			if err != nil {
 				continue
 			}
-			taskType, err1 := odysseus.GetTaskType(int64(taskId))
-			if err1 != nil {
-				continue
+			count, _ := strconv.Atoi(value.Count)
+			for _, taskType := range response {
+				if taskType.TaskId == taskId {
+					taskType.Count = int64(count)
 				}
-			var hardwareRequire models.Hardware
-			eer := json.Unmarshal([]byte(taskType.HardwareRequire), &hardwareRequire)
-			if eer != nil {
-
 			}
-			count, _ := strconv.Atoi(value.Count)
-			retask := models.TaskHeat{
-				TaskId:          taskId,
-				User:            taskType.Username,
-				Pwd:             taskType.Password,
-				Repository:      taskType.ImageUrl,
-				SignUrl:         taskType.SignUrl,
-				ImageName:       taskType.ImageName,
-				ImageId:         taskType.ImageId,
-				HardwareRequire: hardwareRequire,
-				Count:           int64(count),
-			}
-			response = append(response, retask)
 		}
+		//for _, value := range tasks {
+		//	taskId, err := strconv.Atoi(value.Type)
+		//	if err != nil {
+		//		continue
+		//	}
+		//	taskType, err1 := odysseus.GetTaskType(int64(taskId))
+		//	if err1 != nil {
+		//		continue
+		//	}
+		//	var hardwareRequire models.Hardware
+		//	eer := json.Unmarshal([]byte(taskType.HardwareRequire), &hardwareRequire)
+		//	if eer != nil {
+		//
+		//	}
+		//	count, _ := strconv.Atoi(value.Count)
+		//	retask := models.TaskHeat{
+		//		TaskId:          taskId,
+		//		User:            taskType.Username,
+		//		Pwd:             taskType.Password,
+		//		Repository:      taskType.ImageUrl,
+		//		SignUrl:         taskType.SignUrl,
+		//		ImageName:       taskType.ImageName,
+		//		ImageId:         taskType.ImageId,
+		//		HardwareRequire: hardwareRequire,
+		//		Count:           int64(count),
+		//	}
+		//	response = append(response, retask)
+		//}
 		data, err := json.Marshal(response)
 		if err == nil {
-			redis.SetKeyAndData("task:heat", string(data), 0)
+			redis.SetKeyAndData(HeatKey, string(data), 0)
 		}
 	})
 	loopCronTask.Start()
 }
+
+//func Start() {
+//	//defer loopCronTask.Stop()
+//
+//	spec := "*/1 * * * * ?" //"@every 1h"
+//	loopCronTask.AddFunc(spec, func() {
+//		logs.Debug("loopCronTask")
+//		//sql := "SELECT type,count(type) FROM bills GROUP BY type ORDER BY count DESC;"
+//		//tasks, err := postgres.CountTasks(sql)
+//		//if err != nil {
+//		//	return
+//		//}
+//		var response []models.TaskHeat
+//		//for _, value := range tasks {
+//		//	taskId, err := strconv.Atoi(value.Type)
+//		//	if err != nil {
+//		//		continue
+//		//	}
+//		taskType, err1 := odysseus.GetTaskType(int64(12))
+//		if err1 != nil {
+//			//continue
+//		}
+//		var hardwareRequire models.Hardware
+//		eer := json.Unmarshal([]byte(taskType.HardwareRequire), &hardwareRequire)
+//		if eer != nil {
+//
+//		}
+//		//count, _ := strconv.Atoi(value.Count)
+//		retask := models.TaskHeat{
+//			TaskId:          12,
+//			User:            taskType.Username,
+//			Pwd:             taskType.Password,
+//			Repository:      taskType.ImageUrl,
+//			SignUrl:         taskType.SignUrl,
+//			ImageName:       taskType.ImageName,
+//			ImageId:         taskType.ImageId,
+//			HardwareRequire: hardwareRequire,
+//			Count:           int64(0),
+//		}
+//		response = append(response, retask)
+//		//}
+//		data, err := json.Marshal(response)
+//		if err == nil {
+//			redis.SetKeyAndData("task:heat", string(data), 0)
+//		}
+//	})
+//	loopCronTask.Start()
+//}

libs/mysql/mysql.go

@@ -28,6 +28,7 @@ func GetMysqlInstace() *Singleton {
 
 func init() {
 	logs.Debug("mysql lib init")
+	//return
 	orm.Debug = true
 	if err := orm.RegisterDriver("mysql", orm.DRMySQL); err != nil {
 		logs.Error(err.Error())

libs/odysseus/odysseus.go

@@ -15,6 +15,7 @@ import (
 var syncinfo *cachedata.CacheData
 
 func init() {
+	//return
 	host, _ := beego.AppConfig.String("balancehost")
 	db, _ := beego.AppConfig.Int("balancedb")
 	pass, _ := beego.AppConfig.String("balancepass")
@@ -60,3 +61,7 @@ func GetUserBalance(id int64) (int64, error) {
 func GetTaskType(id int64) (*model.TaskType, error) {
 	return syncinfo.GetTaskWithId(id)
 }
+
+func PublichTaskUpdate(path string) error {
+	return syncinfo.PublichTaskUpdate(path)
+}

libs/postgres/postgres.go

@@ -13,6 +13,7 @@ import (
 var ormpost orm.Ormer
 
 func init() {
+	//return
 	logs.Debug("postgres lib init")
 	orm.Debug = true
 	if err := orm.RegisterDriver("postgres", orm.DRPostgres); err != nil {
@@ -55,6 +56,28 @@ func init() {
 	}
 }
 
+func QueryTset(sql string, args ...interface{}) ([]models.Bills, error) {
+	logs.Debug("QueryBills = ", sql)
+	qs := ormpost.Raw(sql, args)
+	var params []orm.Params
+	_, err := qs.Values(&params)
+	if err != nil {
+		return nil, err
+	}
+	arr, err := json.Marshal(params)
+	if err != nil {
+		return nil, err
+	}
+	var containers []models.Bills
+	logs.Debug("QueryBills = ", string(arr))
+	err = json.Unmarshal(arr, &containers)
+	if err != nil {
+		return nil, err
+	}
+
+	return containers, nil
+}
+
 func QueryBills(sql string) ([]models.Bills, error) {
 	logs.Debug("QueryBills = ", sql)
 	qs := ormpost.Raw(sql)

libs/redis/redis.go

 package redis
 
 import (
+	"fmt"
 	"github.com/beego/beego/v2/core/logs"
 	beego "github.com/beego/beego/v2/server/web"
+	"github.com/google/uuid"
 	"gopkg.in/redis.v5"
+	"strconv"
+	"strings"
 	"time"
 )
 
 var redisCache *redis.Client
 
 func init() {
+	//return
 	host, _ := beego.AppConfig.String("tokenhost")
 	db, _ := beego.AppConfig.Int("tokendb")
 	pass, _ := beego.AppConfig.String("tokenpass")
@@ -31,6 +36,59 @@ func createClient(redisHost string, password string, dataBase int) *redis.Client
 	return client
 }
 
+func tryAcquire(rs *redis.Client, lockKey string, lockTimeout time.Duration) (acquired bool, release func(), _ error) {
+	timeout := time.Now().Add(lockTimeout).UnixNano()
+	lockToken := fmt.Sprintf("%d,%s", timeout, uuid.New().String())
+
+	release = func() {
+		// Best effort to check we're releasing the lock we think we have. Note that it
+		// is still technically possible the lock token has changed between the GET and
+		// DEL since these are two separate operations, i.e. when the current lock happen
+		// to be expired at this very moment.
+		get, _ := rs.Get(lockKey).Result()
+		if get == lockToken {
+			_ = rs.Del(lockKey)
+		}
+	}
+	set, err := rs.SetNX(lockKey, lockToken, lockTimeout).Result()
+	if err != nil {
+		return false, nil, err
+	} else if set {
+		return true, release, nil
+	}
+
+	// We didn't get the lock, but we can check if the lock is expired.
+	currentLockToken, err := rs.Get(lockKey).Result()
+	if err == redis.Nil {
+		// Someone else got the lock and released it already.
+		return false, nil, nil
+	} else if err != nil {
+		return false, nil, err
+	}
+
+	currentTimeout, _ := strconv.ParseInt(strings.SplitN(currentLockToken, ",", 2)[0], 10, 64)
+	if currentTimeout > time.Now().UnixNano() {
+		// The lock is still valid.
+		return false, nil, nil
+	}
+
+	// The lock has expired, try to acquire it.
+	get, err := rs.GetSet(lockKey, lockToken).Result()
+	if err != nil {
+		return false, nil, err
+	} else if get != currentLockToken {
+		// Someone else got the lock
+		return false, nil, nil
+	}
+
+	// We got the lock.
+	return true, release, nil
+}
+
+func SetHeat(key string, value interface{}, time time.Duration) {
+
+}
+
 func SetKeyAndData(key string, value interface{}, time time.Duration) error {
 	err := redisCache.Set(key, value, time).Err()
 	if err != nil {

libs/utils/util.go

+package utils
+
+import "time"
+
+func InSameDay(t1, t2 time.Time) bool {
+	y1, m1, d1 := t1.Date()
+	y2, m2, d2 := t2.Date()
+
+	return y1 == y2 && m1 == m2 && d1 == d2
+}
+
+func SplitDate(beginDate, endDate, format string) []string {
+	bDate, _ := time.ParseInLocation(format, beginDate, time.Local)
+	eDate, _ := time.ParseInLocation(format, endDate, time.Local)
+	day := int(eDate.Sub(bDate).Hours() / 24)
+	dlist := make([]string, 0)
+	dlist = append(dlist, beginDate)
+	for i := 1; i < day; i++ {
+		result := bDate.AddDate(0, 0, i)
+		dlist = append(dlist, result.Format(format))
+	}
+	dlist = append(dlist, endDate)
+	return dlist
+}

main.go

@@ -4,9 +4,11 @@ import (
 	"ai_developer_admin/libs/cronjob"
 	"ai_developer_admin/libs/mysql"
 	_ "ai_developer_admin/routers"
+	"fmt"
 	beego "github.com/beego/beego/v2/server/web"
 	"github.com/beego/beego/v2/server/web/filter/cors"
 	_ "github.com/beego/beego/v2/server/web/session/redis"
+	"time"
 )
 
 func init() {
@@ -34,6 +36,28 @@ func main() {
 	//	logs.SetLogger(logs.AdapterFile, config)
 	//}
 
+	str := "2024-02-01T18:41:30"
+	layout := "2006-01-02T15:04:05"
+	t, _ := time.Parse(layout, str)
+	fmt.Print("t = %s", t)
+	//currentTime := time.Now()
+	////if appRequest.EndTime == nil {
+	//tempendTime := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 23, 59, 59, 0, currentTime.Location())
+	//endTime := &tempendTime
+	////}
+	////if appRequest.StartTime == nil {
+	//temp := fmt.Sprintf("-%dh", 24*7)
+	//m, _ := time.ParseDuration(temp)
+	//tempTime := currentTime.Add(m)
+	//tempTime = time.Date(tempTime.Year(), tempTime.Month(), tempTime.Day(), 0, 0, 0, 0, tempTime.Location())
+	//startTime := &tempTime
+	////}
+	//startTimeStr := fmt.Sprintf(startTime.Format("2006-01-02T15:04:05.000000Z"))
+	//endTimeStr := fmt.Sprintf(endTime.Format("2006-01-02T15:04:05.000000Z"))
+	//
+	//dates := utils.SplitDate(startTimeStr, endTimeStr, "2006-01-02T15:04:05.000000Z")
+	//logs.Debug("dates = ", dates)
+
 	cronjob.Start()
 	mysql.GetMysqlInstace()
 	//postgres.Query("")

models/app.go

@@ -43,8 +43,8 @@ type AppRequest struct {
 	Name      string `json:"name,omitempty"`
 	Page      int64  `json:"page,omitempty"`
 	Size      int64  `json:"size,omitempty"`
-	StartTime *time.Time `json:"start_time,omitempty"`
-	EndTime   *time.Time `json:"end_time,omitempty"`
+	StartTime string `json:"start_time,omitempty"`
+	EndTime   string `json:"end_time,omitempty"`
 	ProfitAcc string `json:"profit_acc,omitempty"`
 	WorkerAcc string `json:"worker_acc,omitempty"`
 }

models/task.go

@@ -4,26 +4,42 @@ import (
 	"time"
 )
 
-type Hardware struct {
-	DiskSize   int64    `json:"disk_size,omitempty"`
-	MemorySize int64    `json:"memory_size,omitempty"`
-	Gpus       []string `json:"gpus,omitempty"`
-}
+//type Hardware struct {
+//	DiskSize   int64    `json:"disk_size,omitempty"`
+//	MemorySize int64    `json:"memory_size,omitempty"`
+//	Gpus       []string `json:"gpus,omitempty"`
+//}
+//
+//type CMD struct {
+//	ImageName string `json:"image_name"`
+//	DockerCmd struct {
+//		ContainerPort string `json:"container_port"`
+//	} `json:"docker_cmd"`
+//	ApiUrl string `json:"api_url"`
+//}
 
-type CMD struct {
-	ImageName string `json:"image_name"`
-	DockerCmd struct {
-		ContainerPort string `json:"container_port"`
-	} `json:"docker_cmd"`
-	ApiUrl string `json:"api_url"`
+type ExampleType struct {
+	Title string `json:"title"`
+	Input struct {
+		Prompt string `json:"prompt,omitempty"`
+		Image  string `json:"image,omitempty"`
+	} `json:"input"`
+	Output struct {
+		Text string `json:"text,omitempty"`
+		Url  string `json:"url,omitempty"`
+	} `json:"output"`
 }
 
 type TaskType struct {
 	Id              int       `orm:"column(id);auto"`
+	Name            string    `json:"name";orm:"column(name);size(20)"`
+	Version         string    `json:"version,omitempty";orm:"column(version)"`
 	Desc            string    `json:"desc";orm:"column(desc);size(20)"`
 	Price           int64     `json:"price";orm:"column(price)"`
 	PublicKey       string    `json:"public_key";orm:"column(public_key)"`
 	Complexity      int       `json:"complexity";orm:"column(complexity)"`
+	Type            int       `json:"type";orm:"column(type)"`
+	Kind            int       `json:"kind";orm:"column(kind)"`
 	HardwareRequire string    `json:"hardware_require";orm:"column(hardware_require)"`
 	ImageId         string    `json:"image_id";orm:"column(image_id)"`
 	ImageUrl        string    `json:"image_url";orm:"column(image_url)"`
@@ -34,6 +50,12 @@ type TaskType struct {
 	SignUrl         string    `json:"sign_url";orm:"column(sign_url)"`
 	Username        string    `json:"username";orm:"column(username)"`
 	Password        string    `json:"password";orm:"column(password)"`
+	BaseModel       string    `json:"base_model";orm:"column(base_model)"`
+	Examples        string    `json:"examples";orm:"column(examples)"`
+	ApiDocUrl       string    `json:"api_doc_url";orm:"column(api_doc_url)"`
+	ApiDocContent   string    `json:"api_doc_content";orm:"column(api_doc_content)"`
+	Codes           string    `json:"codes";orm:"column(codes)"`
+	Tags            string    `json:"tags";orm:"column(tags)"`
 	CreatedTime     time.Time `json:"created_time";orm:"column(created_time);type(datetime)"`
 	UpdatedTime     time.Time `json:"updated_time";orm:"column(updated_time);type(datetime)"`
 	Deleted         int       `json:"deleted";orm:"column(deleted);size(1)"`
@@ -41,25 +63,67 @@ type TaskType struct {
 
 type NewTaskType struct {
 	Id              int           `orm:"column(id);auto"`
+	Name            string        `json:"name";orm:"column(desc)"`
+	BaseModel       string        `json:"base_model";orm:"column(desc)"`
+	Version         string        `json:"version";orm:"column(desc)"`
 	Desc            string        `json:"desc";orm:"column(desc);size(20)"`
-	Price           int64     `json:"price";orm:"column(price)"`
+	Price           float64       `json:"price";orm:"column(price)"`
 	PublicKey       string        `json:"public_key";orm:"column(public_key)"`
 	Complexity      int           `json:"complexity";orm:"column(complexity)"`
-	HardwareRequire Hardware  `json:"hardware_require";orm:"column(hardware_require)"`
+	Type            ModelType     `json:"type,omitempty";orm:"column(type)"`
+	Kind            int           `json:"kind";orm:"column(kind)"`
+	TypeDesc        string        `json:"type_desc"`
+	KindDesc        string        `json:"kind_desc"`
+	HardwareRequire interface{}   `json:"hardware_require";orm:"column(hardware_require)"`
 	ImageId         string        `json:"image_id";orm:"column(image_id)"`
 	ImageUrl        string        `json:"image_url";orm:"column(image_url)"`
-	Cmd             CMD       `json:"cmd";orm:"column(cmd)"`
+	Cmd             interface{}   `json:"cmd";orm:"column(cmd)"`
 	Workload        int           `json:"workload";orm:"column(workload)"`
 	ApiPath         string        `json:"api_path";orm:"column(api_path)"`
 	ImageName       string        `json:"image_name";orm:"column(image_name)"`
 	SignUrl         string        `json:"sign_url";orm:"column(sign_url)"`
 	Username        string        `json:"username";orm:"column(username)"`
 	Password        string        `json:"password";orm:"column(password)"`
+	Examples        []ExampleType `json:"examples,omitempty";orm:"column(examples)"`
+	ApiDocUrl       string        `json:"api_doc_url,omitempty";orm:"column(api_doc_url)"`
+	ApiDocContent   string        `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"`
+	Codes           interface{}   `json:"codes,omitempty";orm:"column(codes)"`
+	Tags            []string      `json:"tags,omitempty";orm:"column(tags)"`
 	CreatedTime     time.Time     `json:"created_time";orm:"column(created_time);type(datetime)"`
 	UpdatedTime     time.Time     `json:"updated_time";orm:"column(updated_time);type(datetime)"`
 	Deleted         int           `json:"deleted";orm:"column(deleted);size(1)"`
 }
 
+type Model struct {
+	Id            int    `orm:"column(id);auto"`
+	Tit           string `json:"tit";orm:"column(tit);size(20)"`
+	Version       string `json:"version,omitempty";orm:"column(version)"`
+	Content       string `json:"content";orm:"column(content);size(20)"`
+	Type          int    `json:"type";orm:"column(type)"`
+	ApiPath       string `json:"api_path";orm:"column(api_path)"`
+	BaseModel     string `json:"base_model";orm:"column(base_model)"`
+	Examples      string `json:"examples";orm:"column(examples)"`
+	ApiDocUrl     string `json:"api_doc_url";orm:"column(api_doc_url)"`
+	ApiDocContent string `json:"api_doc_content";orm:"column(api_doc_content)"`
+	Codes         string `json:"codes";orm:"column(codes)"`
+	Tags          string `json:"tags";orm:"column(tags)"`
+}
+
+type ResonseModel struct {
+	Id            int         `json:"id";orm:"column(id)"`
+	Tit           string      `json:"tit";orm:"column(tit);size(20)"`
+	Version       string      `json:"version,omitempty";orm:"column(version)"`
+	Content       string      `json:"content";orm:"column(content);size(20)"`
+	Type          int         `json:"type";orm:"column(type)"`
+	ApiPath       string      `json:"api_path";orm:"column(api_path)"`
+	BaseModel     string      `json:"base_model";orm:"column(base_model)"`
+	Examples      interface{} `json:"examples";orm:"column(examples)"`
+	ApiDocUrl     string      `json:"api_doc_url,omitempty";orm:"column(api_doc_url)"`
+	ApiDocContent string      `json:"api_doc_content,omitempty";orm:"column(api_doc_content)"`
+	Codes         interface{} `json:"codes";orm:"column(codes)"`
+	Tags          interface{} `json:"tags";orm:"column(tags)"`
+}
+
 type Bills struct {
 	//Key       int    `orm:"column(key);auto"`
 	Id        string `json:"id";orm:"column(id)"`
@@ -101,7 +165,7 @@ type TaskHeat struct {
 	Time            time.Time   `json:"time,omitempty"`
 	Count           int64       `json:"count,omitempty"`
 	ApiPath         string      `json:"api_path,omitempty"`
-	HardwareRequire Hardware  `json:"hardware_require,omitempty"`
+	HardwareRequire interface{} `json:"hardware_require,omitempty"`
 	ImageId         string      `json:"image_id,omitempty"`
 	Repository      string      `json:"repository,omitempty"`
 	ImageName       string      `json:"image_name,omitempty"`
@@ -109,6 +173,7 @@ type TaskHeat struct {
 	User            string      `json:"user,omitempty"`
 	Pwd             string      `json:"pwd,omitempty"`
 	TaskId          int         `json:"task_id,omitempty"`
+	Kind            int         `json:"kind";orm:"column(kind)"`
 }
 
 //type Tasks struct {
@@ -123,6 +188,51 @@ type TaskHeat struct {
 //}
 
 type AddTaskType struct {
-	Type   NewTaskType         `json:"type"`
+	TaskTypeIn NewTaskType         `json:"task_type"`
 	Levels     []UserLevelTaskType `json:"levels"`
 }
+
+type ModelType int
+
+const (
+	TXTTOIMG ModelType = iota + 1
+	TXTTOTXT
+	TXTTOVIDEO
+)
+
+func (m ModelType) String() string {
+	switch m {
+	case TXTTOIMG:
+		return "txt2img"
+	case TXTTOTXT:
+		return "txt2txt"
+	case TXTTOVIDEO:
+		return "txt2video"
+	default:
+		return "未知类型"
+	}
+}
+
+type TaskKind int
+
+const (
+	SystemTask   TaskKind = iota
+	ComputeTask           = 1
+	CustomTask            = 2
+	StandardTask          = 3
+)
+
+func (m TaskKind) String() string {
+	switch m {
+	case SystemTask:
+		return "系统任务"
+	case ComputeTask:
+		return "计算任务"
+	case CustomTask:
+		return "自定义任务"
+	case StandardTask:
+		return "标准任务"
+	default:
+		return "未知类型"
+	}
+}

models/user.go

@@ -60,7 +60,7 @@ type UserLevel struct {
 }
 
 type UserLevelTaskType struct {
-	ID                 int       `orm:"column(id);auto"`
+	Id                 int       `orm:"column(id);auto"`
 	FreeCallCountDay   int       `json:"free_call_count_day"orm:"column(free_call_count_day)"`
 	FreeCallCountMonth int       `json:"free_call_count_month"orm:"column(free_call_count_month)"`
 	FreeCallCountYear  int       `json:"free_call_count_year"orm:"column(free_call_count_year)"`