All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.twosigma.webtau.http.datanode.DataNodeBuilder Maven / Gradle / Ivy

There is a newer version: 1.20
Show newest version
/*
 * Copyright 2019 TWO SIGMA OPEN SOURCE, LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.twosigma.webtau.http.datanode;

import com.twosigma.webtau.data.traceable.TraceableValue;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

public class DataNodeBuilder {
    public static DataNode fromMap(DataNodeId id, Map map) {
        return fromValue(id, map);
    }

    public static DataNode fromList(DataNodeId id, List list) {
        return fromValue(id, list);
    }

    private static Map buildMapOfNodes(DataNodeId id, Map map) {
        Map result = new LinkedHashMap<>();
        for (Entry entry : map.entrySet()) {
            String key = entry.getKey();
            Object value = entry.getValue();

            result.put(key, fromValue(id.child(key), value));
        }

        return result;
    }

    private static List buildListOfNodes(DataNodeId id, List values) {
        List result = new ArrayList<>();
        int idx = 0;
        for (Object value : values) {
            result.add(fromValue(id.peer(idx), value));
            idx++;
        }

        return result;
    }

    @SuppressWarnings("unchecked")
    public static DataNode fromValue(DataNodeId id, Object value) {
        if (value instanceof Map) {
            return new StructuredDataNode(id, buildMapOfNodes(id, (Map)value));
        } else if (value instanceof List) {
            return new StructuredDataNode(id, buildListOfNodes(id, (List)value));
        } else {
            return new StructuredDataNode(id, new TraceableValue(value));
        }
    }
}