JointCloud
/
pcm-coordinator
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

Merge branch 'master' of https://gitlink.org.cn/JointCloud/pcm-coordinator 

Former-commit-id: 0a7983d935
pull/326/head
zhangwei 1 year ago
parent
25a394bc95 92451c41d4
commit
9c1528a328
10 changed files with 350 additions and 140 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +55
    -55
      client/types.go
  2. +3
    -40
      internal/logic/core/pulltaskinfologic.go
  3. +2
    -2
      internal/logic/core/pushtaskinfologic.go
  4. +13
    -1
      internal/logic/inference/getadaptersbymodellogic.go
  5. +0
    -11
      internal/storeLink/modelarts.go
  6. +29
    -0
      pkg/models/taskaiasynchronousmodel.go
  7. +98
    -0
      pkg/models/taskaiasynchronousmodel_gen.go
  8. +23
    -31
      pkg/models/taskaimodel_gen.go
  9. +29
    -0
      pkg/models/taskaimodelartsmodel.go
  10. +98
    -0
      pkg/models/taskaimodelartsmodel_gen.go

+ 55
- 55
client/types.go View File

@@ -136,61 +136,61 @@ type CloudInfo struct {
}

type AiInfo struct {
Id int64 `json:"id"` // id
AdapterId int64 `json:"adapterId,omitempty,optional"`
AdapterName string `json:"adapterName,omitempty,optional"`
ClusterId int64 `json:"clusterId,omitempty,optional"`
ClusterIds []int64 `json:"clusterIds,omitempty,optional"`
TaskId int64 `json:"taskId,omitempty"`
TaskName string `json:"taskName,omitempty"`
Replica int32 `json:"replica,omitempty"`
ResourceType string `json:"resourceType,omitempty"`
CpuCoreNum int32 `json:"cpuCoreNum,omitempty"`
TaskType string `json:"taskType,omitempty"`
DatasetsName string `json:"datasetsName,omitempty"`
ProjectId string `json:"project_id,omitempty"`
StrategyName string `json:"strategyName,omitempty"`
ClusterToStaticWeight map[string]int32 `json:"clusterToStaticWeight,omitempty"`
Tops float64 `json:"tops,omitempty"`
ComputeCard string `json:"computeCard,omitempty,optional"`
CodeType string `json:"codeType,omitempty,optional"`
ClusterName string `json:"clusterName,omitempty,optional"`
ModelName string `json:"ModelName,omitempty,optional"`
AlgorithmName string `json:"algorithmName,omitempty,optional"`
Strategy string `json:"strategy,omitempty"`
ImageId string `json:"imageId,omitempty"`
SpecId string `json:"specId,omitempty"`
DatasetsId string `json:"datasetsId,omitempty"`
CodeId string `json:"codeId,omitempty"`
ResourceId string `json:"resourceId,omitempty"`
AlgorithmId string `json:"algorithmId,omitempty"`
MetadataName string `json:"metadataName,omitempty"`
Cmd string `json:"cmd,omitempty"`
Envs []string `json:"envs,omitempty"`
Params []string `json:"params,omitempty"`
Environments string `json:"environments,omitempty"`
Parameters string `json:"parameters,omitempty"`
Name string `json:"name,omitempty"`
Status string `json:"status,omitempty"`
StartTime string `json:"startTime,omitempty"`
RunningTime int64 `json:"runningTime,omitempty"`
Result string `json:"result,omitempty"`
JobId string `json:"jobId,omitempty"`
Datasets string `json:"datasets,omitempty"`
AlgorithmCode string `json:"algorithmCode,omitempty"`
Image string `json:"image,omitempty"`
CreateTime string `json:"createTime,omitempty"`
ImageUrl string `json:"imageUrl,omitempty"`
Command string `json:"command,omitempty"`
FlavorId string `json:"flavorId,omitempty"`
SubscriptionId string `json:"subscriptionId,omitempty"`
ItemVersionId string `json:"itemVersionId,omitempty"`
ObsUrl string `json:"obsUrl,omitempty"`
Id int64 `json:"id"` // id
AdapterId int64 `json:"adapterId,omitempty,optional"`
AdapterName string `json:"adapterName,omitempty,optional"`
ClusterId int64 `json:"clusterId,omitempty,optional"`
//ClusterIds []int64 `json:"clusterIds,omitempty,optional"`
TaskId int64 `json:"taskId,omitempty"`
ClusterName string `json:"clusterName,omitempty,optional"`
ImageId string `json:"imageId,omitempty"`
//ResourceId string `json:"resourceId,omitempty"`
//AlgorithmId string `json:"algorithmId,omitempty"`
//MetadataName string `json:"metadataName,omitempty"`
Command string `json:"command,omitempty"`
//Environments string `json:"environments,omitempty"`
//Parameters string `json:"parameters,omitempty"`
Name string `json:"name,omitempty"`
Status string `json:"status,omitempty"`
StartTime string `json:"startTime,omitempty"`
//RunningTime int64 `json:"runningTime,omitempty"`
JobId string `json:"jobId,omitempty"`
FlavorId string `json:"flavorId,omitempty"`
//TaskName string `json:"taskName,omitempty"`
//Replica int32 `json:"replica,omitempty"`
//ResourceType string `json:"resourceType,omitempty"`
//CpuCoreNum int32 `json:"cpuCoreNum,omitempty"`
//TaskType string `json:"taskType,omitempty"`
//DatasetsName string `json:"datasetsName,omitempty"`
//ProjectId string `json:"project_id,omitempty"`
//StrategyName string `json:"strategyName,omitempty"`
//ClusterToStaticWeight map[string]int32 `json:"clusterToStaticWeight,omitempty"`
//Tops float64 `json:"tops,omitempty"`
//ComputeCard string `json:"computeCard,omitempty,optional"`
//CodeType string `json:"codeType,omitempty,optional"`
//ModelName string `json:"ModelName,omitempty,optional"`
//AlgorithmName string `json:"algorithmName,omitempty,optional"`
//Strategy string `json:"strategy,omitempty"`
//Envs []string `json:"envs,omitempty"`
//Params []string `json:"params,omitempty"`
//SpecId string `json:"specId,omitempty"`
//DatasetsId string `json:"datasetsId,omitempty"`
//CodeId string `json:"codeId,omitempty"`
//Result string `json:"result,omitempty"`
//Datasets string `json:"datasets,omitempty"`
//AlgorithmCode string `json:"algorithmCode,omitempty"`
//Image string `json:"image,omitempty"`
//CreateTime string `json:"createTime,omitempty"`
//ImageUrl string `json:"imageUrl,omitempty"`
//SubscriptionId string `json:"subscriptionId,omitempty"`
//ItemVersionId string `json:"itemVersionId,omitempty"`
//ObsUrl string `json:"obsUrl,omitempty"`
}

type VmInfo struct {


+ 3
- 40
internal/logic/core/pulltaskinfologic.go View File

@@ -2,18 +2,14 @@ package core

import (
"context"
"encoding/json"
"fmt"
"github.com/jinzhu/copier"
"github.com/zeromicro/go-zero/core/logx"
clientCore "gitlink.org.cn/JointCloud/pcm-coordinator/client"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models"
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/models/cloud"
"gitlink.org.cn/JointCloud/pcm-coordinator/pkg/utils"
"gorm.io/gorm"
"log"

"github.com/zeromicro/go-zero/core/logx"
"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
)

type PullTaskInfoLogic struct {
@@ -78,45 +74,12 @@ func (l *PullTaskInfoLogic) PullTaskInfo(req *clientCore.PullTaskInfoReq) (*clie
}

case 1:
var aiModelList []models.TaskAi
var aiModelList []models.TaskAiAsynchronous
err := findModelList(req.AdapterId, l.svcCtx.DbEngin, &aiModelList)
if err != nil {
return nil, err
}
utils.Convert(aiModelList, &resp.AiInfoList)
if len(resp.AiInfoList) > 0 {
for i, aiInfo := range aiModelList {
if resp.AiInfoList[i].Environments != "" {
// 定义一个map来存储解析后的JSON数据
var result map[string]interface{}
// 解析JSON字符串
err := json.Unmarshal([]byte(resp.AiInfoList[i].Environments), &result)
if err != nil {
log.Fatalf("Error parsing JSON: %v", err)
}
// 如果你需要将解析后的map再次转换为JSON字符串，可以使用json.MarshalIndent
formattedJSON, err := json.MarshalIndent(result, "", " ")
aiInfo.Environments = string(formattedJSON)
fmt.Println(aiInfo.Environments)
resp.AiInfoList[i].Environments = aiInfo.Environments
}
if resp.AiInfoList[i].Parameters != "" {
// 定义一个map来存储解析后的JSON数据
var result []interface{}
// 解析JSON字符串
err := json.Unmarshal([]byte(resp.AiInfoList[i].Parameters), &result)
if err != nil {
log.Fatalf("Error parsing JSON: %v", err)
}
// 如果你需要将解析后的map再次转换为JSON字符串，可以使用json.MarshalIndent
formattedJSON, err := json.MarshalIndent(result, "", " ")
aiInfo.Parameters = string(formattedJSON)
fmt.Println(aiInfo.Parameters)
resp.AiInfoList[i].Parameters = aiInfo.Parameters
}

}
}
}
return &resp, nil
}


+ 2
- 2
internal/logic/core/pushtaskinfologic.go View File

@@ -89,8 +89,8 @@ func (l *PushTaskInfoLogic) PushTaskInfo(req *clientCore.PushTaskInfoReq) (*clie
}
case 1:
for _, aiInfo := range req.AiInfoList {
l.svcCtx.DbEngin.Exec("update task_ai set status = ?,start_time = ?,project_id = ?,job_id = ? where participant_id = ? and task_id = ? and name = ?",
aiInfo.Status, aiInfo.StartTime, aiInfo.ProjectId, aiInfo.JobId, req.AdapterId, aiInfo.TaskId, aiInfo.Name)
l.svcCtx.DbEngin.Exec("update task_ai_asynchronous set status = ?,start_time = ?,job_id = ? where cluster_id = ? and task_id = ? and name = ?",
aiInfo.Status, aiInfo.StartTime, aiInfo.JobId, aiInfo.ClusterId, aiInfo.TaskId, aiInfo.Name)
noticeInfo := clientCore.NoticeInfo{
TaskId: aiInfo.TaskId,
AdapterId: aiInfo.AdapterId,


+ 13
- 1
internal/logic/inference/getadaptersbymodellogic.go View File

@@ -38,7 +38,19 @@ func (l *GetAdaptersByModelLogic) GetAdaptersByModel(req *types.GetAdaptersByMod
}

for _, cluster := range clusters.List {
exist := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id][cluster.Id].CheckModelExistence(l.ctx, req.ModelName, req.ModelType)

cmap, found := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id]
if !found {
continue
}

iCluster, found := cmap[cluster.Id]
if !found {
continue
}

exist := iCluster.CheckModelExistence(l.ctx, req.ModelName, req.ModelType)

if exist {
c := &types.ClusterAvail{
ClusterId: cluster.Id,


+ 0
- 11
internal/storeLink/modelarts.go View File

@@ -28,7 +28,6 @@ import (
"gitlink.org.cn/JointCloud/pcm-modelarts/client/modelartsservice"
"gitlink.org.cn/JointCloud/pcm-modelarts/modelarts"
modelartsclient "gitlink.org.cn/JointCloud/pcm-modelarts/modelarts"
"gorm.io/gorm"
"log"
"mime/multipart"
"strconv"
@@ -55,7 +54,6 @@ type ModelArtsLink struct {
Version string
ModelId string
ModelType string
DbEngin *gorm.DB
}

// Version 结构体表示版本号
@@ -63,10 +61,6 @@ type Version struct {
Major, Minor, Patch int
}

type AiStorage struct {
DbEngin *gorm.DB
}

// ParseVersion 从字符串解析版本号
func ParseVersion(versionStr string) (*Version, error) {
parts := strings.Split(versionStr, ".")
@@ -180,11 +174,6 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri
Platform: m.platform,
}
resp, err := m.modelArtsRpc.CreateTrainingJob(ctx, req)
//tx := m.DbEngin.Create(adapterId)
/*if tx.Error != nil {
return tx.Error, nil
}*/

if err != nil {
return nil, err
}


+ 29
- 0
pkg/models/taskaiasynchronousmodel.go View File

@@ -0,0 +1,29 @@
package models

import "github.com/zeromicro/go-zero/core/stores/sqlx"

var _ TaskAiAsynchronousModel = (*customTaskAiAsynchronousModel)(nil)

type (
// TaskAiAsynchronousModel is an interface to be customized, add more methods here,
// and implement the added methods in customTaskAiAsynchronousModel.
TaskAiAsynchronousModel interface {
taskAiAsynchronousModel
withSession(session sqlx.Session) TaskAiAsynchronousModel
}

customTaskAiAsynchronousModel struct {
*defaultTaskAiAsynchronousModel
}
)

// NewTaskAiAsynchronousModel returns a model for the database table.
func NewTaskAiAsynchronousModel(conn sqlx.SqlConn) TaskAiAsynchronousModel {
return &customTaskAiAsynchronousModel{
defaultTaskAiAsynchronousModel: newTaskAiAsynchronousModel(conn),
}
}

func (m *customTaskAiAsynchronousModel) withSession(session sqlx.Session) TaskAiAsynchronousModel {
return NewTaskAiAsynchronousModel(sqlx.NewSqlConnFromSession(session))
}

+ 98
- 0
pkg/models/taskaiasynchronousmodel_gen.go View File

@@ -0,0 +1,98 @@
// Code generated by goctl. DO NOT EDIT.

package models

import (
"context"
"database/sql"
"fmt"
"strings"

"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)

var (
taskAiAsynchronousFieldNames = builder.RawFieldNames(&TaskAiAsynchronous{})
taskAiAsynchronousRows = strings.Join(taskAiAsynchronousFieldNames, ",")
taskAiAsynchronousRowsExpectAutoSet = strings.Join(stringx.Remove(taskAiAsynchronousFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
taskAiAsynchronousRowsWithPlaceHolder = strings.Join(stringx.Remove(taskAiAsynchronousFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
)

type (
taskAiAsynchronousModel interface {
Insert(ctx context.Context, data *TaskAiAsynchronous) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*TaskAiAsynchronous, error)
Update(ctx context.Context, data *TaskAiAsynchronous) error
Delete(ctx context.Context, id int64) error
}

defaultTaskAiAsynchronousModel struct {
conn sqlx.SqlConn
table string
}

TaskAiAsynchronous struct {
Id int64 `db:"id"` // 训练作业资源规格id
TaskId int64 `db:"task_id"` // 任务id
AdapterId int64 `db:"adapter_id"` // 执行任务的适配器id
AdapterName string `db:"adapter_name"` // 适配器名称
ClusterId int64 `db:"cluster_id"` // 集群id
ClusterName string `db:"cluster_name"` // 集群名称
Name string `db:"name"` // 任务名
Replica int64 `db:"replica"` // 执行数
JobId string `db:"job_id"` // 集群返回任务id
StartTime string `db:"start_time"` // 开始时间
RunningTime string `db:"running_time"` // 运行时间
Result string `db:"result"` // 运行结果
DeletedAt string `db:"deleted_at"` // 删除时间
ImageId string `db:"image_id"` // 镜像id
Command string `db:"command"` // 命令行
FlavorId string `db:"flavor_id"` // 训练作业资源规格id
Status string `db:"status"` // 任务状态
}
)

func newTaskAiAsynchronousModel(conn sqlx.SqlConn) *defaultTaskAiAsynchronousModel {
return &defaultTaskAiAsynchronousModel{
conn: conn,
table: "`task_ai_asynchronous`",
}
}

func (m *defaultTaskAiAsynchronousModel) Delete(ctx context.Context, id int64) error {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
_, err := m.conn.ExecCtx(ctx, query, id)
return err
}

func (m *defaultTaskAiAsynchronousModel) FindOne(ctx context.Context, id int64) (*TaskAiAsynchronous, error) {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", taskAiAsynchronousRows, m.table)
var resp TaskAiAsynchronous
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
switch err {
case nil:
return &resp, nil
case sqlx.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}

func (m *defaultTaskAiAsynchronousModel) Insert(ctx context.Context, data *TaskAiAsynchronous) (sql.Result, error) {
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.Command, data.FlavorId, data.Status)
return ret, err
}

func (m *defaultTaskAiAsynchronousModel) Update(ctx context.Context, data *TaskAiAsynchronous) error {
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.Command, data.FlavorId, data.Status, data.Id)
return err
}

func (m *defaultTaskAiAsynchronousModel) tableName() string {
return m.table
}

+ 23
- 31
pkg/models/taskaimodel_gen.go View File

@@ -35,34 +35,26 @@ type (
}

TaskAi struct {
Id int64 `db:"id"` // id
TaskId int64 `db:"task_id"` // 任务id
AdapterId int64 `db:"adapter_id"` // 适配器id
AdapterName string `db:"adapter_name"` // 适配器名称
ClusterId int64 `db:"cluster_id"` // 集群id
ClusterName string `db:"cluster_name"` // 集群名称
Name string `db:"name"` // 任务名
Replica int64 `db:"replica"` // 执行数
JobId string `db:"job_id"` // 集群返回任务id
Strategy string `db:"strategy"` // 主任务使用策略
Status string `db:"status"` // 任务状态
Msg string `db:"msg"` // 集群返回任务信息
CommitTime time.Time `db:"commit_time"` // 提交时间
StartTime string `db:"start_time"` // 开始时间
EndTime string `db:"end_time"` // 结束时间
TaskType string `db:"task_type"`
DeletedAt time.Time `db:"deleted_at"`
Card string `db:"card"`
Remark string `db:"remark"` // 备注
InferUrl string `db:"infer_url"`
ModelName string `db:"model_name"`
AlgorithmId string `db:"algorithm_id"` // 算法id
ImageId string `db:"image_id"` // 镜像id
Command string `db:"command"` // 启动命令
Environments string `db:"environments"` // 训练作业的环境变量
Parameters string `db:"parameters"` // 训练作业的运行参数
FlavorId string `db:"flavor_id"` // 规格id
MetadataName string `db:"metadata_name"` // 训练作业名称
Id int64 `db:"id"` // id
TaskId int64 `db:"task_id"` // 任务id
AdapterId int64 `db:"adapter_id"` // 适配器id
AdapterName string `db:"adapter_name"` // 适配器名称
ClusterId int64 `db:"cluster_id"` // 集群id
ClusterName string `db:"cluster_name"` // 集群名称
Name string `db:"name"` // 任务名
Replica int64 `db:"replica"` // 执行数
JobId string `db:"job_id"` // 集群返回任务id
Strategy string `db:"strategy"` // 主任务使用策略
Status string `db:"status"` // 任务状态
Msg string `db:"msg"` // 集群返回任务信息
CommitTime time.Time `db:"commit_time"` // 提交时间
StartTime string `db:"start_time"` // 开始时间
EndTime string `db:"end_time"` // 结束时间
TaskType string `db:"task_type"`
DeletedAt *time.Time `db:"deleted_at"`
Card string `db:"card"`
InferUrl string `db:"infer_url"`
ModelName string `db:"model_name"`
}
)

@@ -94,14 +86,14 @@ func (m *defaultTaskAiModel) FindOne(ctx context.Context, id int64) (*TaskAi, er
}

func (m *defaultTaskAiModel) Insert(ctx context.Context, data *TaskAi) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType, data.DeletedAt, data.Card, data.Remark, data.InferUrl, data.ModelName, data.AlgorithmId, data.ImageId, data.Command, data.Environments, data.Parameters, data.FlavorId, data.MetadataName)
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiRowsExpectAutoSet)
ret, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType, data.DeletedAt, data.Card, data.InferUrl, data.ModelName)
return ret, err
}

func (m *defaultTaskAiModel) Update(ctx context.Context, data *TaskAi) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiRowsWithPlaceHolder)
_, err := m.conn.ExecCtx(ctx, query, data.TaskId, data.AdapterId, data.AdapterName, data.ClusterId, data.ClusterName, data.Name, data.Replica, data.JobId, data.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType, data.DeletedAt, data.Card, data.Remark, data.InferUrl, data.ModelName, data.AlgorithmId, data.ImageId, data.Command, data.Environments, data.Parameters, data.FlavorId, data.MetadataName, 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.Strategy, data.Status, data.Msg, data.CommitTime, data.StartTime, data.EndTime, data.TaskType, data.DeletedAt, data.Card, data.InferUrl, data.ModelName, data.Id)
return err
}



+ 29
- 0
pkg/models/taskaimodelartsmodel.go View File

@@ -0,0 +1,29 @@
package models

import "github.com/zeromicro/go-zero/core/stores/sqlx"

var _ TaskAiModelartsModel = (*customTaskAiModelartsModel)(nil)

type (
// TaskAiModelartsModel is an interface to be customized, add more methods here,
// and implement the added methods in customTaskAiModelartsModel.
TaskAiModelartsModel interface {
taskAiModelartsModel
withSession(session sqlx.Session) TaskAiModelartsModel
}

customTaskAiModelartsModel struct {
*defaultTaskAiModelartsModel
}
)

// NewTaskAiModelartsModel returns a model for the database table.
func NewTaskAiModelartsModel(conn sqlx.SqlConn) TaskAiModelartsModel {
return &customTaskAiModelartsModel{
defaultTaskAiModelartsModel: newTaskAiModelartsModel(conn),
}
}

func (m *customTaskAiModelartsModel) withSession(session sqlx.Session) TaskAiModelartsModel {
return NewTaskAiModelartsModel(sqlx.NewSqlConnFromSession(session))
}

+ 98
- 0
pkg/models/taskaimodelartsmodel_gen.go View File

@@ -0,0 +1,98 @@
// Code generated by goctl. DO NOT EDIT.

package models

import (
"context"
"database/sql"
"fmt"
"strings"

"github.com/zeromicro/go-zero/core/stores/builder"
"github.com/zeromicro/go-zero/core/stores/sqlx"
"github.com/zeromicro/go-zero/core/stringx"
)

var (
taskAiModelartsFieldNames = builder.RawFieldNames(&TaskAiModelarts{})
taskAiModelartsRows = strings.Join(taskAiModelartsFieldNames, ",")
taskAiModelartsRowsExpectAutoSet = strings.Join(stringx.Remove(taskAiModelartsFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
taskAiModelartsRowsWithPlaceHolder = strings.Join(stringx.Remove(taskAiModelartsFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
)

type (
taskAiModelartsModel interface {
Insert(ctx context.Context, data *TaskAiModelarts) (sql.Result, error)
FindOne(ctx context.Context, id int64) (*TaskAiModelarts, error)
Update(ctx context.Context, data *TaskAiModelarts) error
Delete(ctx context.Context, id int64) error
}

defaultTaskAiModelartsModel struct {
conn sqlx.SqlConn
table string
}

TaskAiModelarts struct {
Id int64 `db:"id"` // 训练作业资源规格id
TaskId int64 `db:"task_id"` // 任务id
AdapterId int64 `db:"adapter_id"` // 执行任务的适配器id
AdapterName string `db:"adapter_name"` // 适配器名称
ClusterId int64 `db:"cluster_id"` // 集群id
ClusterName string `db:"cluster_name"` // 集群名称
Name string `db:"name"` // 任务名
Replica int64 `db:"replica"` // 执行数
JobId string `db:"job_id"` // 集群返回任务id
StartTime string `db:"start_time"` // 开始时间
RunningTime string `db:"running_time"` // 运行时间
Result string `db:"result"` // 运行结果
DeletedAt string `db:"deleted_at"` // 删除时间
ImageId string `db:"image_id"` // 镜像id
Cmd string `db:"cmd"` // 命令行
FlavorId string `db:"flavor_id"` // 训练作业资源规格id
Status string `db:"status"` // 任务状态
}
)

func newTaskAiModelartsModel(conn sqlx.SqlConn) *defaultTaskAiModelartsModel {
return &defaultTaskAiModelartsModel{
conn: conn,
table: "`task_ai_modelarts`",
}
}

func (m *defaultTaskAiModelartsModel) Delete(ctx context.Context, id int64) error {
query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
_, err := m.conn.ExecCtx(ctx, query, id)
return err
}

func (m *defaultTaskAiModelartsModel) FindOne(ctx context.Context, id int64) (*TaskAiModelarts, error) {
query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", taskAiModelartsRows, m.table)
var resp TaskAiModelarts
err := m.conn.QueryRowCtx(ctx, &resp, query, id)
switch err {
case nil:
return &resp, nil
case sqlx.ErrNotFound:
return nil, ErrNotFound
default:
return nil, err
}
}

func (m *defaultTaskAiModelartsModel) Insert(ctx context.Context, data *TaskAiModelarts) (sql.Result, error) {
query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, taskAiModelartsRowsExpectAutoSet)
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)
return ret, err
}

func (m *defaultTaskAiModelartsModel) Update(ctx context.Context, data *TaskAiModelarts) error {
query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, taskAiModelartsRowsWithPlaceHolder)
_, 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)
return err
}

func (m *defaultTaskAiModelartsModel) tableName() string {
return m.table
}

Write Preview
Loading…
Cancel
Save