ฉันจะหลีกเลี่ยงการทำซ้ำรหัสเริ่มต้น hashmap ของ hashmap ได้อย่างไร

ลูกค้าทุกคนมีรหัสและใบแจ้งหนี้จำนวนมากที่มีวันที่เก็บไว้เป็น Hashmap ของลูกค้าโดย id ของ hashmap ของใบแจ้งหนี้ตามวันที่:

HashMap<LocalDateTime, Invoice> allInvoices = allInvoicesAllClients.get(id);

if(allInvoices!=null){
    allInvoices.put(date, invoice);      //<---REPEATED CODE
}else{
    allInvoices = new HashMap<>();
    allInvoices.put(date, invoice);      //<---REPEATED CODE
    allInvoicesAllClients.put(id, allInvoices);
}

โซลูชัน Java ดูเหมือนว่าจะใช้getOrDefault:

HashMap<LocalDateTime, Invoice> allInvoices = allInvoicesAllClients.getOrDefault(
    id,
    new HashMap<LocalDateTime, Invoice> (){{  put(date, invoice); }}
);

แต่ถ้ารับไม่เป็นโมฆะฉันยังต้องการใส่ (วันที่ใบแจ้งหนี้) เพื่อดำเนินการและเพิ่มข้อมูลไปที่ "allInvoicesAllClients" ดังนั้นดูเหมือนจะไม่ช่วยอะไรมาก

Map#computeIfAbsentนี่คือกรณีการใช้งานที่ดีเยี่ยมสำหรับ ตัวอย่างข้อมูลของคุณเทียบเท่ากับ:

allInvoicesAllClients.computeIfAbsent(id, key -> new HashMap<>()).put(date, invoice);

ถ้าidไม่ได้เป็นปัจจุบันเป็นกุญแจสำคัญในallInvoicesAllClientsแล้วมันจะสร้างแผนที่จากidไปอยู่ที่ใหม่และกลับไปใหม่HashMap HashMapถ้าเป็นปัจจุบันเป็นสำคัญแล้วก็จะกลับมาที่มีอยู่idHashMap

computeIfAbsentเป็นทางออกที่ดีสำหรับกรณีนี้โดยเฉพาะ โดยทั่วไปฉันต้องการทราบสิ่งต่อไปนี้เนื่องจากยังไม่มีใครพูดถึง:

hashmap "outer" เพียงเก็บการอ้างอิงถึง hashmap "inner" ดังนั้นคุณสามารถจัดลำดับการดำเนินการใหม่เพื่อหลีกเลี่ยงการทำซ้ำรหัส:

HashMap<LocalDateTime, Invoice> allInvoices = allInvoicesAllClients.get(id);

if (allInvoices == null) {           
    allInvoices = new HashMap<>();
    allInvoicesAllClients.put(id, allInvoices);
}

allInvoices.put(date, invoice);      // <--- no longer repeated

คุณไม่ควรใช้การเริ่มต้นแผนที่ "double brace"

{{  put(date, invoice); }}

ในกรณีนี้คุณควรใช้ computeIfAbsent

allInvoicesAllClients.computeIfAbsent(id, (k) -> new HashMap<>())
                     .put(date, allInvoices);

หากไม่มีแผนที่สำหรับ ID นี้คุณจะแทรกหนึ่งแผนที่ ผลลัพธ์จะเป็นแผนที่ที่มีอยู่หรือคำนวณ จากนั้นคุณสามารถputรายการในแผนที่นั้นพร้อมรับประกันว่าจะไม่เป็นโมฆะ

นี่เป็นคำตอบที่ยาวกว่าคำตอบอื่น ๆ แต่ฉันอ่านได้มากขึ้น:

if(!allInvoicesAllClients.containsKey(id))
    allInvoicesAllClients.put(id, new HashMap<LocalDateTime, Invoice>());

allInvoicesAllClients.get(id).put(date, invoice);

คุณกำลังทำสองสิ่งที่แตกต่างกันที่นี่: ตรวจสอบให้แน่ใจว่าHashMapมีอยู่และเพิ่มรายการใหม่

รหัสที่มีอยู่ทำให้แน่ใจว่าได้แทรกองค์ประกอบใหม่ก่อนที่จะลงทะเบียนแผนที่แฮช แต่ไม่จำเป็นเพราะHashMapไม่สนใจการสั่งซื้อที่นี่ ตัวแปรไม่เป็นเธรดที่ปลอดภัยดังนั้นคุณจะไม่สูญเสียอะไรเลย

ดังนั้นตามที่ @Hezi แนะนำคุณสามารถแยกสองขั้นตอนนี้ออกได้

สิ่งที่ผมจะทำคือการสร้าง Offload ของHashMapกับallInvoicesAllClientsวัตถุดังนั้นวิธีการที่ไม่สามารถกลับgetnull

นอกจากนี้ยังช่วยลดความเป็นไปได้สำหรับการแข่งขันระหว่างหัวข้อที่แยกต่างหากที่จะได้รับทั้งnullคำแนะนำจากgetนั้นตัดสินใจที่จะputใหม่HashMapกับรายการเดียว - สองputอาจจะทิ้งคนแรกที่สูญเสียInvoiceวัตถุ