package com.java110.job.cmd.export;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.java110.core.annotation.Java110Cmd;
|
import com.java110.core.context.ICmdDataFlowContext;
|
import com.java110.core.event.cmd.Cmd;
|
import com.java110.core.event.cmd.CmdEvent;
|
import com.java110.core.factory.GenerateCodeFactory;
|
import com.java110.dto.data.ExportDataDto;
|
import com.java110.dto.user.UserDto;
|
import com.java110.dto.user.UserDownloadFileDto;
|
import com.java110.intf.job.IUserDownloadFileV1InnerServiceSMO;
|
import com.java110.intf.user.IUserV1InnerServiceSMO;
|
import com.java110.job.export.ExportDataQueue;
|
import com.java110.po.user.UserDownloadFilePo;
|
import com.java110.utils.exception.CmdException;
|
import com.java110.utils.util.Assert;
|
import com.java110.utils.util.DateUtil;
|
import com.java110.utils.util.StringUtil;
|
import com.java110.vo.ResultVo;
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import java.util.Arrays;
|
import java.util.List;
|
|
/**
|
* 数据导出处理类
|
*/
|
@Java110Cmd(serviceCode = "export.exportData")
|
public class ExportDataCmd extends Cmd {
|
|
private static final String EXPORT_DATA_PRE = "temp/export/data/";
|
|
public static final String CODE_PREFIX_ID = "10";
|
|
@Autowired
|
private IUserDownloadFileV1InnerServiceSMO userDownloadFileV1InnerServiceSMOImpl;
|
|
|
@Autowired
|
private IUserV1InnerServiceSMO userV1InnerServiceSMOImpl;
|
|
@Override
|
public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
|
Assert.hasKeyAndValue(reqJson, "communityId", "请求中未包含小区");
|
Assert.hasKeyAndValue(reqJson, "pagePath", "请求中未包含页面");
|
}
|
|
@Override
|
public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
|
|
String userId = context.getReqHeaders().get("user-id");
|
String storeId = context.getReqHeaders().get("store-id");
|
|
UserDto userDto = new UserDto();
|
userDto.setUserId(userId);
|
List<UserDto> userDtos = userV1InnerServiceSMOImpl.queryUsers(userDto);
|
|
Assert.listOnlyOne(userDtos, "用户不存在");
|
|
//这里放入 员工ID 和 商户ID
|
|
reqJson.put("userId",userId);
|
reqJson.put("storeId",storeId);
|
if ("dataFeeManualCollection".equals(reqJson.getString("pagePath"))) {
|
// 物业费催缴单
|
List<String> configIds = Arrays.asList(reqJson.getString("configIds").split(","));
|
List<String> roomIds = Arrays.asList(reqJson.getString("roomIds").split(","));
|
for (String configId : configIds) {
|
reqJson.put("configId", configId);
|
for (String roomId : roomIds) {
|
reqJson.put("roomId", roomId);
|
this.create(userId, userDtos, reqJson);
|
}
|
}
|
} else {
|
this.create(userId, userDtos, reqJson);
|
}
|
context.setResponseEntity(ResultVo.createResponseEntity(ResultVo.CODE_OK,"文件正在生成，请到文件下载页面下载"));
|
}
|
|
private void create(String userId, List<UserDto> userDtos, JSONObject reqJson) {
|
ExportDataDto exportDataDto = new ExportDataDto();
|
exportDataDto.setBusinessAdapt(reqJson.getString("pagePath"));
|
exportDataDto.setReqJson(reqJson);
|
String fileName = "";
|
if ("dataFeeManualCollection".equals(reqJson.getString("pagePath"))) {
|
fileName = GenerateCodeFactory.getUUID() + ".docx";
|
} else {
|
fileName = GenerateCodeFactory.getUUID() + ".xlsx";
|
}
|
exportDataDto.setFileName(EXPORT_DATA_PRE
|
+ reqJson.getString("pagePath")
|
+ "/"
|
+ DateUtil.getNow(DateUtil.DATE_FORMATE_STRING_B)
|
+ "/"
|
+ fileName);
|
|
UserDownloadFilePo userDownloadFilePo = new UserDownloadFilePo();
|
userDownloadFilePo.setDownloadId(GenerateCodeFactory.getGeneratorId(CODE_PREFIX_ID));
|
userDownloadFilePo.setDownloadUserId(userId);
|
userDownloadFilePo.setDownloadUserName(userDtos.get(0).getName());
|
userDownloadFilePo.setFileType(reqJson.getString("pagePath"));
|
userDownloadFilePo.setCommunityId(reqJson.getString("communityId"));
|
userDownloadFilePo.setName(fileName);
|
userDownloadFilePo.setState(UserDownloadFileDto.STATE_WAIT);
|
int flag = userDownloadFileV1InnerServiceSMOImpl.saveUserDownloadFile(userDownloadFilePo);
|
|
if (flag < 1) {
|
throw new CmdException("下载文件失败");
|
}
|
exportDataDto.setDownloadId(userDownloadFilePo.getDownloadId());
|
ExportDataQueue.addMsg(exportDataDto);
|
}
|
}
|
