import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.ExcelWriter;
|
import com.alibaba.excel.util.MapUtils;
|
import com.alibaba.excel.write.metadata.WriteSheet;
|
import com.alibaba.excel.write.metadata.fill.FillConfig;
|
import com.alibaba.fastjson.JSON;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import lombok.EqualsAndHashCode;
|
import lombok.Getter;
|
import lombok.Setter;
|
import org.apache.commons.lang3.StringUtils;
|
import org.jeecg.LbProjectCloudApplication;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.util.DateUtils;
|
import org.jeecg.common.util.oConvertUtils;
|
import org.jeecg.modules.project.entity.ProProject;
|
import org.jeecg.modules.project.entity.ProProjectLink;
|
import org.jeecg.modules.project.entity.ProProjectLinkTree;
|
import org.jeecg.modules.project.service.IProProjectLinkService;
|
import org.jeecg.modules.project.service.IProProjectService;
|
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 java.io.File;
|
import java.io.IOException;
|
import java.time.LocalDate;
|
import java.time.Month;
|
import java.time.temporal.ChronoUnit;
|
import java.time.temporal.TemporalAdjusters;
|
import java.util.*;
|
|
@RunWith(SpringRunner.class)
|
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = LbProjectCloudApplication.class)
|
|
public class ProjectTest {
|
@Autowired
|
private IProProjectService proProjectService;
|
|
@Autowired
|
private IProProjectLinkService proLinkService;
|
|
@Test
|
public void queryPageList() {
|
QueryWrapper<ProProject> queryWrapper = new QueryWrapper<>();
|
List<ProProject> list = proProjectService.list(queryWrapper);
|
//生成动态表头
|
List<Map<String, Object>> dynamcTableHead = createDynamcTableHeader(list);
|
//生成数据
|
List<Map<String, Object>> dynamcTableData = createDynamcTableData(list);
|
// System.err.println(dynamcTableHead);
|
// System.err.println(dynamcTableData);
|
|
|
String curWeekStart = DateUtils.weekToDayStartStr(1);
|
String curWeekEnd = DateUtils.weekToDayEndStr(1);
|
System.err.println(curWeekStart);
|
System.err.println(curWeekEnd);
|
for (int i = 0; i < 12; i++) {
|
Calendar c = Calendar.getInstance();
|
c.set(Calendar.YEAR, 2022); //
|
c.set(Calendar.MONTH, i + 1); //
|
System.err.println("周数" + (i + 1) + ":" + c.getActualMaximum(Calendar.WEEK_OF_MONTH));
|
}
|
|
// 随意取一个时间，取了当前时间
|
LocalDate localDate = LocalDate.of(2022, 1, 2);
|
System.out.println("当前时间为：" + localDate);
|
|
// 根据封装好的月份获取一个一月到十二月的Month流，此时流里的对象为Month
|
Arrays.stream(Month.values())
|
// 把每个month都调整到当前这个时间里，此时流的对象为LocalDate
|
.map(month -> month.adjustInto(localDate))
|
// 这里方便里观察此时流的数据，把转换后的时间打印了出来
|
.peek(System.out::println)
|
// 根据时间的所在月的第一天和最后一天作为间隔计算周数
|
.map(temporal -> ChronoUnit.WEEKS.between(temporal.with(TemporalAdjusters.firstDayOfMonth()), temporal.with(TemporalAdjusters.lastDayOfMonth())))
|
// 打印最后的周数
|
.forEach(System.out::println);
|
|
}
|
|
/**
|
* 构建动态表格数据
|
*/
|
private List<Map<String, Object>> createDynamcTableData(List<ProProject> list) {
|
List<Map<String, Object>> res = new ArrayList<>();
|
Map<String, Object> map = new HashMap<>();
|
list.forEach(item -> {
|
map.put("id", item.getId());
|
map.put("xmmc", item.getXmmc());
|
map.put("wcjd", item.getWcjd());
|
String xmcy = item.getXmcy();
|
if (!StringUtils.isEmpty(xmcy)) {
|
String[] split = xmcy.split(",");
|
for (int i = 0; i < split.length; i++) {
|
map.put(split[i], 1);
|
}
|
}
|
String xmfzr = item.getXmfzr();
|
if (!StringUtils.isEmpty(xmfzr)) {
|
map.put(xmfzr, 1);
|
}
|
|
res.add(map);
|
|
});
|
return res;
|
}
|
|
/**
|
* 构建动态表格头部
|
*/
|
private List<Map<String, Object>> createDynamcTableHeader(List<ProProject> records) {
|
List<Map<String, Object>> res = new ArrayList<>();
|
TreeSet<String> userTree = new TreeSet<>();
|
records.forEach(item -> {
|
if (!StringUtils.isEmpty(item.getXmfzr())) {
|
userTree.add(item.getXmfzr());
|
}
|
if (!StringUtils.isEmpty(item.getXmcy())) {
|
String[] split = item.getXmcy().split(",");
|
for (int i = 0; i < split.length; i++) {
|
String user = split[i];
|
userTree.add(user);
|
}
|
|
}
|
});
|
for (String item : userTree) {
|
Map<String, Object> map = new HashMap<>();
|
map.put("name", item);
|
res.add(map);
|
}
|
return res;
|
|
}
|
|
@Test
|
public void getW() {
|
int curYear = DateUtils.getCurYear();
|
int sum = 0;
|
for (int i = 1; i <= 12; i++) {
|
LocalDate date = LocalDate.of(curYear, i, 1);
|
// 获取星期几
|
int week = date.getDayOfWeek().getValue();
|
date = date.with(TemporalAdjusters.lastDayOfMonth());
|
//这个月最后一天
|
int dayOfMonth = date.getDayOfMonth();
|
// 打印
|
int k = 0;
|
for (int j = 1; j <= dayOfMonth; j++) {
|
LocalDate everyDay = LocalDate.of(curYear, i, j);
|
int w = everyDay.getDayOfWeek().getValue();
|
System.err.println("日期：" + everyDay + "-星期：" + w);
|
if (w % 7 == 0) {
|
k++;
|
sum++;
|
String weekToDayStartStr = DateUtils.weekToDayStartStr(sum);
|
String weekToDayEndStr = DateUtils.weekToDayEndStr(sum);
|
System.out.println(i + "月：" + k + "周-No:" + sum + "------" + weekToDayStartStr + "==" + weekToDayEndStr);
|
}
|
}
|
|
}
|
|
}
|
|
//计算菜单等级
|
@Test
|
public void docTest(){
|
String sxh = "10.2.311.4";
|
char[] chars = sxh.toCharArray();
|
char dot = '.';
|
int dotCount = 0 ;
|
for (int i = 0; i < sxh.length(); i++) {
|
if(dot==chars[i]) dotCount ++;
|
}
|
if(dotCount==3){
|
String substring1 = sxh.substring(0, sxh.indexOf("."));
|
sxh = sxh.substring(sxh.indexOf(".")+1);
|
String substring2 = sxh.substring(0, sxh.indexOf("."));
|
sxh = sxh.substring(sxh.indexOf(".")+1);
|
String substring3 = sxh.substring(0, sxh.indexOf("."));
|
sxh = sxh.substring(sxh.indexOf(".")+1);
|
String substring4 = sxh;
|
System.err.println(substring1);
|
System.err.println(substring2);
|
System.err.println(substring3);
|
System.err.println(substring4);
|
System.err.println(sxh);
|
|
}
|
}
|
//
|
@Test
|
public void proLinkTest(){
|
Result<List<ProProjectLinkTree>> result = new Result<List<ProProjectLinkTree>>();
|
Map<String, Object> map = new HashMap<>();
|
map.put("pro", "1565533629802356737");
|
List<ProProjectLink> list = proLinkService.listByMap(map);
|
Collections.sort(list, (o1, o2) -> o1.getSortNo().compareTo(o2.getSortNo()));
|
List<ProProjectLinkTree> res = new ArrayList<>();
|
getTreeList(res, list, null);
|
getSxh2(res,null,false );
|
System.err.println(JSON.toJSON(res));
|
}
|
|
private void getTreeList(List<ProProjectLinkTree> treeList, List<ProProjectLink> metaList, ProProjectLinkTree temp) {
|
for (int i = 0 ; i< metaList.size() ; i ++) {
|
ProProjectLink permission = metaList.get(i);
|
String tempPid = permission.getPid();
|
ProProjectLinkTree tree = new ProProjectLinkTree(permission);
|
if (temp == null && oConvertUtils.isEmpty(tempPid)) {
|
treeList.add(tree);
|
if (!tree.isLeaf()) {
|
getTreeList(treeList, metaList, tree);
|
}
|
} else if (temp != null && tempPid != null && tempPid.equals(temp.getId())) {
|
temp.getChildren().add(tree);
|
if (!tree.isLeaf()) {
|
getTreeList(treeList, metaList, tree);
|
}
|
}
|
|
}
|
}
|
|
//递归计算
|
|
/**
|
* 递归计算序号
|
* @param treeList
|
* @param oxh p序号
|
* @param isChildren 是否子节点
|
*/
|
private void getSxh2(List<ProProjectLinkTree> treeList ,String oxh,boolean isChildren ){
|
for (int i = 0; i < treeList.size(); i++) {
|
ProProjectLinkTree tree = treeList.get(i);
|
if(tree.getPid()==null) oxh = (i+1) +"";
|
if(isChildren){
|
tree.setSxh(oxh + "." + (i+1));
|
}else {
|
tree.setSxh(oxh);
|
}
|
if(tree.getChildren()!=null && tree.getChildren().size()>0){
|
if(tree.getLevel()>1) oxh += "." + (i+1);
|
getSxh2(tree.getChildren(),oxh,true);
|
}
|
|
}
|
}
|
|
/**
|
* 生成序号 和 a-select-tree组件需要的title 目前仅支持三级数据 TODO：递归支持无限极
|
*/
|
private void getSxh(List<ProProjectLinkTree> treeList) {
|
|
for (int i = 0; i < treeList.size(); i++) {
|
ProProjectLinkTree tree1 = treeList.get(i);
|
tree1.setSxh((i + 1) + "");
|
tree1.setTitle(tree1.getSxh() + org.jeecg.common.util.StringUtils.getNullStr(tree1.getXmhj()));
|
if (tree1.getChildren() != null && tree1.getChildren().size() > 0) {
|
List<ProProjectLinkTree> children1 = tree1.getChildren();
|
for (int j = 0; j < children1.size(); j++) {
|
ProProjectLinkTree tree2 = children1.get(j);
|
tree2.setSxh((i +1)+ "." +( j + 1));
|
tree2.setTitle(tree2.getSxh() + org.jeecg.common.util.StringUtils.getNullStr(tree2.getXmhj()));
|
if (tree2.getChildren() != null && tree2.getChildren().size() > 0) {
|
List<ProProjectLinkTree> children2 = tree2.getChildren();
|
for (int k = 0; k < children2.size(); k++) {
|
ProProjectLinkTree tree3 = children2.get(k);
|
tree3.setSxh((i +1)+ "." + ( j + 1) + '.' + (k + 1));
|
tree3.setTitle(tree3.getSxh() + org.jeecg.common.util.StringUtils.getNullStr(tree3.getXmhj()));
|
if (tree3.getChildren() != null && tree3.getChildren().size() > 0) {
|
List<ProProjectLinkTree> children3 = tree3.getChildren();
|
for (int l = 0; l < children3.size(); l++) {
|
ProProjectLinkTree tree4 = children3.get(l);
|
tree4.setSxh((i +1)+ "." + ( j + 1) + '.' + (k + 1)+ '.' + (l + 1));
|
tree4.setTitle(tree4.getSxh() + org.jeecg.common.util.StringUtils.getNullStr(tree4.getXmhj()));
|
}
|
}
|
}
|
}
|
}
|
}
|
|
}
|
|
}
|
|
|
@Test
|
public void exportExcel(){
|
String roopath = System.getProperty("user.dir");
|
roopath = roopath.substring(0,roopath.indexOf("/lb-boot/")+8) ;
|
|
String templateFileName = roopath+ File.separator + "export-template"+ File.separator + "weekly"+ File.separator + "complexFillWithTable.xlsx";
|
|
System.err.println(templateFileName);
|
String fileName = roopath + File.separator + "export"+ File.separator + System.currentTimeMillis() + ".xlsx";
|
System.err.println(fileName);
|
// 方案1
|
|
|
|
ExcelWriter excelWriter = EasyExcel.write(fileName).withTemplate(templateFileName).build();
|
WriteSheet writeSheet = EasyExcel.writerSheet().build();
|
// 直接写入数据
|
excelWriter.fill(data(), writeSheet);
|
excelWriter.fill(data(), writeSheet);
|
|
// 写入list之前的数据
|
Map<String, Object> map = new HashMap<String, Object>();
|
map.put("date", "2019年10月9日13:28:28");
|
excelWriter.fill(map, writeSheet);
|
|
// list 后面还有个统计 想办法手动写入
|
// 这里偷懒直接用list 也可以用对象
|
List<List<String>> totalListList = new ArrayList<List<String>>();
|
List<String> totalList = new ArrayList<String>();
|
totalListList.add(totalList);
|
totalList.add(null);
|
totalList.add(null);
|
totalList.add(null);
|
// 第四列
|
totalList.add("统计:1000");
|
// 这里是write 别和fill 搞错了
|
excelWriter.write(totalListList, writeSheet);
|
excelWriter.finish();
|
// 总体上写法比较复杂 但是也没有想到好的版本 异步的去写入excel 不支持行的删除和移动，也不支持备注这种的写入，所以也排除了可以
|
// 新建一个 然后一点点复制过来的方案，最后导致list需要新增行的时候，后面的列的数据没法后移，后续会继续想想解决方案
|
|
|
}
|
|
private List<FillData> data() {
|
List<FillData> list = new ArrayList<FillData>();
|
for (int i = 0; i < 10; i++) {
|
FillData fillData = new FillData();
|
list.add(fillData);
|
fillData.setName("张三");
|
fillData.setNumber(5.2);
|
}
|
return list;
|
}
|
|
|
@Getter
|
@Setter
|
@EqualsAndHashCode
|
public class FillData {
|
private String name;
|
private double number;
|
private Date date;
|
|
}
|
|
|
|
@Test
|
public void simple(){
|
// 模板注意 用{} 来表示你要用的变量 如果本来就有"{","}" 特殊字符 用"\{","\}"代替
|
String roopath = System.getProperty("user.dir");
|
roopath = roopath.substring(0,roopath.indexOf("/lb-boot/")+8) ;
|
|
String templateFileName = roopath+ File.separator + "export-template"+ File.separator + "weekly"+ File.separator + "weekly.xlsx";
|
|
System.err.println(templateFileName);
|
String fileName = roopath + File.separator + "export"+ File.separator + System.currentTimeMillis() + ".xlsx";
|
System.err.println(fileName); // 这里 会填充到第一个sheet， 然后文件流会自动关闭
|
FillData fillData = new FillData();
|
fillData.setName("张三");
|
fillData.setNumber(5.2);
|
EasyExcel.write(fileName).withTemplate(templateFileName).sheet(1).doFill(fillData);
|
|
}
|
|
|
@Test
|
public void exportWeekly(){
|
Calendar calendar = Calendar.getInstance();
|
calendar.setTime(new Date());
|
int week = calendar.get(Calendar.WEEK_OF_MONTH);
|
int month = calendar.get(Calendar.MONTH);
|
int year = calendar.get(Calendar.YEAR);
|
|
System.err.println(year);
|
System.err.println(month);
|
System.err.println(week);
|
|
}
|
|
|
|
}
|
