guoxin
1 year ago
parent
3f2007cf9f
commit
47f9f95818
@ -0,0 +1,47 @@ |
|||||||
|
package com.keyware.shandan.bianmu.export; |
||||||
|
|
||||||
|
import com.keyware.shandan.common.util.StringUtils; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Optional; |
||||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||||
|
|
||||||
|
public class ExportCache { |
||||||
|
private final static ConcurrentHashMap<String, Map<String, ExportProgress>> cache = new ConcurrentHashMap<>(); |
||||||
|
|
||||||
|
public static void startExport(ExportProgress progress) { |
||||||
|
Map<String, ExportProgress> userCache = cache.get(progress.getUserId()); |
||||||
|
if (userCache == null) { |
||||||
|
userCache = new HashMap<>(); |
||||||
|
} |
||||||
|
userCache.put(progress.getExportId(), progress); |
||||||
|
cache.put(progress.getUserId(), userCache); |
||||||
|
new Thread(progress).start(); |
||||||
|
} |
||||||
|
|
||||||
|
public static <T extends ExportProgress> T getCache(String userId, String exportId) { |
||||||
|
Map<String, ExportProgress> userCache = cache.get(userId); |
||||||
|
if (userCache == null) { |
||||||
|
return null; |
||||||
|
} |
||||||
|
if (StringUtils.hasText(exportId)) { |
||||||
|
return (T) userCache.get(exportId); |
||||||
|
} |
||||||
|
Optional<ExportProgress> optional = userCache.values().stream().findFirst(); |
||||||
|
if (optional.isPresent()) { |
||||||
|
return (T) optional.get(); |
||||||
|
} |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
public static <T extends ExportProgress> T popCache(String userId, String exportId) { |
||||||
|
Map<String, ExportProgress> userCache = cache.get(userId); |
||||||
|
if (userCache == null) { |
||||||
|
return null; |
||||||
|
} else { |
||||||
|
ExportProgress progress = userCache.remove(exportId); |
||||||
|
return (T) progress; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,90 @@ |
|||||||
|
package com.keyware.shandan.bianmu.export; |
||||||
|
|
||||||
|
import cn.hutool.core.io.FileUtil; |
||||||
|
import com.keyware.shandan.common.util.UUIDUtil; |
||||||
|
import com.keyware.shandan.frame.config.component.AppContext; |
||||||
|
import com.keyware.shandan.frame.properties.CustomProperties; |
||||||
|
import lombok.Data; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.io.Serializable; |
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
|
||||||
|
@Data |
||||||
|
public abstract class ExportProgress implements Runnable, Serializable { |
||||||
|
|
||||||
|
private String exportId = UUIDUtil.getUUID(); |
||||||
|
private String userId; |
||||||
|
private String title; |
||||||
|
private Integer currentStep = 0; |
||||||
|
protected final String tempDir; |
||||||
|
|
||||||
|
protected boolean error = false; |
||||||
|
|
||||||
|
public ExportProgress(String userId) { |
||||||
|
this.userId = userId; |
||||||
|
this.title = "正在查询需要导出的数据"; |
||||||
|
CustomProperties customProperties = AppContext.getContext().getBean(CustomProperties.class); |
||||||
|
tempDir = customProperties.getFileStorage().getTempPath(); |
||||||
|
} |
||||||
|
|
||||||
|
public abstract void run(); |
||||||
|
|
||||||
|
/** |
||||||
|
* 进度总步数 |
||||||
|
*/ |
||||||
|
public abstract long getStepTotal(); |
||||||
|
|
||||||
|
public abstract String getDownloadPath(); |
||||||
|
|
||||||
|
/** |
||||||
|
* 偏移量自增 |
||||||
|
*/ |
||||||
|
public void autoAddStep() { |
||||||
|
this.currentStep++; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 获取百分比 |
||||||
|
* |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
public int getPercentage() { |
||||||
|
return Math.round(currentStep / (float) getStepTotal() * 100); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 判断是否完成 |
||||||
|
* |
||||||
|
* @return |
||||||
|
*/ |
||||||
|
public boolean getIsDone() { |
||||||
|
return currentStep == getStepTotal(); |
||||||
|
} |
||||||
|
|
||||||
|
protected void setDone() { |
||||||
|
currentStep = getCurrentStep(); |
||||||
|
} |
||||||
|
|
||||||
|
public String getMsg() { |
||||||
|
return "(" + getPercentage() + "%) " + this.title; |
||||||
|
} |
||||||
|
|
||||||
|
public Map<String, Object> getProgress() { |
||||||
|
Map<String, Object> map = new HashMap<>(); |
||||||
|
map.put("exportId", exportId); |
||||||
|
map.put("title", title); |
||||||
|
map.put("msg", getMsg()); |
||||||
|
map.put("isDone", getIsDone()); |
||||||
|
map.put("isError", error); |
||||||
|
return map; |
||||||
|
} |
||||||
|
|
||||||
|
public void clean() { |
||||||
|
File file = new File(getDownloadPath()); |
||||||
|
if (file.exists()) { |
||||||
|
FileUtil.del(file); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,119 @@ |
|||||||
|
package com.keyware.shandan.bianmu.export; |
||||||
|
|
||||||
|
import cn.hutool.core.collection.CollUtil; |
||||||
|
import cn.hutool.poi.excel.BigExcelWriter; |
||||||
|
import cn.hutool.poi.excel.ExcelUtil; |
||||||
|
import cn.hutool.poi.excel.ExcelWriter; |
||||||
|
import com.alibaba.fastjson.JSONArray; |
||||||
|
import com.alibaba.fastjson.JSONObject; |
||||||
|
import com.keyware.shandan.bianmu.entity.MetadataBasicVo; |
||||||
|
import com.keyware.shandan.bianmu.service.MetadataService; |
||||||
|
import com.keyware.shandan.common.util.StreamUtil; |
||||||
|
import com.keyware.shandan.common.util.StringUtils; |
||||||
|
import com.keyware.shandan.frame.config.component.AppContext; |
||||||
|
import org.apache.ibatis.session.ResultContext; |
||||||
|
import org.apache.ibatis.session.ResultHandler; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.List; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.function.Consumer; |
||||||
|
import java.util.stream.Collectors; |
||||||
|
|
||||||
|
public class MetaTableExport extends ExportProgress implements ResultHandler<HashMap<String, Object>> { |
||||||
|
private final MetadataBasicVo metadata; |
||||||
|
private Consumer<Object> queryFunc; |
||||||
|
private final String fileName; |
||||||
|
private JSONArray colsArray; |
||||||
|
private final MetadataService metadataService; |
||||||
|
|
||||||
|
// 所有行的集合
|
||||||
|
private List<List<String>> rowList = new ArrayList<>(); |
||||||
|
// 字段列名集合
|
||||||
|
private List<String> colNameList = new ArrayList<>(); |
||||||
|
private int dataTotal; |
||||||
|
|
||||||
|
public MetaTableExport(String userId, MetadataBasicVo metadata) { |
||||||
|
super(userId); |
||||||
|
this.metadataService = AppContext.getContext().getBean(MetadataService.class); |
||||||
|
this.metadata = metadata; |
||||||
|
String name = metadata.getMetadataComment(); |
||||||
|
if (!StringUtils.hasText(name)) { |
||||||
|
name = metadata.getMetadataName(); |
||||||
|
} |
||||||
|
fileName = tempDir + File.separator + name + ".xlsx"; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void run() { |
||||||
|
this.setTitle("正在读取数据表信息"); |
||||||
|
colsArray = metadataService.getColumns(metadata.getId()); |
||||||
|
|
||||||
|
// 添加第一行，字段列名注释集合，当做表格的第一行
|
||||||
|
rowList.add(colsArray.stream().map((json) -> { |
||||||
|
JSONObject field = (JSONObject) json; |
||||||
|
String comment = field.getString("comment"); |
||||||
|
String colName = field.getString("columnName"); |
||||||
|
colNameList.add(colName); |
||||||
|
return colName + (StringUtils.hasText(comment) ? "[" + comment + "]" : ""); |
||||||
|
}).collect(Collectors.toList())); |
||||||
|
|
||||||
|
this.setTitle("正在读取数据到临时文件"); |
||||||
|
|
||||||
|
// 执行数据查询
|
||||||
|
try { |
||||||
|
queryFunc.accept(null); |
||||||
|
|
||||||
|
File file = new File(fileName); |
||||||
|
if(file.exists()){ |
||||||
|
file.delete(); |
||||||
|
} |
||||||
|
List<List<String>> rows = CollUtil.newArrayList(rowList); |
||||||
|
//通过工具类创建writer
|
||||||
|
ExcelWriter writer = ExcelUtil.getWriter(file); |
||||||
|
//一次性写出内容
|
||||||
|
writer.write(rows); |
||||||
|
//关闭writer，释放内存
|
||||||
|
writer.close(); |
||||||
|
this.setTitle("数据文件准备完毕，开始下载"); |
||||||
|
this.setDone(); |
||||||
|
} catch (Exception e) { |
||||||
|
this.error = true; |
||||||
|
this.setTitle("准备数据时出现异常"); |
||||||
|
this.clean(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public long getStepTotal() { |
||||||
|
return this.dataTotal; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String getDownloadPath() { |
||||||
|
return this.fileName; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void handleResult(ResultContext<? extends HashMap<String, Object>> context) { |
||||||
|
Map<String, Object> data = context.getResultObject(); |
||||||
|
// 遍历字段列名，获取对应数据并拼接成列的集合
|
||||||
|
List<String> cells = colNameList.stream().map(col -> { |
||||||
|
Object value = data.get(col); |
||||||
|
value = value == null ? "" : value; |
||||||
|
return String.valueOf(value); |
||||||
|
}).collect(Collectors.toList()); |
||||||
|
rowList.add(cells); |
||||||
|
this.autoAddStep(); |
||||||
|
} |
||||||
|
|
||||||
|
public void queryData(Consumer<Object> query) { |
||||||
|
this.queryFunc = query; |
||||||
|
} |
||||||
|
|
||||||
|
public void setDataCount(int count) { |
||||||
|
this.dataTotal = count; |
||||||
|
} |
||||||
|
} |
||||||
Reference in new issue