CY
/
mcwl-ai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(resource): 增加模型类型字段

master
yang 2025-02-05 15:20:45 +08:00
parent bb64605dc4
commit 0d9a485c19
3 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mcwl-resource/src/main/java/com/mcwl/resource/domain/ModelProduct.java
View File

@ -47,10 +47,10 @@ public class ModelProduct extends BaseEntity {
@ApiModelProperty(value = "用户id") @ApiModelProperty(value = "用户id")
private Long userId; private Long userId;
/** /**
* 模型类型ID * 模型类型
*/ */
@ApiModelProperty(value = "模型类型ID") @ApiModelProperty(value = "模型类型")
private Long modelTypeId; private Long modelType;
/** /**
* 垂类 * 垂类
*/ */

9
mcwl-resource/src/main/java/com/mcwl/resource/domain/vo/ModelVo.java
View File

@ -10,13 +10,18 @@ public class ModelVo {
/** /**
* 模型id * 模型id
*/ */
@ApiModelProperty(value = "模型id", required = true) @ApiModelProperty(value = "模型id")
private Long id; private Long id;
/** /**
* 模型名称 * 模型名称
*/ */
@ApiModelProperty(value = "模型名称", required = true) @ApiModelProperty(value = "模型名称")
private String modelName; private String modelName;
/**
* 模型类型
*/
@ApiModelProperty(value = "模型类型")
private String modelType;
/** /**
* 封面图 * 封面图
*/ */

7
mcwl-resource/src/main/java/com/mcwl/resource/service/impl/ModelServiceImpl.java
View File

@ -155,6 +155,13 @@ public class ModelServiceImpl extends ServiceImpl<ModelMapper,ModelProduct> impl
for (ModelProduct modelImage : modelImageList) { for (ModelProduct modelImage : modelImageList) {
ModelVo modelVo = new ModelVo(); ModelVo modelVo = new ModelVo();
BeanUtil.copyProperties(modelImage, modelVo); BeanUtil.copyProperties(modelImage, modelVo);
// 设置模型类型
if (Objects.nonNull(modelImage.getModelType())) {
String modelTypeName = DictInit.getDictValue(DictConstants.MODEL_CATEGORY, modelImage.getModelType().toString());
modelVo.setModelType(modelTypeName);
}
// 获取用户信息 // 获取用户信息
SysUser sysUser = sysUserService.selectUserById(modelImage.getUserId()); SysUser sysUser = sysUserService.selectUserById(modelImage.getUserId());
if (Objects.isNull(sysUser)) { if (Objects.isNull(sysUser)) {