package com.doumee.service.business.impl;
|
|
import com.doumee.core.conditoner.ConditionerUtil;
|
import com.doumee.core.conditoner.model.request.CompanyGsManageRequest;
|
import com.doumee.core.conditoner.model.response.CompanyGsInfoResponse;
|
import com.doumee.core.conditoner.model.response.ConditionerBaseResponse;
|
import com.doumee.dao.business.dto.YwConditionerReportQueryDTO;
|
import com.doumee.dao.business.dto.YwConditionerUsageReportPageDTO;
|
import com.doumee.dao.business.dto.YwConditionerUsageReportVO;
|
import com.doumee.service.business.ConditionerBizService;
|
import com.doumee.service.business.YwConditionerReportService;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.OutputStreamWriter;
|
import java.io.PrintWriter;
|
import java.net.URLEncoder;
|
import java.nio.charset.StandardCharsets;
|
import java.util.Collections;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class YwConditionerReportServiceImpl implements YwConditionerReportService {
|
|
@Autowired
|
private ConditionerBizService conditionerBizService;
|
|
@Override
|
public YwConditionerUsageReportPageDTO findPage(YwConditionerReportQueryDTO query) {
|
return conditionerBizService.queryUsageReport(query);
|
}
|
|
@Override
|
public String syncUsage(YwConditionerReportQueryDTO query) {
|
return conditionerBizService.syncUsage(query);
|
}
|
|
@Override
|
public List<Map<String, Object>> merchantOptions() {
|
conditionerBizService.ensureLogin();
|
CompanyGsManageRequest req = new CompanyGsManageRequest();
|
ConditionerBaseResponse<List<CompanyGsInfoResponse>> resp = ConditionerUtil.getGs(req);
|
if (resp == null || resp.getData() == null) {
|
return Collections.emptyList();
|
}
|
return resp.getData().stream().map(gs -> {
|
Map<String, Object> m = new LinkedHashMap<>();
|
m.put("gsId", gs.getId());
|
m.put("gsName", gs.getGs_name());
|
return m;
|
}).collect(Collectors.toList());
|
}
|
|
@Override
|
public void exportExcel(YwConditionerReportQueryDTO query, HttpServletResponse response) {
|
if (query == null) {
|
query = new YwConditionerReportQueryDTO();
|
}
|
query.setPage(1);
|
query.setCapacity(1_000_000);
|
YwConditionerUsageReportPageDTO page = findPage(query);
|
try {
|
String fileName = URLEncoder.encode("空调用量报表.csv", StandardCharsets.UTF_8.name());
|
response.setContentType("text/csv;charset=UTF-8");
|
response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
|
PrintWriter writer = new PrintWriter(new OutputStreamWriter(response.getOutputStream(), StandardCharsets.UTF_8));
|
writer.write('\ufeff');
|
List<String> cols = page.getDateColumns() != null ? page.getDateColumns() : Collections.emptyList();
|
StringBuilder header = new StringBuilder("设备ID,设备名称,总时长(h),总电量(kWh),总电费(元)");
|
for (String col : cols) {
|
header.append(',').append(col).append("电量(kWh),")
|
.append(col).append("时长(h),")
|
.append(col).append("电费(元)");
|
}
|
writer.println(header);
|
if (page.getRecords() != null) {
|
for (YwConditionerUsageReportVO row : page.getRecords()) {
|
StringBuilder line = new StringBuilder();
|
line.append(csv(row.getDevId())).append(',')
|
.append(csv(formatDevDisplayName(row))).append(',')
|
.append(csv(row.getTotalTime())).append(',')
|
.append(csv(row.getTotalDl())).append(',')
|
.append(csv(row.getTotalDf()));
|
for (String col : cols) {
|
String dateKey = colToDateKey(query, col);
|
YwConditionerUsageReportVO.YwConditionerUsageDailyVO day =
|
row.getDaily() != null && dateKey != null ? row.getDaily().get(dateKey) : null;
|
line.append(',').append(csv(day != null ? day.getDl() : ""))
|
.append(',').append(csv(day != null ? day.getTime() : ""))
|
.append(',').append(csv(day != null ? day.getDf() : ""));
|
}
|
writer.println(line);
|
}
|
}
|
writer.flush();
|
} catch (Exception e) {
|
throw new RuntimeException("导出失败", e);
|
}
|
}
|
|
private static String csv(Object val) {
|
if (val == null) {
|
return "";
|
}
|
String s = String.valueOf(val).replace("\"", "\"\"");
|
if (s.contains(",") || s.contains("\"") || s.contains("\n")) {
|
return "\"" + s + "\"";
|
}
|
return s;
|
}
|
|
private static String formatDevDisplayName(YwConditionerUsageReportVO row) {
|
if (row == null) {
|
return "-";
|
}
|
StringBuilder sb = new StringBuilder();
|
appendDisplayPart(sb, row.getFloorName());
|
appendDisplayPart(sb, row.getRoomName());
|
appendDisplayPart(sb, row.getDevName());
|
return sb.length() == 0 ? "-" : sb.toString();
|
}
|
|
private static void appendDisplayPart(StringBuilder sb, String part) {
|
if (StringUtils.isBlank(part)) {
|
return;
|
}
|
if (sb.length() > 0) {
|
sb.append('/');
|
}
|
sb.append(part.trim());
|
}
|
|
private String colToDateKey(YwConditionerReportQueryDTO query, String col) {
|
if (query != null && "day".equalsIgnoreCase(query.getReportType())) {
|
return col;
|
}
|
String month = query != null ? query.getMonth() : null;
|
if (month == null || col == null || !col.contains(".")) {
|
return col;
|
}
|
String[] parts = col.split("\\.");
|
if (parts.length != 2) {
|
return col;
|
}
|
String year = month.split("-")[0];
|
int m = Integer.parseInt(parts[0]);
|
int d = Integer.parseInt(parts[1]);
|
return String.format("%s-%02d-%02d", year, m, d);
|
}
|
}
|
