org.openlca.proto.io.input.UnitGroupImport Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of olca-proto-io Show documentation
The newest version!
package org.openlca.proto.io.input;

import java.util.Objects;

import org.openlca.core.database.IDatabase;
import org.openlca.core.io.EntityResolver;
import org.openlca.core.model.Category;
import org.openlca.core.model.Exchange;
import org.openlca.core.model.FlowProperty;
import org.openlca.core.model.ModelType;
import org.openlca.core.model.RootEntity;
import org.openlca.core.model.UnitGroup;
import org.openlca.proto.ProtoUnitGroup;
import org.openlca.util.Strings;

class UnitGroupImport implements EntityResolver {

  private final ProtoImport imp;

  UnitGroupImport(ProtoImport imp) {
    this.imp = imp;
  }

	public void importAll() {
		for (var refId : imp.reader.getIds(ModelType.UNIT_GROUP)) {
			get(UnitGroup.class, refId);
		}
	}

	@Override
	public IDatabase db() {
		return imp.db();
	}

	@Override
	@SuppressWarnings("unchecked")
	public  T get(Class type, String refId) {

		// first return null for default-flow property
		// to avoid cyclic calls
		if (Objects.equals(type, FlowProperty.class))
			return null;

		if (!Objects.equals(type, UnitGroup.class))
			return imp.get(type, refId);

		var item = imp.fetch(type, refId);
		if (item.isError() || item.isVisited())
			return item.entity();

		var proto = (ProtoBox) item.proto();

		// read / update a unit group
		UnitGroup group = null;
		if (item.isNew()) {
			group = imp.db().insert(proto.read(this));
		} else if (item.entity() instanceof UnitGroup g){
			proto.update(g, this);
			group = imp.db().update(g);
		}
		if (group == null)
			return null;
		imp.visited(group);

		// set a possible default flow property
		var propId = proto.message().getDefaultFlowProperty().getId();
		if (Strings.notEmpty(propId)) {
			var prop = imp.get(FlowProperty.class, propId);
			if (prop != null) {
				group.defaultFlowProperty = prop;
				group = imp.db().update(group);
				imp.visited(group);
			}
		}

		return (T) group;
	}

	@Override
	public Category getCategory(ModelType type, String path) {
		return imp.getCategory(type, path);
	}

	@Override
	public void resolveProvider(String providerId, Exchange exchange) {
		imp.resolveProvider(providerId, exchange);
	}
}