大屏接口修改

This commit is contained in:
mashuai 2023-12-09 16:37:03 +08:00
parent d259b39b1f
commit 5d6579fb82
1 changed files with 8 additions and 1 deletions

View File

@ -3,9 +3,11 @@ package com.bonus.zlpt.bigscreen.service.impl;
import com.bonus.zlpt.bigscreen.domain.vo.CompanyInfoVo; import com.bonus.zlpt.bigscreen.domain.vo.CompanyInfoVo;
import com.bonus.zlpt.bigscreen.mapper.BmMachinistInfoMapper; import com.bonus.zlpt.bigscreen.mapper.BmMachinistInfoMapper;
import com.bonus.zlpt.bigscreen.service.BmMachinistInfoService; import com.bonus.zlpt.bigscreen.service.BmMachinistInfoService;
import com.bonus.zlpt.company.api.enums.CompanyLtdEnum;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@ -27,7 +29,12 @@ public class BmMachinistInfoServiceImpl implements BmMachinistInfoService {
*/ */
@Override @Override
public List<CompanyInfoVo> selectMachinistInfoList() { public List<CompanyInfoVo> selectMachinistInfoList() {
return infoMapper.selectMachinistInfoList(); List<CompanyInfoVo> list = infoMapper.selectMachinistInfoList();
ArrayList<CompanyInfoVo> companyInfoVos = new ArrayList<>();
for (CompanyInfoVo companyInfoVo : list) {
companyInfoVo.setOwnCo(CompanyLtdEnum.getNameByCode(Integer.parseInt(companyInfoVo.getOwnCo())));
}
return companyInfoVos;
} }
} }