Java의 구조 문서에서 경로 배열을 사용하여 값을 찾는 방법

때로는 JSON, YAML과 같은 구조적 문서에서 값을 찾아야 합니다. 또한 문서의 경로로 값을 찾아야 할 때가 있습니다.

구조 문서는 파서 라이브러리를 사용하여 Map-List 객체로 쉽게 변환할 수 있습니다. 그 때문에 키와 인덱스 배열을 사용하여 문서의 값을 얻을 수 있습니다.

하기 위해서,

먼저 JSON 또는 YAML을 Java Map-List 구조로 구문 분석할 수 있는 구문 분석기 라이브러리가 필요합니다. 해당 라이브러리는 Maven Central과 같은 인터넷 라이브러리 저장소에서 쉽게 찾을 수 있습니다.

이 게시물에서는 JSON 문서를 대상으로 한다고 가정합니다.
아래에 있어요

{
  "id": "0001",
  "type": "donut",
  "name": "Cake",
  "ppu": 0.55,
  "batters": {
    "aaa": {
      "bbb": {
        "ccc": {
          "value": 123.0
        }
      }
    },
    "batter": [
      {
        "id": "1001",
        "type": "Regular"
      },
      {
        "id": "1002",
        "type": "Chocolate"
      },
      {
        "id": "1003",
        "type": "Blueberry"
      },
      {
        "id": "1004",
        "type": "Devil\u0027s Food"
      }
    ]
  },

재귀 알고리즘을 사용하여 값을 찾고 경로가 값을 찾을 수 없거나 경로가 존재하지 않으면 null을 반환합니다.

완성된 코드는 아래에 있습니다.

    /**
     * To find value with path array on structural data
     * 
     * @param obj
     * @param keys
     * @return
     */
    public static Object findValue(Object obj, Object[] keys) {
        if (obj instanceof List) {
            List list = (List) obj;
            if (keys.length == 1) {
                if (keys[0] instanceof Integer && list.size() > (int) keys[0]) {
                    return list.get((int) keys[0]);
                }
            } else if (keys.length > 1 && list.size() > 0) {
                int idx = (int)keys[0];
                if(idx < list.size()) {
                    return findValue(list.get((int) keys[0]), Arrays.copyOfRange(keys, 1, keys.length));
                }
            }
        } else if (obj instanceof Map) {
            Map map = (Map) obj;
            if (keys.length == 1 && map.containsKey(keys[0])) {
                return map.get(keys[0]);
            } else if (keys.length > 1 && map.containsKey(keys[0])) {
                if(keys.length > 0) {
                    return findValue(map.get(keys[0]), Arrays.copyOfRange(keys, 1, keys.length));
                }
            }
        }
        return null;
    }

자, 이제 테스트를 시작하겠습니다

위의 JSON 문서는 'sample.json'이라는 파일에 저장합니다.
아래 코드는 'sample.json'을 읽고 결과는 아래의 이미지입니다.

Gson gson = new Gson();
Map<String, Object> map = gson.fromJson(new FileReader("simple.json"), Map.class);
Object[] keys = new Object[] {"batters", "batter", 3, "type"};

Object obj = findValue(map, keys);
System.out.println(obj);