• Home
• com.github.ddth
• ddth-commons-core
• 0.9.3
• source code
• DPathUtils.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.github.ddth.commons.utils.DPathUtils Maven / Gradle / Ivy

package com.github.ddth.commons.utils;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ContainerNode;
import com.fasterxml.jackson.databind.node.MissingNode;
import com.fasterxml.jackson.databind.node.NullNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.fasterxml.jackson.databind.node.POJONode;

/**
 * Utility to access data from a hierarchy structure.
 * 
 * 

 * Notation:
 * 
 * 
 * 

 * 
. (the dot character): path separator
 * 
name: access a map's attribute specified by
 * name
 * 
[i]: access i'th element of a list/array (0-based)
 * 
 * Example: employees.[1].first_name. The dot right before {@code []} can be omitted:
 * employees[1].first_name.
 * 
 * 

 * Sample usage: assuming you have the following data structure:
 * 
 * 
 * 
 * Map<String, Object> company = new HashMap<String, Object>();
 * company.put("name", "Monster Corp.");
 * company.put("year", 2003);
 * 
 * List<Map<String, Object>> employees = new ArrayList<Map<String, Object>>();
 * company.put("employees", employees);
 * 
 * Map<String, Object> employee1 = new HashMap<String, Object>();
 * employee1.put("first_name", "Mike");
 * employee1.put("last_name", "Wazowski");
 * employee1.put("email", "[email protected]");
 * employee1.put("age", 29);
 * employees.add(employee1);
 * 
 * Map<String, Object> employee2 = new HashMap<String, Object>();
 * employee2.put("first_name", "Sulley");
 * employee2.put("last_name", "Sullivan");
 * employee2.put("email", "[email protected]");
 * employee2.put("age", 30);
 * employees.add(employee2);
 * 
 * 
 * 

 * You can access company's attributes:
 * 
 * 
 * 
 * String companyName = DPathUtils.getValue(company, "name", String.class);
 * // got string "Monster Corp."
 * 
 * Integer companyYear = DPathUtils.getValue(company, "year", Integer.class);
 * // got integer 2003
 * 
 * 
 * 

 * You can access two employees:
 * 
 * 
 * 
 * Object user1 = DPathUtils.getValue(company, "employees.[0]");
 * Map<String, Object> user2 = DPathUtils.getValue(company, "employees.[1]", Map.class);
 * 
 * 
 * 

 * Or, employee's attributes:
 * 
 * 
 * 
 * String firstName1 = DPathUtils.getValue(company, "employees.[0].first_name", String.class);
 * Object email2 = DPathUtils.getValue(company, "employees.[1].email");
 * Long age2 = DPathUtils.getValue(company, "employees.[1].age", Long.class);
 * 
 * 
 * @author Thanh Nguyen 
 * @since 0.1.0
 */
public class DPathUtils {

    private final static Pattern PATTERN_INDEX = Pattern.compile("^\\[(.*?)\\]$");
    private final static Pattern PATTERN_END_INDEX = Pattern.compile("^(.*)(\\[.*?\\])$");
    public final static char PATH_SEPARATOR = '.';

    /**
     * Split {@code DPath} string to tokens.
     * 
     * 

     * 
{@code "a.b.c.[i].d"} or {@code "a.b.c[i].d"} will be split into
     * {@code ["a","b","c","[i]","d"]}
     * 
{@code "a.b.c.[i].[j].d"} or {@code "a.b.c[i].[j].d"} or
     * {@code "a.b.c[i][j].d"} or {@code "a.b.c.[i][j].d"} will be split into
     * {@code ["a","b","c","[i]","[j]","d"]}
     * 
     * 
     * @param dpath
     * @return
     * @since 0.6.1
     */
    @SuppressWarnings("serial")
    public final static String[] splitDpath(String dpath) {
        String[] tokens = StringUtils.split(dpath, PATH_SEPARATOR);
        List tokenList = new ArrayList() {
            {
                for (String token : tokens) {
                    List temp = new ArrayList() {
                        {
                            String _token = token;
                            Matcher m = PATTERN_END_INDEX.matcher(_token);
                            while (m.matches()) {
                                _token = m.group(1);
                                add(0, m.group(2));
                                m = PATTERN_END_INDEX.matcher(_token);
                            }
                            if (!StringUtils.isBlank(_token)) {
                                add(0, _token);
                            }
                        }
                    };
                    addAll(temp);
                }
            }
        };
        return tokenList.toArray(ArrayUtils.EMPTY_STRING_ARRAY);
    }

    static Object extractValue(POJONode pojoNode) {
        Object value = pojoNode;
        while (value instanceof POJONode) {
            value = ((POJONode) value).getPojo();
        }
        return value;
    }

    private static Object extractValue(Object target, String index) {
        if (target == null) {
            return null;
        }
        if (target instanceof JsonNode) {
            return extractValue((JsonNode) target, index);
        }
        Matcher m = PATTERN_INDEX.matcher(index);
        if (m.matches()) {
            try {
                int i = Integer.parseInt(m.group(1));
                if (target instanceof Object[]) {
                    return ((Object[]) target)[i];
                }
                if (target instanceof List) {
                    return ((List) target).get(i);
                }
                throw new IllegalArgumentException("Expect an array or list for index [" + index
                        + "] but received [" + target.getClass() + "] instead.");
            } catch (IndexOutOfBoundsException e) {
                // rethrow for now
                throw e;
            } catch (NumberFormatException e) {
                throw new IllegalArgumentException("Invalid index value: " + index, e);
            }
        }
        if (target instanceof Map) {
            return ((Map) target).get(index);
        }
        throw new IllegalArgumentException(
                "Unsupported type [" + target.getClass() + "] or invalid index [" + index + "]");
    }

    private static Object extractValue(JsonNode node, String index) {
        if (node == null || node instanceof NullNode || node instanceof MissingNode) {
            return null;
        }
        if (node instanceof POJONode) {
            return extractValue(extractValue((POJONode) node), index);
        }
        Matcher m = PATTERN_INDEX.matcher(index);
        if (m.matches()) {
            try {
                int i = Integer.parseInt(m.group(1));
                if (node instanceof ArrayNode) {
                    if (i >= 0 && i < node.size()) {
                        return node.get(i);
                    } else {
                        throw new IndexOutOfBoundsException(String.valueOf(i));
                    }
                }
                throw new IllegalArgumentException("Expect an ArrayNode for index [" + index
                        + "] but received [" + node.getClass() + "] instead.");
            } catch (NumberFormatException e) {
                throw new IllegalArgumentException("Invalid index value: " + index, e);
            }
        }
        if (node instanceof ObjectNode) {
            return node.get(index);
        }
        throw new IllegalArgumentException(
                "Unsupported type [" + node.getClass() + "] or invalid index [" + index + "]");
    }

    /**
     * @param dPath
     * @param cursor
     * @param index
     * @since 0.9.2
     */
    @SuppressWarnings("unchecked")
    private static void deleteFieldValue(String dPath, Object cursor, String index) {
        if (cursor instanceof POJONode) {
            cursor = extractValue((POJONode) cursor);
        }
        if (cursor instanceof Map) {
            Map temp = (Map) cursor;
            temp.remove(index);
        } else if (cursor instanceof ObjectNode) {
            ObjectNode temp = (ObjectNode) cursor;
            temp.remove(index);
        } else {
            throw new IllegalArgumentException("Target object of type [" + cursor.getClass()
                    + "] is not writable with path [" + dPath + "]!");
        }
    }

    /**
     * @param dPath
     * @param cursor
     * @param index
     * @since 0.9.2
     */
    @SuppressWarnings("unchecked")
    private static void deleteFieldValue(String dPath, Object cursor, int index) {
        if (cursor instanceof POJONode) {
            cursor = extractValue((POJONode) cursor);
        }
        if (cursor instanceof List) {
            List