private static Procurement getProcurement(List<Procurement> procurementList, int productId) {
for (Procurement procurement : procurementList) {
if (procurement.getProductId() == productId) {
return procurement;
}
}
return null;
}
public static void exportProducts(
String fileName,
List<Product> productList,
Map<Type, List<Category>> typeMap,
List<Procurement> procurementList) {
logger.info("Export product to excel");
ExcelGenerator generator = new ExcelGenerator(fileName);
List<String[]> headerList = new ArrayList<>();
List<String> sheetNames = new ArrayList<>();
for (Type type : typeMap.keySet()) {
sheetNames.add(type.getType());
headerList.add(new String[] {"名字", "描述", "大类", "小类", "采购价", "系数", "销售价", "单位"});
}
List<List<String[]>> contents = new ArrayList<>();
for (Type type : typeMap.keySet()) {
List<Category> categoryList = typeMap.get(type);
List<String[]> content = new ArrayList<>();
for (Category category : categoryList) {
List<Product> subProductList = getProductListByCategory(productList, category);
for (Product product : subProductList) {
Procurement procurement = getProcurement(procurementList, product.getId());
double procprice = product.getPrice();
double procindex = 1.0;
if (procurement != null) {
procprice = procurement.getProcprice();
procindex = procurement.getProcindex();
}
content.add(
new String[] {
product.getName(),
product.getDescription(),
product.getType().getType(),
product.getCategory().getCategory(),
String.valueOf(procprice),
String.valueOf(procindex),
product.getPrice().toString(),
product.getUnit()
});
}
}
contents.add(content);
}
try {
generator.generate(sheetNames, headerList, contents);
} catch (IOException e) {
e.printStackTrace();
return;
}
}