|
@@ -1,3 +1,4 @@
|
|
|
+import jnr.ffi.annotations.In;
|
|
|
import org.junit.Test;
|
|
|
|
|
|
import java.util.*;
|
|
@@ -70,6 +71,16 @@ public class RecommendationSystemTest {
|
|
|
public void testByExcel() {
|
|
|
LinkedList<Map<String, Object>> list = ExcelUtil.readExcel("/Users/alvin/Downloads/授用信各资方家数表.xlsx", "用信家数明细表", 0, 0);
|
|
|
|
|
|
+ HashMap<String, Object> olds = new HashMap<>();
|
|
|
+ for (Map<String, Object> map : list) {
|
|
|
+ for (String s : map.keySet()) {
|
|
|
+ if (map.get(s) != null && !"企业名称".equalsIgnoreCase(s)) {
|
|
|
+ olds.put(map.get("企业名称") + s, (int) Double.parseDouble(map.get(s) + ""));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
HashSet<String> ignore = new HashSet<>();
|
|
|
ignore.add("企业名称");
|
|
|
ignore.add("总计");
|
|
@@ -99,9 +110,16 @@ public class RecommendationSystemTest {
|
|
|
Map<String, Double> recommendations = rs.recommendItems(companyName);
|
|
|
recommendations.entrySet().removeIf(entry -> entry.getValue() == 0.0);
|
|
|
|
|
|
+
|
|
|
HashMap<String, Object> map = new HashMap<>();
|
|
|
- map.put("企业名称", companyName);
|
|
|
map.putAll(recommendations);
|
|
|
+
|
|
|
+ for (String s : map.keySet()) {
|
|
|
+ if (olds.containsKey(companyName + s)) {
|
|
|
+ map.put(s, "已有业务: " + olds.get(companyName + s));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ map.put("企业名称", companyName);
|
|
|
outRes.add(map);
|
|
|
}
|
|
|
|