yixian
7 years ago
parent
ff80fbe56d
commit
bca940a8c5
Binary file not shown.
@ -0,0 +1,83 @@
|
||||
package au.com.royalpay.payment.manage.task;
|
||||
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.alibaba.fastjson.JSONArray;
|
||||
import com.alibaba.fastjson.JSONObject;
|
||||
import org.apache.commons.io.FileUtils;
|
||||
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
|
||||
import org.apache.poi.ss.usermodel.*;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.util.*;
|
||||
import java.util.function.BinaryOperator;
|
||||
import java.util.stream.Collectors;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
/**
|
||||
* Create by yixian at 2018-05-01 17:15
|
||||
*/
|
||||
public class FillExcelTask {
|
||||
|
||||
private File rpayCat = new File("C:\\Users\\yixian\\develop\\royalpayv2\\manage\\src\\main\\ui\\static\\data\\royalpayindustry.json");
|
||||
private File wxCat = new File("C:\\Users\\yixian\\develop\\royalpayv2\\manage\\src\\main\\resources\\wx_industry.json");
|
||||
|
||||
private Map<String, String> royalCatMap;
|
||||
private Map<String, String> wxCatMap;
|
||||
private File excel = new File("C:\\Users\\yixian\\Documents\\bd_merchants.xlsx");
|
||||
|
||||
@Before
|
||||
public void initCategory() throws IOException {
|
||||
String rpayCatStr = FileUtils.readFileToString(rpayCat, "utf-8");
|
||||
JSONArray rpayCatJson = JSON.parseArray(rpayCatStr);
|
||||
List<JSONObject> flatCollect = rpayCatJson.stream().map(item -> (JSONObject) item)
|
||||
.flatMap(item -> loadChildren(item).stream()).collect(Collectors.toList());
|
||||
royalCatMap = rpayCatJson.stream().map(item -> (JSONObject) item)
|
||||
.flatMap(item -> loadChildren(item).stream())
|
||||
.collect(Collectors.toMap(item -> item.getString("mccCode"), item -> item.getString("label"), (a, b) -> a));
|
||||
|
||||
String wxCatStr = FileUtils.readFileToString(wxCat, "utf-8");
|
||||
JSONObject wxCatJson = JSON.parseObject(wxCatStr);
|
||||
wxCatMap = wxCatJson.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> (String) entry.getValue()));
|
||||
|
||||
}
|
||||
|
||||
private List<JSONObject> loadChildren(JSONObject entity) {
|
||||
if (entity.containsKey("children") && !entity.getJSONArray("children").isEmpty()) {
|
||||
List<JSONObject> arr = new ArrayList<>(Arrays.asList(entity));
|
||||
arr.addAll(entity.getJSONArray("children").stream().map(item -> (JSONObject) item).flatMap(item -> loadChildren(item).stream()).collect(Collectors.toList()));
|
||||
return arr;
|
||||
}
|
||||
return Collections.singletonList(entity);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void replaceCat() throws IOException, InvalidFormatException {
|
||||
Workbook wb = WorkbookFactory.create(excel);
|
||||
Sheet sheet = wb.getSheetAt(0);
|
||||
for (int rn = 1; rn <= sheet.getLastRowNum(); rn++) {
|
||||
Row row = sheet.getRow(rn);
|
||||
Cell wxCatCell = row.getCell(2);
|
||||
Cell rpCatCell = row.getCell(3);
|
||||
fillCateLabel(wxCatCell, wxCatMap);
|
||||
fillCateLabel(rpCatCell, royalCatMap);
|
||||
}
|
||||
wb.write(new FileOutputStream("C:\\Users\\yixian\\Documents\\bd_merchants_final.xlsx"));
|
||||
}
|
||||
|
||||
private void fillCateLabel(Cell rpCatCell, Map<String, String> royalCatMap) {
|
||||
if (rpCatCell == null) {
|
||||
return;
|
||||
}
|
||||
if (rpCatCell.getCellType() == Cell.CELL_TYPE_NUMERIC) {
|
||||
String typeNum = (int) rpCatCell.getNumericCellValue() + "";
|
||||
if (royalCatMap.get(typeNum) != null) {
|
||||
rpCatCell.setCellType(Cell.CELL_TYPE_STRING);
|
||||
rpCatCell.setCellValue(royalCatMap.get(typeNum));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue