如何比较两个HashMap <String,List <String >>
我想比较从一个Excel文件(键是第1列,值是第2列)读取的数据,正在被放入一个HashMap从SQL查询获得的数据。 起初我使用HashMap <String,String>,因为我只需要比较<key,value>对,但现在我必须比较<key,list>,而且我有点卡住了。 这是我读取xls文件的代码:
public class ReadExcel { HashMap<String, List<String>> result = new HashMap<String, List<String>>(); public HashMap<String, List<String>> process() { try { result.clear(); FileInputStream file = new FileInputStream(new File("C:/some.xlsx")); //Create Workbook instance holding reference to .xlsx file XSSFWorkbook workbook = new XSSFWorkbook(file); //Get first/desired sheet from the workbook XSSFSheet sheet = workbook.getSheetAt(0); //Iterate through each rows one by one Iterator<Row> rowIterator = sheet.iterator(); while (rowIterator.hasNext()) { List<String> xlsList = new ArrayList<String>(); Row row = rowIterator.next(); Cell cell = row.getCell(1); Cell cell2 = row.getCell(2); String key =""; String value=""; xlsList.clear(); switch (cell.getCellType()) { case Cell.CELL_TYPE_NUMERIC: key = getStringCellValue(cell); value = getNumericCellValue(cell2); break; case Cell.CELL_TYPE_STRING: key = getStringCellValue(cell); value = getStringCellValue(cell2); break; } xlsList.add(value); result.put(key, xlsList); } } }
例如在我的excel文件中
row 1: column 1 = car, column 2 = blue row 2: column 1 = car, column 2 = yellow.
当我运行excel阅读器时,它将“蓝色,黄色”值放在HashMap的“car”键下。 当我有例如:
row 1: column 1 = car, column 2 = blue row 2: column 1 = car, column 2 = yellow row 3: column 1 = year, column 2 = 1990 row 4: column 1 = year, column 2 = 1999
它只显示: car=[yellow], year=[1999]
。 它只需要最后的值,如果不是重复的键,这工作得很好。
第一个问题:我怎样才能做得更好? 如果我有第1行相同的密钥只取一次密钥,并存储第2行的所有值,如果第1行的密钥是相同的?
这里是我从SQL数据库中提取数据的代码:
public class DB { HashMap<String, List<String>> result = new HashMap<String, List<String>>(); public HashMap<String, List<String>> process() { result.clear(); Connection conn = null; Statement stmt = null; List<String> carColour = new ArrayList<String>(); try { Class.forName("oracle.jdbc.driver.OracleDriver"); conn = DriverManager.getConnection(DB_URL, USER, PASS); stmt = conn.createStatement(); String sql1 = "SOME SQL SELECT STATEMENT THAT RETURNS 2 or more lines, by that i mean "car" has 2 values or more"; ResultSet rs = stmt.executeQuery(sql1); while(rs.next()){ carColour.add(rs.getString("colour")); // i select the column "colour" result.put("car", carColour); // i put "car" as key, and "blue" and "yellow" as values ................................... } } catch... }
这个数据库的代码工作正常,但如果我想提取更多的列,如“颜色”或“年”,我必须为他们所有创build列表,如果我得到像20列提取,可能是非常时间耗时。
第二个问题:我怎样才能做到这一点,而不是创build20个列表(使用相同的列表也许和list.clear();它?)(这取决于,因为如果我有30列,我只需要20,我可以getString (“列”)全部删除我不想要的,但是如何?
这里是比较的代码:
final Map<String, Boolean> comparisonResult = compareEntries(dbResult, xlsResult); for(final Entry<String, Boolean> entry : comparisonResult.entrySet()){ if (entry.getValue() == false){ System.out.println("------------------------------------------------------------------------"); System.out.println("| Comparison FAILED | Value not matching! Column name --> " + entry.getKey() + " |"); } } System.out.println("------------------------------------------------------------------------"); System.out.println("DB consistency check finished.");
………………………………………….. …………………………
public static <K extends Comparable<? super K>, V> Map<K, Boolean> compareEntries(final Map<K, V> dbResult, final Map<K, V> xlsResult){ final Collection<K> allKeys = new HashSet<K>(); allKeys.addAll(dbResult.keySet()); allKeys.addAll(xlsResult.keySet()); final Map<K, Boolean> result = new TreeMap<K, Boolean>(); for(final K key : allKeys){ result.put(key, dbResult.containsKey(key) == xlsResult.containsKey(key) && Boolean.valueOf(equal(dbResult.get(key), xlsResult.get(key)))); } return result; } private static boolean equal(final Object obj1, final Object obj2){ return obj1 == obj2 || (obj1 != null && obj1.equals(obj2)); }
最后一个问题:我如何改进代码来比较两个HashMap(string,列表<String>),或者我怎样才能更好地一步一步做到这一点? 谢谢!
您的ReadExcel
阅读循环有一个严重的问题:而不是重复使用一个List
,这是已经映射的一个关键,你总是创build一个新的,因此你会最终列表只有一个单一(最后一个)值。
这是你如何解决这个问题(使用Java 8):
// ... while (rowIterator.hasNext()) { Row row = rowIterator.next(); Cell keyCell = row.getCell(1); Cell valCell = row.getCell(2); String key = getStringCellValue(keyCell); switch (cell.getCellType()) { case Cell.CELL_TYPE_NUMERIC: value = getNumericCellValue(valCell); break; case Cell.CELL_TYPE_STRING: value = getStringCellValue(valCell); break; } } // this line only compiles since Java 8 result.computeIfAbsent(key, k -> new HashSet<Object>()).add(value); }
如果您尚未使用Java 8,请快速下载它,或者使用以下代码replace上面的最后一行:
Set<Object> list = result.get(key); if (list == null) result.put(key, list = new HashSet<Object>()); list.add(value);
Map
对象的比较相当简单 – 你可以直接使用equals
比较它们:
if (map1.equals(map2)) { // both maps are equal! } else { // maps are NOT equal! }
当然,这只是产生false
。 如果你想知道细节(哪些条目不同),你需要自己做。