package com.iailab.module.mcs.controller.admin;
|
|
import com.iailab.common.annotation.LogOperation;
|
import com.iailab.common.enums.IsEnableEnum;
|
import com.iailab.framework.common.page.PageData;
|
import com.iailab.common.utils.Constant;
|
import com.iailab.framework.common.util.object.ConvertUtils;
|
import com.iailab.framework.common.pojo.CommonResult;
|
import com.iailab.framework.common.util.validation.ValidationUtils;
|
import com.iailab.framework.common.validation.group.AddGroup;
|
import com.iailab.framework.common.validation.group.DefaultGroup;
|
import com.iailab.framework.common.validation.group.UpdateGroup;
|
import com.iailab.framework.security.core.util.SecurityFrameworkUtils;
|
import com.iailab.module.mcs.dto.StModelDTO;
|
import com.iailab.module.mcs.dto.StModelOutDTO;
|
import com.iailab.module.mcs.dto.StModelParamDTO;
|
import com.iailab.module.mcs.dto.StModelSettingDTO;
|
import com.iailab.module.mcs.entity.StModelOutEntity;
|
import com.iailab.module.mcs.entity.StModelParamEntity;
|
import com.iailab.module.mcs.entity.StModelSettingEntity;
|
import com.iailab.module.mcs.service.StModelOutService;
|
import com.iailab.module.mcs.service.StModelParamService;
|
import com.iailab.module.mcs.service.StModelService;
|
import com.iailab.module.mcs.service.StModelSettingService;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.Parameter;
|
import io.swagger.v3.oas.annotations.Parameters;
|
import javax.annotation.Resource;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.IOException;
|
import java.util.*;
|
|
import static com.iailab.framework.common.pojo.CommonResult.success;
|
|
|
/**
|
* @author lirm 1343021927@qq.com
|
* @since 1.0.0 2023-05-10
|
*/
|
@RestController
|
@RequestMapping("/model/mcs/st-model")
|
public class StModelController {
|
@Resource
|
private StModelService stModelService;
|
|
@Resource
|
private StModelParamService stModelParamService;
|
|
@Resource
|
private StModelSettingService stModelSettingService;
|
|
@Resource
|
private StModelOutService stModelOutService;
|
|
@GetMapping("page")
|
@Operation(summary = "分页")
|
@Parameters({
|
@Parameter(name = Constant.PAGE, description = "当前页码，从1开始", required = true),
|
@Parameter(name = Constant.LIMIT, description = "每页显示记录数", required = true),
|
@Parameter(name = Constant.ORDER_FIELD, description = "排序字段"),
|
@Parameter(name = Constant.ORDER, description = "排序方式，可选值(asc、desc)")
|
})
|
public CommonResult<PageData<StModelDTO>> page(@RequestParam Map<String, Object> params) {
|
PageData<StModelDTO> page = stModelService.page(params);
|
|
return success(page);
|
}
|
|
@GetMapping("{id}")
|
@Operation(summary = "信息")
|
public CommonResult<StModelDTO> get(@PathVariable("id") String id) {
|
StModelDTO data = stModelService.get(id);
|
|
List<StModelParamDTO> tParamPage = stModelParamService.getByModelId(id);
|
data.setModelParamList(tParamPage);
|
|
Map<String, Object> params = new HashMap<String, Object>(1);
|
params.put("modelId", data.getId());
|
List<StModelSettingDTO> tSettingPage = stModelSettingService.getAll(params);
|
data.setModelSettingList(tSettingPage);
|
|
Map<String, Object> out = new HashMap<String, Object>(1);
|
out.put("modelId", data.getId());
|
List<StModelOutDTO> tOutPage = stModelOutService.getAll(out);
|
data.setModelOutList(tOutPage);
|
|
return new CommonResult<StModelDTO>().setData(data);
|
}
|
|
@PostMapping
|
@Operation(summary = "保存")
|
@LogOperation("保存")
|
@Transactional
|
public CommonResult save(@RequestBody StModelDTO dto) {
|
//效验数据
|
ValidationUtils.validate(dto, AddGroup.class, DefaultGroup.class);
|
|
dto.setId(UUID.randomUUID().toString());
|
List<StModelParamDTO> tParamList = dto.getModelParamList();
|
if (tParamList != null) {
|
for (int i = 0; i < tParamList.size(); i++) {
|
StModelParamDTO tParamDTO = new StModelParamDTO();
|
tParamDTO = tParamList.get(i);
|
StModelParamEntity tParamEntity = new StModelParamEntity();
|
tParamEntity.setModelId(dto.getId());
|
tParamEntity.setParamName(tParamDTO.getParamName());
|
tParamEntity.setParamId(tParamDTO.getParamId());
|
tParamEntity.setParamOrder(tParamDTO.getParamOrder());
|
tParamEntity.setParamPortOrder(tParamDTO.getParamPortOrder());
|
tParamEntity.setDataLength(tParamDTO.getDataLength());
|
tParamEntity.setParamType(tParamDTO.getParamType());
|
stModelParamService.insert(tParamEntity);
|
}
|
}
|
List<StModelSettingDTO> tSettingList = dto.getModelSettingList();
|
if (tSettingList != null) {
|
for (int i = 0; i < tSettingList.size(); i++) {
|
StModelSettingEntity tSettingEntity = ConvertUtils.sourceToTarget(tSettingList.get(i), StModelSettingEntity.class);
|
tSettingEntity.setModelId(dto.getId());
|
stModelSettingService.insert(tSettingEntity);
|
}
|
}
|
|
List<StModelOutDTO> tOutList = dto.getModelOutList();
|
if (tOutList != null) {
|
for (int i = 0; i < tOutList.size(); i++) {
|
StModelOutEntity tOutEntity = ConvertUtils.sourceToTarget(tOutList.get(i), StModelOutEntity.class);
|
tOutEntity.setModelId(dto.getId());
|
tOutEntity.setSort(i+1);
|
stModelOutService.insert(tOutEntity);
|
}
|
}
|
dto.setCreator(SecurityFrameworkUtils.getLoginUserId());
|
dto.setCreateDate(new Date());
|
dto.setUpdater(SecurityFrameworkUtils.getLoginUserId());
|
dto.setUpdateDate(new Date());
|
dto.setIsEnable(IsEnableEnum.ENABLE.value());
|
stModelService.save(dto);
|
return new CommonResult();
|
}
|
|
@PutMapping
|
@Operation(summary = "修改")
|
@LogOperation("修改")
|
@Transactional
|
public CommonResult update(@RequestBody StModelDTO dto) {
|
//效验数据
|
ValidationUtils.validate(dto, UpdateGroup.class, DefaultGroup.class);
|
|
List<StModelParamDTO> tParamList = dto.getModelParamList();
|
if (tParamList != null) {
|
stModelParamService.deleteByModelId(dto.getId());
|
for (int i = 0; i < tParamList.size(); i++) {
|
StModelParamDTO tParamDTO = new StModelParamDTO();
|
tParamDTO = tParamList.get(i);
|
StModelParamEntity tParamEntity = new StModelParamEntity();
|
tParamEntity.setModelId(dto.getId());
|
tParamEntity.setParamName(tParamDTO.getParamName());
|
tParamEntity.setParamId(tParamDTO.getParamId());
|
tParamEntity.setParamOrder(tParamDTO.getParamOrder());
|
tParamEntity.setParamPortOrder(tParamDTO.getParamPortOrder());
|
tParamEntity.setDataLength(tParamDTO.getDataLength());
|
tParamEntity.setParamType(tParamDTO.getParamType());
|
|
stModelParamService.insert(tParamEntity);
|
}
|
}
|
List<StModelSettingDTO> tSettingList = dto.getModelSettingList();
|
if (tSettingList != null) {
|
stModelSettingService.deleteByModelId(dto.getId());
|
for (int i = 0; i < tSettingList.size(); i++) {
|
StModelSettingEntity tSettingEntity = ConvertUtils.sourceToTarget(tSettingList.get(i), StModelSettingEntity.class);
|
tSettingEntity.setModelId(dto.getId());
|
stModelSettingService.insert(tSettingEntity);
|
}
|
}
|
List<StModelOutDTO> tOutList = dto.getModelOutList();
|
if (tOutList != null) {
|
stModelOutService.deleteByModelId(dto.getId());
|
for (int i = 0; i < tOutList.size(); i++) {
|
StModelOutEntity tOutEntity = ConvertUtils.sourceToTarget(tOutList.get(i), StModelOutEntity.class);
|
tOutEntity.setModelId(dto.getId());
|
tOutEntity.setSort(i+1);
|
stModelOutService.insert(tOutEntity);
|
}
|
}
|
|
Long id = SecurityFrameworkUtils.getLoginUserId();
|
dto.setUpdater(id);
|
dto.setUpdateDate(new Date());
|
stModelService.update(dto);
|
|
return new CommonResult<Object>();
|
}
|
|
@DeleteMapping
|
@Operation(summary = "删除")
|
@LogOperation("删除")
|
@Transactional
|
public CommonResult delete(@RequestBody String[] ids) {
|
List<String> idsList = Arrays.asList(ids);
|
for (int i = 0; i < idsList.size(); i++) {
|
stModelParamService.deleteByModelId(idsList.get(i));//删除模型参数表数据
|
stModelSettingService.deleteByModelId(idsList.get(i));//删除模型设置表数据
|
}
|
|
stModelService.deleteByIds(ids);//删除模型表数据
|
|
return new CommonResult();
|
}
|
|
@PutMapping("/enable")
|
@Operation(summary = "启用")
|
@LogOperation("启用")
|
@Transactional
|
public CommonResult enable(@RequestBody String[] ids) {
|
stModelService.enableByIds(ids);
|
return new CommonResult();
|
}
|
|
@PutMapping("/disable")
|
@Operation(summary = "禁用")
|
@LogOperation("禁用")
|
@Transactional
|
public CommonResult disable(@RequestBody String[] ids) {
|
stModelService.disableByIds(ids);
|
return new CommonResult();
|
}
|
|
|
/**
|
* 上传模型
|
*/
|
@Operation(summary = "上传模型")
|
@PostMapping("/upload")
|
public CommonResult<Map<String, Object>> upload(@RequestParam("file") MultipartFile file) throws IOException {
|
Map<String, Object> data = stModelService.upload(file);
|
return new CommonResult<Map<String, Object>>().setData(data);
|
}
|
|
@Operation(summary = "上传模型")
|
@GetMapping("/relation")
|
public CommonResult getRelation() {
|
Map<String, Object> data = stModelService.getRelation();
|
return new CommonResult().setData(data);
|
}
|
}
|
