lyf
/
fxkh-txl-service
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

通过行政区划下载excel新增

master
徐杰盟 2024-05-28 13:05:31 +08:00
parent 497a655ee8
commit 8efc4678c4
1 changed files with 51 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/main/java/com/whdc/controller/AddressBookController.java
View File

@ -3,11 +3,9 @@ package com.whdc.controller;
import cn.dev33.satoken.stp.StpUtil;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.whdc.exception.MyException;
import com.whdc.model.dto.AddressBootDto;
import com.whdc.model.entity.AbUdR;
import com.whdc.model.entity.AddressBook;
import com.whdc.model.entity.Organization;
import com.whdc.model.entity.User;
import com.whdc.model.entity.*;
import com.whdc.model.enums.Role;
import com.whdc.model.enums.VersionsType;
import com.whdc.model.group.Insert;
@ -429,22 +427,26 @@ public class AddressBookController {
}
@ApiOperation(value = "通过行政区划下载excel")
@GetMapping(value = "getExcel/{adnm}")
public ResultJson getExcel(@PathVariable("adnm") String adnm, HttpServletResponse response) {
public void getExcel(@PathVariable("adnm") String adnm, HttpServletResponse response) {
List<AdcdTree> tree = adinfoService.tree(adnm, adnm);
// 查询全部标识
if ("all".equals(adnm)) {
adnm = null;
}
List<AdcdTree> tree = adinfoService.tree(null, null);
if (CollectionUtils.isEmpty(tree)) {
return ResultJson.error("当前行政区划不存在");
throw new MyException("当前行政区划不存在");
}
List<AddressBook> addressBooks = service.lambdaQuery().isNotNull(AddressBook::getAdcd)
.list();
if (CollectionUtils.isEmpty(addressBooks)) {
return ResultJson.error("行政区划责任人为空");
throw new MyException("行政区划责任人为空");
}
Map<String, List<AddressBook>> abMap = addressBooks.stream().collect(Collectors.groupingBy(AddressBook::getAdcd, Collectors.toList()));
@ -452,32 +454,56 @@ public class AddressBookController {
// 行政区划责任人数据
List<ExcelABVo> data = Lists.newArrayList();
tree.forEach(t ->{
String finalAdnm = adnm;
for (int i = 0; i < tree.size(); i++) {
AdcdTree t = tree.get(i);
String adcd = t.getAdcd();
String cityAdnm = t.getAdnm();
List<AddressBook> books = abMap.get(adcd);
if (CollectionUtils.isEmpty(books)) {
continue;
}
books.forEach(b -> {
if (StringUtils.isNotBlank(finalAdnm)) {
if (cityAdnm.equals(finalAdnm)) {
data.add(new ExcelABVo(cityAdnm, cityAdnm, b.getName(), b.getPosition(), b.getPhone()));
}
} else {
data.add(new ExcelABVo(cityAdnm, cityAdnm, b.getName(), b.getPosition(), b.getPhone()));
}
});
t.getChildren().forEach(tc ->{
List<AdcdTree> children = t.getChildren();
if (CollectionUtils.isNotEmpty(children)) {
for (int i1 = 1; i1 < children.size(); i1++) {
AdcdTree tc = children.get(i1);
String tcAdcd = tc.getAdcd();
String tcCityAdnm = tc.getAdnm();
List<AddressBook> tcBooks = abMap.get(tcAdcd);
books.forEach(b ->{
if (CollectionUtils.isEmpty(tcBooks)) {
continue;
}
tcBooks.forEach(b -> {
if (StringUtils.isNotBlank(finalAdnm)) {
if (cityAdnm.equals(finalAdnm) || tcCityAdnm.equals(finalAdnm)) {
data.add(new ExcelABVo(cityAdnm, tcCityAdnm, b.getName(), b.getPosition(), b.getPhone()));
}
} else {
data.add(new ExcelABVo(cityAdnm, tcCityAdnm, b.getName(), b.getPosition(), b.getPhone()));
}
});
});
});
}
}
};
if (CollectionUtils.isEmpty(data)) {
return ResultJson.error("当前行政区划责任人为空");
throw new MyException("当前行政区划责任人为空");
}
ExcelCommon.exportExcel(data,
"防汛抗旱通讯录", "防汛抗旱通讯录", ExcelABVo.class, "防汛抗旱通讯录_" + adnm + "_" + LocalDateTime.now() + ".xlsx",
response);
return ResultJson.ok("");
}
}