floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: update AiInfo task_ai_async and 

Former-commit-id: c4a6452facc465db3c0753ea96e8e972b7eb1bf9
This commit is contained in:
qiwang 2024-10-24 17:41:50 +08:00
parent 1da1c53ac4
commit 92451c41d4
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
client/types.go
View File

@ -136,20 +136,20 @@ type CloudInfo struct {
} }
type AiInfo struct { type AiInfo struct {
Id int64 `json:"id"` // id Id int64 `json:"id"` // id
AdapterId int64 `json:"adapterId,omitempty,optional"` AdapterId int64 `json:"adapterId,omitempty,optional"`
AdapterName string `json:"adapterName,omitempty,optional"` AdapterName string `json:"adapterName,omitempty,optional"`
ClusterId int64 `json:"clusterId,omitempty,optional"` ClusterId int64 `json:"clusterId,omitempty,optional"`
ClusterIds []int64 `json:"clusterIds,omitempty,optional"` //ClusterIds []int64 `json:"clusterIds,omitempty,optional"`
TaskId int64 `json:"taskId,omitempty"` TaskId int64 `json:"taskId,omitempty"`
ClusterName string `json:"clusterName,omitempty,optional"` ClusterName string `json:"clusterName,omitempty,optional"`
ImageId string `json:"imageId,omitempty"` ImageId string `json:"imageId,omitempty"`
//ResourceId string `json:"resourceId,omitempty"` //ResourceId string `json:"resourceId,omitempty"`
AlgorithmId string `json:"algorithmId,omitempty"` //AlgorithmId string `json:"algorithmId,omitempty"`
//MetadataName string `json:"metadataName,omitempty"` //MetadataName string `json:"metadataName,omitempty"`
Command string `json:"command,omitempty"` Command string `json:"command,omitempty"`
Environments string `json:"environments,omitempty"` //Environments string `json:"environments,omitempty"`
Parameters string `json:"parameters,omitempty"` //Parameters string `json:"parameters,omitempty"`
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
Status string `json:"status,omitempty"` Status string `json:"status,omitempty"`

6
pkg/models/taskaiasynchronousmodel_gen.go
View File

@ -48,7 +48,7 @@ type (
Result string `db:"result"` // 运行结果 Result string `db:"result"` // 运行结果
DeletedAt string `db:"deleted_at"` // 删除时间 DeletedAt string `db:"deleted_at"` // 删除时间
ImageId string `db:"image_id"` // 镜像id ImageId string `db:"image_id"` // 镜像id
Cmd string `db:"cmd"` // 命令行 Command string `db:"command"` // 命令行
FlavorId string `db:"flavor_id"` // 训练作业资源规格id FlavorId string `db:"flavor_id"` // 训练作业资源规格id
Status string `db:"status"` // 任务状态 Status string `db:"status"` // 任务状态
} }
@ -83,13 +83,13 @@ func (m *defaultTaskAiAsynchronousModel) FindOne(ctx context.Context, id int64)
func (m *defaultTaskAiAsynchronousModel) Insert(ctx context.Context, data *TaskAiAsynchronous) (sql.Result, error) { func (m *defaultTaskAiAsynchronousModel) Insert(ctx context.Context, data *TaskAiAsynchronous) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiAsynchronousRowsExpectAutoSet) query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiAsynchronousRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.Id, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Cmd, data.FlavorId, data.Status) ret, err := m.conn.ExecCtx(ctx, query, data.Id, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Command, data.FlavorId, data.Status)
return ret, err return ret, err
} }
func (m *defaultTaskAiAsynchronousModel) Update(ctx context.Context, data *TaskAiAsynchronous) error { func (m *defaultTaskAiAsynchronousModel) Update(ctx context.Context, data *TaskAiAsynchronous) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiAsynchronousRowsWithPlaceHolder) query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiAsynchronousRowsWithPlaceHolder)
_, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Cmd, data.FlavorId, data.Status, data.Id) _, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.StartTime, data.RunningTime, data.Result, data.DeletedAt, data.ImageId, data.Command, data.FlavorId, data.Status, data.Id)
return err return err
} }