Browse Source

合并最新代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.7.1
zouap 3 years ago
parent
commit
7bb947cebd
3 changed files with 23 additions and 1 deletions
  1. +16
    -0
      models/ai_model_manage.go
  2. +1
    -1
      options/locale/locale_zh-CN.ini
  3. +6
    -0
      routers/repo/ai_model_convert.go

+ 16
- 0
models/ai_model_manage.go View File

@@ -397,6 +397,22 @@ func QueryModelConvertByRepoID(repoId int64) ([]*AiModelConvert, error) {
return aiModelManageConvert, nil
}

func QueryModelConvertByUserID(userID int64) ([]*AiModelConvert, error) {
sess := x.NewSession()
defer sess.Close()
var cond = builder.NewCond()
cond = cond.And(
builder.Eq{"ai_model_convert.user_id": userID},
)
sess.OrderBy("ai_model_convert.created_unix DESC")
aiModelManageConvert := make([]*AiModelConvert, 0)
if err := sess.Table(new(AiModelConvert)).Where(cond).
Find(&aiModelManageConvert); err != nil {
return nil, fmt.Errorf("Find: %v", err)
}
return aiModelManageConvert, nil
}

func QueryModelConvert(opts *AiModelQueryOptions) ([]*AiModelConvert, int64, error) {
sess := x.NewSession()
defer sess.Close()


+ 1
- 1
options/locale/locale_zh-CN.ini View File

@@ -1234,7 +1234,7 @@ model.list=模型列表
model.manage.create_new_convert_task=创建模型转换任务

modelconvert.manage.create_error1=相同的名称模型转换任务已经存在。
modelconvert.manage.create_error2=只能创建一个在运行的模型转换任务。
modelconvert.manage.create_error2=只能创建一个在运行的模型转换任务。
modelconvert.manage.model_not_exist=选择的模型不存在。
modelconvert.manage.no_operate_right=无操作权限。



+ 6
- 0
routers/repo/ai_model_convert.go View File

@@ -108,6 +108,12 @@ func SaveModelConvert(ctx *context.Context) {
})
return
}
}
}

convertList, err = models.QueryModelConvertByUserID(ctx.User.ID)
if err == nil {
for _, convert := range convertList {
if isRunningTask(convert.Status) {
log.Info("convert.Status=" + convert.Status + " convert.id=" + convert.ID)
ctx.JSON(200, map[string]string{


Loading…
Cancel
Save