package com.zzg.web.controller.land;
|
|
import com.zzg.common.core.domain.AjaxResult;
|
import com.zzg.common.core.domain.entity.system.SysUser;
|
import com.zzg.common.core.domain.model.LoginUser;
|
import com.zzg.common.utils.ExcelUtils;
|
import com.zzg.common.utils.SecurityUtils;
|
import com.zzg.common.utils.uuid.IdUtils;
|
import com.zzg.system.domain.SysCity;
|
import com.zzg.system.domain.vo.City;
|
import com.zzg.system.service.system.ISysCityService;
|
import com.zzg.system.service.system.ISysUserService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.annotation.Resource;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Optional;
|
import java.util.stream.Collectors;
|
|
@Api(tags = "测试")
|
@RestController
|
@RequestMapping("demo")
|
public class DemoController {
|
|
@Resource
|
ISysUserService userService;
|
@Resource
|
ISysCityService sysCityService;
|
|
@ApiOperation("用户列表")
|
@GetMapping("list")
|
public AjaxResult getUserList() {
|
List<SysUser> list = userService.list();
|
|
return AjaxResult.success(list);
|
}
|
|
@ApiOperation("获取当前登录用户信息")
|
@GetMapping("getLoginUser")
|
public AjaxResult getLoginUser() {
|
LoginUser loginUser = SecurityUtils.getLoginUser();
|
return AjaxResult.success(loginUser);
|
}
|
|
@ApiOperation("初始化村组信息")
|
@GetMapping("/initGroups")
|
public AjaxResult initGroups() throws IOException {
|
File file = new File("D:\\Work\\崇州征地项目\\崇州村组(2).xlsx");
|
FileInputStream fileInputStream = new FileInputStream(file);
|
|
List<City> list = Optional.ofNullable(ExcelUtils.importExcel(fileInputStream, 0, 1, City.class)).orElse(new ArrayList<>());
|
Map<String, List<City>> collect = list.stream().collect(Collectors.groupingBy(City::getTown));
|
|
List<SysCity> groupList = new ArrayList<>();
|
SysCity district = new SysCity();
|
district.setId("0");
|
district.setPid("-1");
|
district.setName("崇州市");
|
district.setLevel(0);
|
district.setState(0);
|
district.setRanking(1);
|
groupList.add(district);
|
|
Integer a = 1;
|
for (Map.Entry<String, List<City>> entry : collect.entrySet()) {
|
String townName = entry.getKey();
|
SysCity town = new SysCity();
|
town.setId(IdUtils.fastSimpleUUID());
|
town.setPid("0");
|
town.setName(townName);
|
town.setLevel(1);
|
town.setState(0);
|
town.setRanking(a);
|
a += 1;
|
groupList.add(town);
|
List<City> value = entry.getValue();
|
Integer b = 1;
|
for (City c : value) {
|
SysCity village = new SysCity();
|
village.setId(IdUtils.fastSimpleUUID());
|
village.setPid(town.getId());
|
village.setName(c.getVillage());
|
village.setLevel(2);
|
village.setState(0);
|
village.setRanking(b);
|
b += 1;
|
groupList.add(village);
|
|
for (int i = 1; i <= c.getNum(); i++) {
|
SysCity group = new SysCity();
|
group.setId(IdUtils.fastSimpleUUID());
|
group.setPid(village.getId());
|
group.setName(i + "组");
|
group.setLevel(3);
|
group.setState(0);
|
group.setRanking(i);
|
groupList.add(group);
|
}
|
if (c.getType().equals("社区")) {
|
SysCity group = new SysCity();
|
group.setId(IdUtils.fastSimpleUUID());
|
group.setPid(village.getId());
|
group.setName("社区集体");
|
group.setLevel(3);
|
group.setState(0);
|
group.setRanking(c.getNum() + 1);
|
groupList.add(group);
|
} else {
|
SysCity group = new SysCity();
|
group.setId(IdUtils.fastSimpleUUID());
|
group.setPid(village.getId());
|
group.setName("村集体");
|
group.setLevel(3);
|
group.setState(0);
|
group.setRanking(c.getNum() + 1);
|
groupList.add(group);
|
}
|
}
|
}
|
sysCityService.saveBatch(groupList);
|
|
return AjaxResult.success();
|
}
|
|
}
|
