| @@ -397,6 +397,22 @@ func QueryModelConvertByRepoID(repoId int64) ([]*AiModelConvert, error) { | |||||
| return aiModelManageConvert, nil | 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) { | func QueryModelConvert(opts *AiModelQueryOptions) ([]*AiModelConvert, int64, error) { | ||||
| sess := x.NewSession() | sess := x.NewSession() | ||||
| defer sess.Close() | defer sess.Close() | ||||
| @@ -1234,7 +1234,7 @@ model.list=模型列表 | |||||
| model.manage.create_new_convert_task=创建模型转换任务 | model.manage.create_new_convert_task=创建模型转换任务 | ||||
| modelconvert.manage.create_error1=相同的名称模型转换任务已经存在。 | modelconvert.manage.create_error1=相同的名称模型转换任务已经存在。 | ||||
| modelconvert.manage.create_error2=只能创建一个在运行的模型转换任务。 | |||||
| modelconvert.manage.create_error2=只能创建一个正在运行的模型转换任务。 | |||||
| modelconvert.manage.model_not_exist=选择的模型不存在。 | modelconvert.manage.model_not_exist=选择的模型不存在。 | ||||
| modelconvert.manage.no_operate_right=无操作权限。 | modelconvert.manage.no_operate_right=无操作权限。 | ||||
| @@ -108,6 +108,12 @@ func SaveModelConvert(ctx *context.Context) { | |||||
| }) | }) | ||||
| return | return | ||||
| } | } | ||||
| } | |||||
| } | |||||
| convertList, err = models.QueryModelConvertByUserID(ctx.User.ID) | |||||
| if err == nil { | |||||
| for _, convert := range convertList { | |||||
| if isRunningTask(convert.Status) { | if isRunningTask(convert.Status) { | ||||
| log.Info("convert.Status=" + convert.Status + " convert.id=" + convert.ID) | log.Info("convert.Status=" + convert.Status + " convert.id=" + convert.ID) | ||||
| ctx.JSON(200, map[string]string{ | ctx.JSON(200, map[string]string{ | ||||