Commit 6ada879d authored by kongfm's avatar kongfm

更新tzs导出相关代码

parent db2e3734
......@@ -4,6 +4,7 @@ import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
......@@ -13,6 +14,7 @@ import java.util.Map;
import javax.servlet.http.HttpServletResponse;
import com.google.common.collect.Lists;
import org.apache.poi.ss.usermodel.BorderStyle;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.IndexedColors;
......@@ -382,4 +384,70 @@ public class ExcelUtil {
return new HorizontalCellStyleStrategy(headWriteCellStyle, contentWriteCellStyle);
}
/**
* 生成excel模板
*
* @param response
* @param fileName 下载的文件名,
* @param sheetName sheet名
* @param data 导出的数据
* @param model 导出的头
* @param flag true模板填充下拉 false 不填充
*/
public static void createTemplateWithHeaders(HttpServletResponse response, String fileName, String sheetName,
List<? extends Object> data, Class<?> model, DataSources dataDictionaryMapper, boolean flag, List<List<String>> heads, List<String> headstr, String fileType) {
HorizontalCellStyleStrategy horizontalCellStyleStrategy = setMyCellStyle();
try {
// 下拉列表集合
Map<Integer, String[]> explicitListConstraintMap = new HashMap<>();
if (flag) {
// 循环获取对应列得下拉列表信息
Field[] declaredFields = model.getDeclaredFields();
for (int i = 0; i < declaredFields.length; i++) {
Field field = declaredFields[i];
// 解析注解信息
ExplicitConstraint explicitConstraint = field.getAnnotation(ExplicitConstraint.class);
resolveExplicitConstraint(explicitListConstraintMap, explicitConstraint, dataDictionaryMapper);
}
}
List<List<Object>> listData = Lists.newArrayList();
for (Object t : data) {
List<Object> rowLine = new ArrayList<>();
for(String head: headstr) {
List<Object> row = new ArrayList<>();
String getMethodName = "get" + head.substring(0,1).toUpperCase() + head.substring(1);
Class clazz = t.getClass();
Method getMethod;
getMethod = clazz.getMethod(getMethodName, new Class[]{} );
Object value = getMethod.invoke(t, new Object[]{});
rowLine.add(value);
}
listData.add(rowLine);
}
ExcelTypeEnum typeEnum = null;
if("1039".equals(fileType)) {
typeEnum = ExcelTypeEnum.XLS;
} else if("1040".equals(fileType)) {
typeEnum = ExcelTypeEnum.XLSX;
}
EasyExcel.write(getOutputStream(fileName, response, typeEnum))
.excelType(typeEnum).sheet(sheetName)
.registerWriteHandler(new TemplateCellWriteHandlerDate(explicitListConstraintMap))
.registerWriteHandler(new TemplateCellWriteHandler())
.registerWriteHandler(horizontalCellStyleStrategy)
.head(heads).doWrite(listData);
} catch (Exception e) {
e.printStackTrace();
throw new RuntimeException("系统异常!");
}
}
}
package com.yeejoin.amos.boot.module.tzs.api.dto;
import com.alibaba.excel.annotation.ExcelIgnore;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
/**
* 导出dto类
*/
@Data
@ApiModel(value="ExportDto", description="导出Dto类")
public class ExportDto {
@ExcelIgnore
private static final long serialVersionUID = 1L;
@ApiModelProperty(value = "文件名称")
private String fileName;
@ApiModelProperty(value = "文件类型")
private String fileType;
@ApiModelProperty(value = "导出数据类型 0 全部数据 1 选中数据")
private String dataType;
@ApiModelProperty(value = "导出数据id")
private String exportId;
@ApiModelProperty(value = "导出字段")
private String exportArray;
@ApiModelProperty(value = "导出类型")
private String exportType;
}
package com.yeejoin.amos.boot.module.tzs.api.dto;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import com.yeejoin.amos.boot.biz.common.dto.BaseDto;
import lombok.Data;
import lombok.EqualsAndHashCode;
import java.util.Date;
/**
* 模板表
*
* @author system_generator
* @date 2021-10-25
*/
@Data
@EqualsAndHashCode(callSuper = true)
@ApiModel(value="TemplateExportDto", description="模板表")
public class TemplateExportDto extends BaseDto {
private static final long serialVersionUID = 1L;
@ApiModelProperty(value = "模板名称")
private String templateName;
@ApiModelProperty(value = "模板字段")
private String templateContent;
@ApiModelProperty(value = "模板类型")
private String templateType;
@ApiModelProperty(value = "是否标准模板")
private Boolean isStandard;
}
......@@ -20,8 +20,11 @@ import java.util.List;
@TableName("tcb_use_unit")
@ApiModel(value = "UseUnitDto", description = "UseUnitDto")
public class UseUnitDto extends BaseDto {
private static final long serialVersionUID = 1L;
@ApiModelProperty(value = "统一社会信用代码")
private String socialCreditCode;
......@@ -73,6 +76,12 @@ public class UseUnitDto extends BaseDto {
@ApiModelProperty(value = "原始表id(来自历史数据库)")
private String originalId;
@ApiModelProperty(value = "经度")
private String longitude;
@ApiModelProperty(value = "纬度")
private String latitude;
@ApiModelProperty(value = "人员信息")
List<DutyPersonDto> dutyPersonList;
}
package com.yeejoin.amos.boot.module.tzs.api.entity;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.yeejoin.amos.boot.biz.common.entity.BaseEntity;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;
import java.util.Date;
/**
* 模板表
*
* @author system_generator
* @date 2021-10-25
*/
@Data
@EqualsAndHashCode(callSuper = true)
@Accessors(chain = true)
@TableName("tz_template_export")
public class TemplateExport extends BaseEntity {
private static final long serialVersionUID = 1L;
/**
* 模板名称
*/
@TableField("template_name")
private String templateName;
/**
* 模板字段
*/
@TableField("template_content")
private String templateContent;
/**
* 模板类型
*/
@TableField("template_type")
private String templateType;
/**
* 是否标准模板
*/
@TableField("is_standard")
private Boolean isStandard;
}
package com.yeejoin.amos.boot.module.tzs.api.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorWlInfoDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.Elevator;
import org.apache.ibatis.annotations.Param;
import java.util.List;
......@@ -27,4 +29,6 @@ public interface ElevatorMapper extends BaseMapper<Elevator> {
);
int queryElevatorListCount();
List<ElevatorDto> selectExportData(@Param("ids") List<String> ids);
}
package com.yeejoin.amos.boot.module.tzs.api.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.yeejoin.amos.boot.module.tzs.api.dto.MaintenanceUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.MaintenanceUnit;
import org.apache.ibatis.annotations.Param;
import java.util.List;
/**
* Mapper 接口
......@@ -11,4 +15,5 @@ import com.yeejoin.amos.boot.module.tzs.api.entity.MaintenanceUnit;
*/
public interface MaintenanceUnitMapper extends BaseMapper<MaintenanceUnit> {
List<MaintenanceUnitDto> selectExportData(@Param("ids") List<String> ids);
}
package com.yeejoin.amos.boot.module.tzs.api.mapper;
import com.yeejoin.amos.boot.module.tzs.api.entity.TemplateExport;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
/**
* 模板表 Mapper 接口
*
* @author system_generator
* @date 2021-10-25
*/
public interface TemplateExportMapper extends BaseMapper<TemplateExport> {
}
package com.yeejoin.amos.boot.module.tzs.api.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.yeejoin.amos.boot.module.tzs.api.dto.UseUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.UseUnit;
import org.apache.ibatis.annotations.Param;
import java.util.List;
/**
* Mapper 接口
......@@ -11,4 +15,5 @@ import com.yeejoin.amos.boot.module.tzs.api.entity.UseUnit;
*/
public interface UseUnitMapper extends BaseMapper<UseUnit> {
List<UseUnitDto> selectExportData(@Param("ids") List<String> ids);
}
package com.yeejoin.amos.boot.module.tzs.api.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.Elevator;
import java.util.List;
/**
* 服务类
*
......@@ -23,4 +26,6 @@ public interface IElevatorService extends IService<Elevator> {
* @return
*/
String saveElevatorQrCode(Long elevatorId);
List<ElevatorDto> selectExportData(String exportId);
}
package com.yeejoin.amos.boot.module.tzs.api.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.yeejoin.amos.boot.module.tzs.api.dto.MaintenanceUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.MaintenanceUnit;
import java.util.List;
/**
* 服务类
*
......@@ -11,4 +14,5 @@ import com.yeejoin.amos.boot.module.tzs.api.entity.MaintenanceUnit;
*/
public interface IMaintenanceUnitService extends IService<MaintenanceUnit> {
List<MaintenanceUnitDto> selectExportData(String exportId);
}
package com.yeejoin.amos.boot.module.tzs.api.service;
/**
* 模板表接口类
*
* @author system_generator
* @date 2021-10-25
*/
public interface ITemplateExportService {
}
package com.yeejoin.amos.boot.module.tzs.api.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.yeejoin.amos.boot.module.tzs.api.dto.UseUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.UseUnit;
import java.util.List;
/**
* 服务类
*
......@@ -11,4 +14,5 @@ import com.yeejoin.amos.boot.module.tzs.api.entity.UseUnit;
*/
public interface IUseUnitService extends IService<UseUnit> {
List<UseUnitDto> selectExportData(String exportId);
}
......@@ -82,4 +82,16 @@
) te
</select>
<select id="selectExportData" resultType="com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorDto">
select t.* , d.name as useSiteCategory
from tcb_elevator t left join cb_data_dictionary d on d.code = t.use_site_category where t.is_delete = 0
<if test="ids != null and ids.size() > 0">
and t.sequence_nbr in
<foreach item="item" collection="ids" separator="," open="(" close=")" index=""> #{item}
</foreach>
</if>
</select>
</mapper>
......@@ -2,4 +2,15 @@
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.yeejoin.amos.boot.module.tzs.api.mapper.MaintenanceUnitMapper">
<select id="selectExportData" resultType="com.yeejoin.amos.boot.module.tzs.api.dto.MaintenanceUnitDto">
select u.*
from cb_maintenance_unit u where u.is_delete = 0
<if test="ids != null and ids.size() > 0">
and u.sequence_nbr in
<foreach item="item" collection="ids" separator="," open="(" close=")" index=""> #{item}
</foreach>
</if>
</select>
</mapper>
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.yeejoin.amos.boot.module.tzs.api.mapper.TemplateExportMapper">
</mapper>
......@@ -2,4 +2,32 @@
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.yeejoin.amos.boot.module.tzs.api.mapper.UseUnitMapper">
<select id="selectExportData" resultType="com.yeejoin.amos.boot.module.tzs.api.dto.UseUnitDto">
select
u.social_credit_code as socialCreditCode,
u.use_unit_name as useUnitName,
u.real_estate_brand as realEstateBrand,
u.property_company_brand as propertyCompanyBrand,
u.province as province,
u.city as city,
u.district as district,
u.region_code as regionCode,
u.address as address,
u.principal as principal,
u.longitude as longitude,
u.latitude as latitude,
u.principal_phone as principalPhone,
u.management as management,
u.manager as manager,
u.manager_phone as managerPhone
from tcb_use_unit u where u.is_delete = 0
<if test="ids != null and ids.size() > 0">
and u.sequence_nbr in
<foreach item="item" collection="ids" separator="," open="(" close=")" index=""> #{item}
</foreach>
</if>
</select>
</mapper>
......@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.SystemClock;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.google.common.collect.Lists;
import com.yeejoin.amos.boot.biz.common.bo.ReginParams;
import com.yeejoin.amos.boot.biz.common.controller.BaseController;
import com.yeejoin.amos.boot.biz.common.utils.DateUtils;
......@@ -24,16 +25,22 @@ import com.yeejoin.amos.boot.module.tzs.api.dto.AlertPaperInfoDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.DutySeatDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ESAlertCalledDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ESAlertCalledRequestDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ExportDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.AlertCalled;
import com.yeejoin.amos.boot.module.tzs.api.entity.DispatchPaper;
import com.yeejoin.amos.boot.module.tzs.api.entity.Elevator;
import com.yeejoin.amos.boot.module.tzs.api.entity.TemplateExport;
import com.yeejoin.amos.boot.module.tzs.api.entity.WechatRelation;
import com.yeejoin.amos.boot.module.tzs.api.enums.AlertStageEnums;
import com.yeejoin.amos.boot.module.tzs.api.service.IMaintenanceUnitService;
import com.yeejoin.amos.boot.module.tzs.api.service.IUseUnitService;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.AlertCalledServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.AlertFormValueServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.DispatchPaperServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.ESAlertCalledService;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.ElevatorServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.TemplateExportServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.WechatRelationServiceImpl;
import com.yeejoin.amos.boot.module.tzs.biz.utils.AlertBeanDtoVoUtils;
import com.yeejoin.amos.boot.module.tzs.biz.utils.BeanDtoVoUtils;
......@@ -653,4 +660,6 @@ public class AlertCalledController extends BaseController {
return ResponseHelper.buildResponse(iAlertCalledService.getSeatInfos());
}
}
package com.yeejoin.amos.boot.module.tzs.biz.controller;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.google.common.collect.Lists;
import com.yeejoin.amos.boot.module.common.api.excel.ExcelUtil;
import com.yeejoin.amos.boot.module.tzs.api.dto.ElevatorDto;
import com.yeejoin.amos.boot.module.tzs.api.dto.ExportDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.TemplateExport;
import com.yeejoin.amos.boot.module.tzs.api.service.IMaintenanceUnitService;
import com.yeejoin.amos.boot.module.tzs.api.service.IUseUnitService;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.ElevatorServiceImpl;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.Api;
import org.springframework.web.bind.annotation.RestController;
import com.yeejoin.amos.boot.biz.common.controller.BaseController;
import java.util.List;
import com.yeejoin.amos.boot.module.tzs.biz.service.impl.TemplateExportServiceImpl;
import org.typroject.tyboot.core.foundation.utils.ValidationUtil;
import org.typroject.tyboot.core.restful.exception.instance.BadRequest;
import org.typroject.tyboot.core.restful.utils.ResponseHelper;
import org.typroject.tyboot.core.restful.utils.ResponseModel;
import org.springframework.beans.factory.annotation.Autowired;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.springframework.web.bind.annotation.*;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.yeejoin.amos.boot.module.tzs.api.dto.TemplateExportDto;
import org.typroject.tyboot.core.restful.doc.TycloudOperation;
import org.typroject.tyboot.core.foundation.enumeration.UserType;
/**
* 模板表
*
* @author system_generator
* @date 2021-10-25
*/
@RestController
@Api(tags = "导出Api")
@RequestMapping(value = "/template-export")
public class TemplateExportController extends BaseController {
@Autowired
TemplateExportServiceImpl templateExportServiceImpl;
@Autowired
IMaintenanceUnitService iMaintenanceUnitService;
@Autowired
IUseUnitService iUseUnitService;
@Autowired
ElevatorServiceImpl elevatorServiceImpl;
/**
* 新增模板表
*
* @return
*/
@TycloudOperation(ApiLevel = UserType.AGENCY)
@PostMapping(value = "/save")
@ApiOperation(httpMethod = "POST", value = "新增模板", notes = "新增模板")
public ResponseModel<TemplateExportDto> save(@RequestBody TemplateExportDto model) {
if (ValidationUtil.isEmpty(model.getTemplateName())
|| ValidationUtil.isEmpty(model.getTemplateContent())
|| ValidationUtil.isEmpty(model.getTemplateType())
) {
throw new BadRequest("参数校验失败.");
}
// 判断 同一类型模板名称不能重复
TemplateExport temp = templateExportServiceImpl.getOne(new LambdaQueryWrapper<TemplateExport>().eq(TemplateExport::getTemplateType,model.getTemplateType()).eq(TemplateExport::getTemplateName,model.getTemplateName()));
if(temp != null) {
throw new BadRequest("模板名称已存在.");
}
model.setIsStandard(false); // 新增模板表均不为标准模板
model = templateExportServiceImpl.createWithModel(model);
return ResponseHelper.buildResponse(model);
}
/**
* 根据sequenceNbr更新
*
* @param sequenceNbr 主键
* @return
*/
@TycloudOperation(ApiLevel = UserType.AGENCY)
@PutMapping(value = "/{sequenceNbr}")
@ApiOperation(httpMethod = "PUT", value = "根据sequenceNbr更新模板", notes = "根据sequenceNbr更新模板")
public ResponseModel<TemplateExportDto> updateBySequenceNbrTemplateExport(@RequestBody TemplateExportDto model,@PathVariable(value = "sequenceNbr") Long sequenceNbr) {
model.setSequenceNbr(sequenceNbr);
if (ValidationUtil.isEmpty(model.getTemplateType())
) {
throw new BadRequest("模板类型为必填项");
}
if(StringUtils.isNotBlank(model.getTemplateName())) {
TemplateExport temp = templateExportServiceImpl.getOne(new LambdaQueryWrapper<TemplateExport>().eq(TemplateExport::getTemplateType,model.getTemplateType()).
eq(TemplateExport::getTemplateName,model.getTemplateName()).
ne(TemplateExport::getSequenceNbr,sequenceNbr));
if(temp != null) {
throw new BadRequest("模板名称已存在.");
}
}
model.setIsStandard(false); // 更新模板表均不为标准模板
return ResponseHelper.buildResponse(templateExportServiceImpl.updateWithModel(model));
}
/**
* 根据sequenceNbr删除
*
* @param sequenceNbr 主键
* @return
*/
@TycloudOperation(ApiLevel = UserType.AGENCY)
@DeleteMapping(value = "/{sequenceNbr}")
@ApiOperation(httpMethod = "DELETE", value = "根据sequenceNbr删除模板", notes = "根据sequenceNbr删除模板")
public ResponseModel<Boolean> deleteBySequenceNbr(HttpServletRequest request, @PathVariable(value = "sequenceNbr") Long sequenceNbr){
// 不可删除标准模板
return ResponseHelper.buildResponse(templateExportServiceImpl.removeById(sequenceNbr));
}
/**
* 列表全部数据查询
*
* @return
*/
@TycloudOperation(ApiLevel = UserType.AGENCY)
@ApiOperation(httpMethod = "GET",value = "模板表列表全部数据查询", notes = "模板表列表全部数据查询")
@GetMapping(value = "/list/{type}")
public ResponseModel<List<TemplateExport>> selectForList(@PathVariable String type) {
List<TemplateExport> list = templateExportServiceImpl.list(new LambdaQueryWrapper<TemplateExport>().eq(TemplateExport::getIsDelete,0).eq(TemplateExport::getTemplateType,type).orderByDesc(TemplateExport::getIsStandard));
return ResponseHelper.buildResponse(list);
}
@TycloudOperation(ApiLevel = UserType.AGENCY, needAuth = false)
@ApiOperation(httpMethod = "POST", value = "根据模板导出", notes = "根据模板导出")
@PostMapping("/exportBytemolate")
public void testExport(@RequestBody ExportDto exportDto, HttpServletResponse response) {
if (ValidationUtil.isEmpty(exportDto.getExportArray())
|| ValidationUtil.isEmpty(exportDto.getDataType())
|| ValidationUtil.isEmpty(exportDto.getFileName())
|| ValidationUtil.isEmpty(exportDto.getExportType())
|| ValidationUtil.isEmpty(exportDto.getFileType())
) {
throw new BadRequest("参数校验失败.");
}
// 导出模块 电梯 维保单位 救援机构
if("1042".equals(exportDto.getDataType())) { // 选择列导出
if(ValidationUtil.isEmpty(exportDto.getExportId())) {
throw new BadRequest("未选择需要导出的数据.");
}
}
System.out.println(exportDto.getExportArray());
JSONArray jsonArray = JSONArray.parseArray(exportDto.getExportArray());
List<List<String>> heads = Lists.newArrayList();
List<String> headstr = Lists.newArrayList();
List list = null;
String sheetName = null;
for(int i = 0; i < jsonArray.size() ; i ++) {
JSONObject temp = jsonArray.getJSONObject(i);
headstr.add(temp.getString("key"));
List<String> tempList = Lists.newArrayList(temp.getString("value"));
heads.add(tempList);
}
String fileName = exportDto.getFileName();
if("ELEVATOR".equals(exportDto.getExportType())) { // 查询电梯数据
list = elevatorServiceImpl.selectExportData(exportDto.getExportId());
sheetName = "电梯信息";
} else if("MAINTENANCE_COMPANY".equals(exportDto.getExportType())) { // 查询维保单位数据
list = iMaintenanceUnitService.selectExportData(exportDto.getExportId());
sheetName = "维保单位";
} else if("USE_UNIT".equals(exportDto.getExportType())) { // 查询使用单位数据
sheetName = "使用单位";
list = iUseUnitService.selectExportData(exportDto.getExportId());
}
ExcelUtil.createTemplateWithHeaders(response, fileName, sheetName, list, ElevatorDto.class, null, false, heads, headstr, exportDto.getFileType());
}
}
......@@ -19,6 +19,7 @@ import com.yeejoin.amos.boot.module.tzs.api.mapper.ElevatorMapper;
import com.yeejoin.amos.boot.module.tzs.api.service.IElevatorService;
import com.yeejoin.amos.component.feign.model.FeignClientResult;
import com.yeejoin.amos.feign.systemctl.Systemctl;
import org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
......@@ -30,6 +31,7 @@ import org.typroject.tyboot.core.rdbms.service.BaseService;
import org.typroject.tyboot.core.restful.exception.instance.BadRequest;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
......@@ -195,6 +197,16 @@ public class ElevatorServiceImpl extends BaseService<ElevatorDto, Elevator, Elev
return filePath;
}
@Override
public List<ElevatorDto> selectExportData(String exportId) {
List<String> ids = null;
if(StringUtils.isNotEmpty(exportId)) {
String[] idStr = exportId.split(",");
ids = Arrays.asList(idStr);
}
return baseMapper.selectExportData(ids);
}
public List<ElevatorWlInfoDto> queryElevatorList(ElevatorWlInfoDto esElevatorDto) {
return elevatorMapper.queryElevatorList(esElevatorDto.getAddress(),
......
package com.yeejoin.amos.boot.module.tzs.biz.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yeejoin.amos.boot.module.tzs.api.dto.MaintenanceUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.MaintenanceUnit;
import com.yeejoin.amos.boot.module.tzs.api.mapper.MaintenanceUnitMapper;
import com.yeejoin.amos.boot.module.tzs.api.service.IMaintenanceUnitService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import java.util.Arrays;
import java.util.List;
/**
* 服务实现类
*
......@@ -15,4 +20,13 @@ import org.springframework.stereotype.Service;
@Service
public class MaintenanceUnitServiceImpl extends ServiceImpl<MaintenanceUnitMapper, MaintenanceUnit> implements IMaintenanceUnitService {
@Override
public List<MaintenanceUnitDto> selectExportData(String exportId) {
List<String> ids = null;
if(StringUtils.isNotEmpty(exportId)) {
String[] idStr = exportId.split(",");
ids = Arrays.asList(idStr);
}
return baseMapper.selectExportData(ids);
}
}
package com.yeejoin.amos.boot.module.tzs.biz.service.impl;
import com.yeejoin.amos.boot.module.tzs.api.entity.TemplateExport;
import com.yeejoin.amos.boot.module.tzs.api.mapper.TemplateExportMapper;
import com.yeejoin.amos.boot.module.tzs.api.service.ITemplateExportService;
import com.yeejoin.amos.boot.module.tzs.api.dto.TemplateExportDto;
import org.typroject.tyboot.core.rdbms.service.BaseService;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import java.util.List;
/**
* 模板表服务实现类
*
* @author system_generator
* @date 2021-10-25
*/
@Service
public class TemplateExportServiceImpl extends BaseService<TemplateExportDto,TemplateExport,TemplateExportMapper> implements ITemplateExportService {
/**
* 分页查询
*/
public Page<TemplateExportDto> queryForTemplateExportPage(Page<TemplateExportDto> page) {
return this.queryForPage(page, null, false);
}
/**
* 列表查询 示例
*/
public List<TemplateExportDto> queryForTemplateExportList() {
return this.queryForList("" , false);
}
}
\ No newline at end of file
......@@ -4,11 +4,15 @@ import com.yeejoin.amos.boot.module.tzs.api.dto.UseUnitDto;
import com.yeejoin.amos.boot.module.tzs.api.entity.UseUnit;
import com.yeejoin.amos.boot.module.tzs.api.mapper.UseUnitMapper;
import com.yeejoin.amos.boot.module.tzs.api.service.IUseUnitService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import org.typroject.tyboot.core.rdbms.service.BaseService;
import java.util.Arrays;
import java.util.List;
/**
* 服务实现类
* 服务实现类return null;
*
* @author tb
* @date 2021-06-01
......@@ -16,4 +20,13 @@ import org.typroject.tyboot.core.rdbms.service.BaseService;
@Service
public class UseUnitServiceImpl extends BaseService<UseUnitDto, UseUnit, UseUnitMapper> implements IUseUnitService {
@Override
public List<UseUnitDto> selectExportData(String exportId) {
List<String> ids = null;
if(StringUtils.isNotEmpty(exportId)) {
String[] idStr = exportId.split(",");
ids = Arrays.asList(idStr);
}
return baseMapper.selectExportData(ids);
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment