D-Net

package eu.dnetlib.data.mapreduce.hbase.index.config;

import java.io.StringReader;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;

import com.google.common.collect.Maps;

import com.google.common.collect.Sets;

import com.google.gson.Gson;

import com.typesafe.config.Config;

import com.typesafe.config.ConfigFactory;

import com.typesafe.config.ConfigObject;

import eu.dnetlib.data.mapreduce.OptionalConfig;

import eu.dnetlib.data.mapreduce.util.RelDescriptor;

import eu.dnetlib.data.proto.TypeProtos.Type;

import org.apache.commons.collections.CollectionUtils;

public class IndexConfig extends OptionalConfig {

	public static IndexConfig load(final String s) {

		return new IndexConfig(ConfigFactory.parseReader(new StringReader(s)));

	}

	public IndexConfig(final Config config) {

		super(config);

	}

	@SuppressWarnings("unchecked")

	public EntityConfigTable getConfigMap() {

		final ConfigObject modelMap = getConfig().getObject("index.conf");

		final Map<String, EntityConfig> map = Maps.transformValues(modelMap, value -> {

			final HashMap<RelDescriptor, LinkDescriptor> links = Maps.newHashMap();

			for (final Object o : (List<?>) ((Map<String, ?>) value.unwrapped()).get("links")) {

				final Map<String, ?> map1 = (Map<String, ?>) o;

				final RelDescriptor relDescriptor = new RelDescriptor((String) map1.get("relType"));

				final Set<String> fields = Sets.newHashSet((List<String>) map1.get("fields"));

				final Type target = Type.valueOf((String) map1.get("targetEntity"));

				final Boolean isChild = ((String) map1.get("expandAs")).equals("child");

				final Boolean symmetric = (Boolean) map1.get("symmetric");

				final LinkDescriptor ld = new LinkDescriptor(relDescriptor, target, isChild, symmetric, fields);

				if (map1.containsKey("max")) {

					ld.setMax((Integer) map1.get("max"));

				}

				links.put(relDescriptor, ld);

			}

			final List<String> includeFields = (List<String>) ((Map<String, ?>) value.unwrapped()).get("includeFields");

			final List<String> excludeFields = (List<String>) ((Map<String, ?>) value.unwrapped()).get("excludeFields");

			if(CollectionUtils.isNotEmpty(includeFields) & CollectionUtils.isNotEmpty(excludeFields)){

				throw new RuntimeException("Cannot create the index configuration: includeFields and excludeFields are both not null");

			}

			return new EntityConfig((Boolean) ((Map<String, ?>) value.unwrapped()).get("dups"), links, includeFields, excludeFields);

		});

		final EntityConfigTable res = new EntityConfigTable();

		for (final String key : map.keySet()) {

			res.put(Type.valueOf(key), map.get(key));

		}

		return res;

	}

	@Override

	public String toString() {

		return new Gson().toJson(this);

	}

}