JavaのMapでnull値をデフォルト値に置き換える方法

JavaのMapに含まれるnull値を、あらかじめ定義しておいたデフォルト値に一括で置き換える方法を紹介します。Stream APIを活用することで、簡潔かつ読みやすいコードでこの処理を実現できます。

サンプルコード

import java.util.*;
import java.util.stream.*;
public class Demo{
   public static <T, K> Map<K, T> null_vals(Map<K, T> my_map, T def_val){
      my_map = my_map.entrySet().stream().map(entry -> {
         if (entry.getValue() == null)
         entry.setValue(def_val);
         return entry;
      })
      .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
      return my_map;
   }
   public static void main(String[] args){
      Map<Integer, Integer> my_map = new HashMap<>();
      my_map.put(1, null);
      my_map.put(2, 56);
      my_map.put(3, null);
      my_map.put(4, 99);
      int defaultValue = 0;
      System.out.println("null値を含むマップ : "+ my_map);
      my_map = null_vals(my_map, defaultValue);
      System.out.println("null値を置き換えた後のマップ : " + my_map);
   }
}

実行結果

null値を含むマップ : {1=null, 2=56, 3=null, 4=99}
null値を置き換えた後のマップ : {1=0, 2=56, 3=0, 4=99}

コードの解説

Demo クラスには null_vals というジェネリックメソッドが定義されています。このメソッドは、引数として受け取ったMapの各エントリを走査し、値がnullである場合に、事前に指定されたデフォルト値へ置き換えます。

処理の流れは以下の通りです。

1. entrySet().stream() でMapのエントリに対するストリームを生成します。
2. map() の中で各エントリをチェックし、値がnullなら setValue() を使ってデフォルト値に置き換えます。
3. Collectors.toMap() を使って、処理済みのエントリを新しいMapとして収集します。

mainメソッドでは、HashMap のインスタンスを生成し、put メソッドでキーと値のペアを追加しています。この例では、キー1と3に対応する値がnullになっています。その後、null_vals メソッドを呼び出すことで、これらのnull値がすべてデフォルト値（0）に置き換えられ、最終的なMapが出力されます。

なお、Java 8以降で利用可能な getOrDefault() メソッドや merge() メソッドを使えば、値の取得時や更新時にnullを回避する別のアプローチも可能です。用途に応じて使い分けるとよいでしょう。