jianjun
4 years ago
5 changed files with 237 additions and 402 deletions
@ -1,47 +0,0 @@ |
|||
/** |
|||
* Copyright 2018 人人开源 https://www.renren.io
|
|||
* <p> |
|||
* This program is free software: you can redistribute it and/or modify |
|||
* it under the terms of the GNU General Public License as published by |
|||
* the Free Software Foundation, either version 3 of the License, or |
|||
* (at your option) any later version. |
|||
* <p> |
|||
* This program is distributed in the hope that it will be useful, |
|||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
* GNU General Public License for more details. |
|||
* <p> |
|||
* You should have received a copy of the GNU General Public License |
|||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
|
|||
package com.epmet.redis; |
|||
|
|||
import com.epmet.commons.tools.redis.RedisUtils; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.stereotype.Component; |
|||
|
|||
/** |
|||
* 用户基础信息 |
|||
* |
|||
* @author generator generator@elink-cn.com |
|||
* @since v1.0.0 2021-10-26 |
|||
*/ |
|||
@Component |
|||
public class IcResiUserRedis { |
|||
@Autowired |
|||
private RedisUtils redisUtils; |
|||
|
|||
public void delete(Object[] ids) { |
|||
|
|||
} |
|||
|
|||
public void set(){ |
|||
|
|||
} |
|||
|
|||
public String get(String id){ |
|||
return null; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,183 @@ |
|||
package com.epmet.epmetuser.test; |
|||
|
|||
import cn.afterturn.easypoi.excel.entity.ExportParams; |
|||
import cn.afterturn.easypoi.excel.entity.params.ExcelExportEntity; |
|||
import cn.afterturn.easypoi.excel.export.ExcelExportService; |
|||
import com.alibaba.fastjson.JSON; |
|||
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; |
|||
import com.epmet.commons.tools.exception.EpmetErrorCode; |
|||
import com.epmet.commons.tools.exception.RenException; |
|||
import com.epmet.commons.tools.utils.Result; |
|||
import com.epmet.dto.form.CustomerFormQueryDTO; |
|||
import com.epmet.dto.result.CustomerFormResultDTO; |
|||
import com.epmet.dto.result.FormGroupDTO; |
|||
import com.epmet.dto.result.FormItem; |
|||
import com.epmet.feign.OperCustomizeOpenFeignClient; |
|||
import org.apache.commons.lang3.StringUtils; |
|||
import org.apache.poi.xssf.usermodel.XSSFWorkbook; |
|||
import org.jetbrains.annotations.NotNull; |
|||
import org.junit.Test; |
|||
import org.junit.runner.RunWith; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
import org.springframework.test.context.junit4.SpringRunner; |
|||
import org.springframework.web.bind.annotation.RequestParam; |
|||
|
|||
import java.io.FileOutputStream; |
|||
import java.util.*; |
|||
|
|||
@SpringBootTest |
|||
@RunWith(SpringRunner.class) |
|||
public class IcResiUserControllerTest { |
|||
private static final String BASE_TABLE_NAME = "BASE_TABLE_NAME"; |
|||
@Autowired |
|||
private OperCustomizeOpenFeignClient operCustomizeOpenFeignClient; |
|||
|
|||
@Test |
|||
public void downloadTemplate(@RequestParam String customerId) throws Exception { |
|||
CustomerFormResultDTO resultForm = getResiFormItems(customerId); |
|||
|
|||
XSSFWorkbook workbook = new XSSFWorkbook(); |
|||
Map<String, List<ExcelExportEntity>> sheetHeaderMap = buildHeaderByItem(resultForm); |
|||
//Workbook workbook = null;
|
|||
for (Map.Entry<String, List<ExcelExportEntity>> entry : sheetHeaderMap.entrySet()) { |
|||
String sheetName = entry.getKey(); |
|||
List<ExcelExportEntity> headers = entry.getValue(); |
|||
System.out.println("headers:"+sheetName+JSON.toJSONString(headers)); |
|||
ExportParams exportParams = new ExportParams(null,sheetName); |
|||
//exportParams.setAutoSize(true);
|
|||
List<Map<String,String>> dataSet = new ArrayList<>(); |
|||
HashMap<String, String> map = new HashMap<>(); |
|||
map.put("1","2"); |
|||
dataSet.add(map); |
|||
new ExcelExportService().createSheetForMap(workbook, exportParams, headers, dataSet); |
|||
} |
|||
|
|||
FileOutputStream fos = new FileOutputStream("//Users/liujianjun/Downloads/基础信息表/Dow.tt.xls"); |
|||
workbook.write(fos); |
|||
fos.close(); |
|||
} |
|||
|
|||
@NotNull |
|||
private CustomerFormResultDTO getResiFormItems(String customerId) { |
|||
CustomerFormQueryDTO queryDTO = new CustomerFormQueryDTO(); |
|||
queryDTO.setFormCode("resi_base_info"); |
|||
queryDTO.setCustomerId(customerId); |
|||
Result<CustomerFormResultDTO> resultForm = operCustomizeOpenFeignClient.getCustomerForm(queryDTO); |
|||
if (resultForm == null || !resultForm.success() ||resultForm.getData() == null){ |
|||
throw new RenException(EpmetErrorCode.SERVER_ERROR.getCode()); |
|||
} |
|||
System.out.println(JSON.toJSONString(resultForm.getData())); |
|||
return resultForm.getData(); |
|||
} |
|||
|
|||
@NotNull |
|||
private Map<String,List<ExcelExportEntity>> buildHeaderByItem(CustomerFormResultDTO resultForm) { |
|||
//form中的itemlist 为一级表头 但是要排除每个item中含有childGroup的
|
|||
|
|||
List<FormItem> itemList = resultForm.getItemList(); |
|||
List<FormGroupDTO> groupList = resultForm.getGroupList(); |
|||
|
|||
Map<String,List<ExcelExportEntity>> everySheetHeaderMap = new LinkedHashMap<>(); |
|||
|
|||
List<ExcelExportEntity> firstSheetHeaderList = new ArrayList<>(); |
|||
//Map<String, String> groupNameMap = groupList.stream().collect(Collectors.toMap(FormGroupDTO::getGroupId,FormGroupDTO::getLabel));
|
|||
itemList.forEach(item->{ |
|||
if (StringUtils.isBlank(item.getColumnName())){ |
|||
return; |
|||
} |
|||
|
|||
if (item.getChildGroup() == null){ |
|||
ExcelExportEntity header = new ExcelExportEntity(item.getLabel(),item.getColumnName().concat(String.valueOf(item.getColumnNum())),30); |
|||
header.setNeedMerge(true); |
|||
firstSheetHeaderList.add(header); |
|||
return; |
|||
} |
|||
everySheetHeaderMap.putIfAbsent(resultForm.getFormName(),firstSheetHeaderList); |
|||
|
|||
//这些是动态的 formGroup
|
|||
if (item.getChildGroup() != null){ |
|||
//baseTableName单独的一个sheet
|
|||
System.out.println("childGroup:"+item.getLabel()); |
|||
if (BASE_TABLE_NAME.equals(item.getTableName())){ |
|||
List<ExcelExportEntity> otherSheetHeaderList = new ArrayList<>(); |
|||
//这里是设置除基础信息之外的sheet的表头
|
|||
item.getChildGroup().getItemList().forEach(item2->{ |
|||
if (!BASE_TABLE_NAME.equals(item2.getTableName())){ |
|||
everySheetHeaderMap.putIfAbsent(item.getLabel(),otherSheetHeaderList); |
|||
} |
|||
ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum()))); |
|||
|
|||
otherSheetHeaderList.add(secondHeader); |
|||
if (!"radio".equals(item2.getItemType()) && CollectionUtils.isNotEmpty(item2.getOptions())){ |
|||
secondHeader.setNeedMerge(true); |
|||
List<ExcelExportEntity> thirdHeaderList = new ArrayList<>(); |
|||
item2.getOptions().forEach(child->{ |
|||
ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel()); |
|||
thirdHeaderList.add(thirdHeader); |
|||
}); |
|||
secondHeader.setList(thirdHeaderList); |
|||
} |
|||
|
|||
}); |
|||
} |
|||
} |
|||
}); |
|||
groupList.forEach(item->{ |
|||
/* if (!"兴趣爱好".equals(item.getLabel()) && !"宗教信仰".equals(item.getLabel())){ |
|||
return; |
|||
}*/ |
|||
if (!BASE_TABLE_NAME.equals(item.getTableName())){ |
|||
return; |
|||
} |
|||
ExcelExportEntity header = new ExcelExportEntity(item.getLabel(),item.getTableName()); |
|||
header.setNeedMerge(true); |
|||
List<ExcelExportEntity> secondHeaderList = new ArrayList<>(); |
|||
item.getItemList().forEach(item2->{ |
|||
ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum()))); |
|||
secondHeader.setNeedMerge(true); |
|||
secondHeaderList.add(secondHeader); |
|||
if (com.baomidou.mybatisplus.core.toolkit.CollectionUtils.isNotEmpty(item2.getOptions())){ |
|||
List<ExcelExportEntity> thirdHeaderList = new ArrayList<>(); |
|||
item2.getOptions().forEach(child->{ |
|||
ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel(),child.getValue()+ new Random(1).nextInt(2)); |
|||
thirdHeader.setNeedMerge(true); |
|||
thirdHeaderList.add(thirdHeader); |
|||
}); |
|||
secondHeader.setList(thirdHeaderList); |
|||
} |
|||
|
|||
}); |
|||
header.setList(secondHeaderList); |
|||
firstSheetHeaderList.add(header); |
|||
System.out.println(JSON.toJSONString(firstSheetHeaderList)); |
|||
}); |
|||
return everySheetHeaderMap; |
|||
} |
|||
|
|||
private void buildHeader(Map<String, List<ExcelExportEntity>> everySheetHeaderMap, FormItem item, ExcelExportEntity header) { |
|||
List<ExcelExportEntity> firstSheetHeaderList = new ArrayList<>(); |
|||
List<ExcelExportEntity> secondHeaderList = new ArrayList<>(); |
|||
item.getChildGroup().getItemList().forEach(item2->{ |
|||
if (!BASE_TABLE_NAME.equals(item2.getTableName())){ |
|||
everySheetHeaderMap.putIfAbsent(item.getLabel(),firstSheetHeaderList); |
|||
} |
|||
ExcelExportEntity secondHeader = new ExcelExportEntity(item2.getLabel(),item2.getColumnName().concat(String.valueOf(item2.getColumnNum()))); |
|||
secondHeader.setNeedMerge(true); |
|||
secondHeaderList.add(secondHeader); |
|||
if (CollectionUtils.isNotEmpty(item2.getOptions())){ |
|||
List<ExcelExportEntity> thirdHeaderList = new ArrayList<>(); |
|||
item2.getOptions().forEach(child->{ |
|||
ExcelExportEntity thirdHeader = new ExcelExportEntity(child.getLabel()); |
|||
thirdHeaderList.add(thirdHeader); |
|||
}); |
|||
secondHeader.setList(thirdHeaderList); |
|||
} |
|||
|
|||
}); |
|||
header.setList(secondHeaderList); |
|||
firstSheetHeaderList.add(header); |
|||
everySheetHeaderMap.putIfAbsent(item.getLabel(),firstSheetHeaderList); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue