|
@@ -11,10 +11,7 @@ import com.bizmatics.common.mvc.base.AbstractCrudService;
|
|
|
import com.bizmatics.model.*;
|
|
|
import com.bizmatics.model.system.SysUser;
|
|
|
import com.bizmatics.model.vo.PlatformAreaVo;
|
|
|
-import com.bizmatics.persistence.mapper.AlarmPowerMapper;
|
|
|
-import com.bizmatics.persistence.mapper.DeviceMapper;
|
|
|
-import com.bizmatics.persistence.mapper.PlatformAreaMapper;
|
|
|
-import com.bizmatics.persistence.mapper.SiteMapper;
|
|
|
+import com.bizmatics.persistence.mapper.*;
|
|
|
import com.bizmatics.persistence.mapper.system.SysUserMapper;
|
|
|
import com.bizmatics.service.*;
|
|
|
import com.bizmatics.service.enums.DeviceStatusCode;
|
|
@@ -64,6 +61,8 @@ public class SiteServiceImpl extends AbstractCrudService<SiteMapper, Site> imple
|
|
|
private SysUserMapper userMapper;
|
|
|
@Autowired
|
|
|
private HtAnalogDataService htAnalogDataService;
|
|
|
+ @Autowired
|
|
|
+ private ElectricityRateConfigMapper electricityRateConfigMapper;
|
|
|
|
|
|
@Override
|
|
|
public DeviceCountVO selectCount() {
|
|
@@ -81,15 +80,28 @@ public class SiteServiceImpl extends AbstractCrudService<SiteMapper, Site> imple
|
|
|
public List<Site> list(String name) {
|
|
|
Integer userId = SecurityUtils.getLoginUser().getUser().getUserId().intValue();
|
|
|
List<SysUser> tenantDaya = userMapper.getTenantId(SecurityUtils.getLoginUser().getUser().getUserId());
|
|
|
- if (tenantDaya.size() < 0) {
|
|
|
+ if (tenantDaya.isEmpty()) {
|
|
|
throw new BusinessException("无此租户,请联系管理员");
|
|
|
}
|
|
|
List<Site> siteList = new ArrayList<>();
|
|
|
- if (tenantDaya.get(0).getUserType().equals("01")) {
|
|
|
+ if ("01".equals(tenantDaya.get(0).getUserType())) {
|
|
|
siteList = baseMapper.listOne(name, tenantDaya.get(0).getTenantId());
|
|
|
- } else if (tenantDaya.get(0).getUserType().equals("00")) {
|
|
|
+ } else if ("00".equals(tenantDaya.get(0).getUserType())) {
|
|
|
siteList = baseMapper.list(userId, name, tenantDaya.get(0).getTenantId());
|
|
|
}
|
|
|
+
|
|
|
+ // 查询是否存在费用
|
|
|
+ List<Integer> siteIds = siteList.stream()
|
|
|
+ .map(Site::getId)
|
|
|
+ .collect(Collectors.toList());
|
|
|
+
|
|
|
+ Set<Integer> exists = electricityRateConfigMapper.selectBatchIds(siteIds)
|
|
|
+ .stream()
|
|
|
+ .map(ElectricityRateConfig::getSiteId)
|
|
|
+ .collect(Collectors.toSet());
|
|
|
+
|
|
|
+ siteList.forEach(site -> site.setCostTypeStatus(exists.contains(site.getId())));
|
|
|
+
|
|
|
return siteList;
|
|
|
}
|
|
|
|