iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mcs/pre/controller/admin/MmPredictItemController.java

@@ -5,8 +5,13 @@
import com.iailab.framework.common.pojo.PageResult;
import com.iailab.framework.common.util.object.BeanUtils;
import com.iailab.module.model.mcs.pre.dto.MmPredictItemDTO;
import com.iailab.module.model.mcs.pre.entity.MmItemOutputEntity;
import com.iailab.module.model.mcs.pre.entity.MmItemTypeEntity;
import com.iailab.module.model.mcs.pre.entity.MmPredictItemEntity;
import com.iailab.module.model.mcs.pre.service.MmItemOutputService;
import com.iailab.module.model.mcs.pre.service.MmItemTypeService;
import com.iailab.module.model.mcs.pre.service.MmPredictItemService;
import com.iailab.module.model.mcs.pre.service.MmResultTableService;
import com.iailab.module.model.mcs.pre.vo.CountItemtypeVO;
import com.iailab.module.model.mcs.pre.vo.MmPredictItemPageReqVO;
import com.iailab.module.model.mcs.pre.vo.MmPredictItemRespVO;
@@ -19,7 +24,6 @@
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import static com.iailab.framework.common.pojo.CommonResult.success;

@@ -34,6 +38,15 @@
    @Autowired
    private MmPredictItemService mmPredictItemService;

    @Autowired
    private MmItemTypeService mmItemTypeService;

    @Autowired
    private MmItemOutputService mmItemOutputService;

    @Autowired
    private MmResultTableService mmResultTableService;

    /**
     * 预测项列表
     */
@@ -41,13 +54,23 @@
    @PreAuthorize("@ss.hasPermission('model:pre-predict:query')")
    public CommonResult<PageResult<MmPredictItemRespVO>> page(@Validated MmPredictItemPageReqVO reqVO) {
        PageResult<MmPredictItemEntity> page = mmPredictItemService.queryPage(reqVO);
        return success(BeanUtils.toBean(page, MmPredictItemRespVO.class));
        PageResult<MmPredictItemRespVO> result = BeanUtils.toBean(page, MmPredictItemRespVO.class);
        for (MmPredictItemRespVO item : result.getList()){

            MmItemTypeEntity mmItemTypeEntity = mmItemTypeService.info(item.getItemtypeid());
            item.setItemtypename(mmItemTypeEntity== null ? "" :mmItemTypeEntity.getItemtypename());

            MmItemOutputEntity mmItemOutputEntity = mmItemOutputService.getByItemid(item.getId());
            item.setTagname(mmItemOutputEntity== null ? "" :mmItemOutputEntity.getTagname());
            item.setResulttableid(mmItemOutputEntity== null ? "" :mmItemOutputEntity.getResulttableid());
            item.setTablename(item == null ? "" : mmResultTableService.info(item.getResulttableid()).getTablename());
        }
        return success(result);
    }

    @GetMapping("/list")
    @PreAuthorize("@ss.hasPermission('model:pre-predict:query')")
    public CommonResult<List<MmPredictItemEntity>> list(@RequestParam Map<String, Object> params) {
        List<MmPredictItemEntity> list = mmPredictItemService.list(params);
    public CommonResult<List<MmPredictItemEntity>> list() {
        List<MmPredictItemEntity> list = mmPredictItemService.list();

        return success(list);
    }