Java tutorial
/* Copyright (c) 2014 Boundless and others. * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Distribution License v1.0 * which accompanies this distribution, and is available at * https://www.eclipse.org/org/documents/edl-v10.html * * Contributors: * Gabriel Roldan (Boundless) - initial implementation */ package org.locationtech.geogig.rest.repository; import static org.locationtech.geogig.rest.repository.RESTUtils.getGeogig; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.Reader; import java.math.BigDecimal; import java.math.BigInteger; import java.text.SimpleDateFormat; import java.util.Date; import java.util.Map.Entry; import java.util.UUID; import org.geotools.feature.simple.SimpleFeatureBuilder; import org.geotools.geometry.jts.WKTReader2; import org.locationtech.geogig.model.FieldType; import org.locationtech.geogig.model.NodeRef; import org.locationtech.geogig.model.ObjectId; import org.locationtech.geogig.model.RevCommit; import org.locationtech.geogig.model.RevFeature; import org.locationtech.geogig.model.RevFeatureBuilder; import org.locationtech.geogig.model.RevFeatureType; import org.locationtech.geogig.model.RevObject; import org.locationtech.geogig.model.RevTree; import org.locationtech.geogig.plumbing.FindTreeChild; import org.locationtech.geogig.plumbing.RevObjectParse; import org.locationtech.geogig.repository.Repository; import org.locationtech.geogig.rest.RestletException; import org.locationtech.geogig.web.api.CommandSpecException; import org.opengis.feature.simple.SimpleFeature; import org.opengis.feature.simple.SimpleFeatureType; import org.opengis.feature.type.PropertyDescriptor; import org.restlet.data.MediaType; import org.restlet.data.Status; import org.restlet.resource.Representation; import org.restlet.resource.Resource; import org.restlet.resource.StringRepresentation; import com.google.common.base.Optional; import com.google.common.base.Preconditions; import com.google.common.collect.ImmutableList; import com.google.common.io.Closeables; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import com.google.gson.JsonPrimitive; import com.vividsolutions.jts.geom.Geometry; import com.vividsolutions.jts.geom.GeometryCollection; import com.vividsolutions.jts.geom.LineString; import com.vividsolutions.jts.geom.MultiLineString; import com.vividsolutions.jts.geom.MultiPoint; import com.vividsolutions.jts.geom.MultiPolygon; import com.vividsolutions.jts.geom.Point; import com.vividsolutions.jts.geom.Polygon; /** * */ public class MergeFeatureResource extends Resource { @Override public boolean allowPost() { return true; } private Optional<NodeRef> parseID(ObjectId commitId, String path, Repository geogig) { Optional<RevObject> object = geogig.command(RevObjectParse.class).setObjectId(commitId).call(); RevCommit commit = null; if (object.isPresent() && object.get() instanceof RevCommit) { commit = (RevCommit) object.get(); } else { throw new CommandSpecException("Couldn't resolve id: " + commitId.toString() + " to a commit"); } object = geogig.command(RevObjectParse.class).setObjectId(commit.getTreeId()).call(); if (object.isPresent()) { RevTree tree = (RevTree) object.get(); return geogig.command(FindTreeChild.class).setParent(tree).setChildPath(path).call(); } else { throw new CommandSpecException("Couldn't resolve commit's treeId"); } } public void post(Representation entity) { InputStream input = null; try { input = getRequest().getEntity().getStream(); final Repository ggig = getGeogig(getRequest()).get(); final Reader body = new InputStreamReader(input); final JsonParser parser = new JsonParser(); final JsonElement conflictJson = parser.parse(body); if (conflictJson.isJsonObject()) { final JsonObject conflict = conflictJson.getAsJsonObject(); String featureId = null; RevFeature ourFeature = null; RevFeatureType ourFeatureType = null; RevFeature theirFeature = null; RevFeatureType theirFeatureType = null; JsonObject merges = null; if (conflict.has("path") && conflict.get("path").isJsonPrimitive()) { featureId = conflict.get("path").getAsJsonPrimitive().getAsString(); } Preconditions.checkState(featureId != null); if (conflict.has("ours") && conflict.get("ours").isJsonPrimitive()) { String ourCommit = conflict.get("ours").getAsJsonPrimitive().getAsString(); Optional<NodeRef> ourNode = parseID(ObjectId.valueOf(ourCommit), featureId, ggig); if (ourNode.isPresent()) { Optional<RevObject> object = ggig.command(RevObjectParse.class) .setObjectId(ourNode.get().getObjectId()).call(); Preconditions.checkState(object.isPresent() && object.get() instanceof RevFeature); ourFeature = (RevFeature) object.get(); object = ggig.command(RevObjectParse.class).setObjectId(ourNode.get().getMetadataId()) .call(); Preconditions.checkState(object.isPresent() && object.get() instanceof RevFeatureType); ourFeatureType = (RevFeatureType) object.get(); } } if (conflict.has("theirs") && conflict.get("theirs").isJsonPrimitive()) { String theirCommit = conflict.get("theirs").getAsJsonPrimitive().getAsString(); Optional<NodeRef> theirNode = parseID(ObjectId.valueOf(theirCommit), featureId, ggig); if (theirNode.isPresent()) { Optional<RevObject> object = ggig.command(RevObjectParse.class) .setObjectId(theirNode.get().getObjectId()).call(); Preconditions.checkState(object.isPresent() && object.get() instanceof RevFeature); theirFeature = (RevFeature) object.get(); object = ggig.command(RevObjectParse.class).setObjectId(theirNode.get().getMetadataId()) .call(); Preconditions.checkState(object.isPresent() && object.get() instanceof RevFeatureType); theirFeatureType = (RevFeatureType) object.get(); } } if (conflict.has("merges") && conflict.get("merges").isJsonObject()) { merges = conflict.get("merges").getAsJsonObject(); } Preconditions.checkState(merges != null); Preconditions.checkState(ourFeatureType != null || theirFeatureType != null); SimpleFeatureBuilder featureBuilder = new SimpleFeatureBuilder( (SimpleFeatureType) (ourFeatureType != null ? ourFeatureType.type() : theirFeatureType.type())); ImmutableList<PropertyDescriptor> descriptors = (ourFeatureType == null ? theirFeatureType : ourFeatureType).descriptors(); for (Entry<String, JsonElement> entry : merges.entrySet()) { int descriptorIndex = getDescriptorIndex(entry.getKey(), descriptors); if (descriptorIndex != -1 && entry.getValue().isJsonObject()) { PropertyDescriptor descriptor = descriptors.get(descriptorIndex); JsonObject attributeObject = entry.getValue().getAsJsonObject(); if (attributeObject.has("ours") && attributeObject.get("ours").isJsonPrimitive() && attributeObject.get("ours").getAsBoolean()) { featureBuilder.set(descriptor.getName(), ourFeature == null ? null : ourFeature.get(descriptorIndex).orNull()); } else if (attributeObject.has("theirs") && attributeObject.get("theirs").isJsonPrimitive() && attributeObject.get("theirs").getAsBoolean()) { featureBuilder.set(descriptor.getName(), theirFeature == null ? null : theirFeature.get(descriptorIndex).orNull()); } else if (attributeObject.has("value") && attributeObject.get("value").isJsonPrimitive()) { JsonPrimitive primitive = attributeObject.get("value").getAsJsonPrimitive(); if (primitive.isString()) { try { Object object = valueFromString( FieldType.forBinding(descriptor.getType().getBinding()), primitive.getAsString()); featureBuilder.set(descriptor.getName(), object); } catch (Exception e) { throw new Exception("Unable to convert attribute (" + entry.getKey() + ") to required type: " + descriptor.getType().getBinding().toString()); } } else if (primitive.isNumber()) { try { Object value = valueFromNumber( FieldType.forBinding(descriptor.getType().getBinding()), primitive.getAsNumber()); featureBuilder.set(descriptor.getName(), value); } catch (Exception e) { throw new Exception("Unable to convert attribute (" + entry.getKey() + ") to required type: " + descriptor.getType().getBinding().toString()); } } else if (primitive.isBoolean()) { try { Object value = valueFromBoolean( FieldType.forBinding(descriptor.getType().getBinding()), primitive.getAsBoolean()); featureBuilder.set(descriptor.getName(), value); } catch (Exception e) { throw new Exception("Unable to convert attribute (" + entry.getKey() + ") to required type: " + descriptor.getType().getBinding().toString()); } } else if (primitive.isJsonNull()) { featureBuilder.set(descriptor.getName(), null); } else { throw new Exception( "Unsupported JSON type for attribute value (" + entry.getKey() + ")"); } } } } SimpleFeature feature = featureBuilder.buildFeature(NodeRef.nodeFromPath(featureId)); RevFeature revFeature = RevFeatureBuilder.build(feature); ggig.objectDatabase().put(revFeature); getResponse() .setEntity(new StringRepresentation(revFeature.getId().toString(), MediaType.TEXT_PLAIN)); } } catch (Exception e) { throw new RestletException(e.getMessage(), Status.SERVER_ERROR_INTERNAL, e); } finally { if (input != null) Closeables.closeQuietly(input); } } private Object valueFromNumber(FieldType type, Number value) throws Exception { switch (type) { case NULL: return null; case BOOLEAN: return new Boolean(value.doubleValue() != 0); case BYTE: return new Byte(value.byteValue()); case SHORT: return new Short(value.shortValue()); case INTEGER: return new Integer(value.intValue()); case LONG: return new Long(value.longValue()); case FLOAT: return new Float(value.floatValue()); case DOUBLE: return new Double(value.doubleValue()); case STRING: return value.toString(); case BOOLEAN_ARRAY: boolean boolArray[] = { value.doubleValue() != 0 }; return boolArray; case BYTE_ARRAY: byte byteArray[] = { value.byteValue() }; return byteArray; case SHORT_ARRAY: short shortArray[] = { value.shortValue() }; return shortArray; case INTEGER_ARRAY: int intArray[] = { value.intValue() }; return intArray; case LONG_ARRAY: long longArray[] = { value.longValue() }; return longArray; case FLOAT_ARRAY: float floatArray[] = { value.floatValue() }; return floatArray; case DOUBLE_ARRAY: double doubleArray[] = { value.doubleValue() }; return doubleArray; case STRING_ARRAY: String stringArray[] = { value.toString() }; return stringArray; case DATETIME: return new Date(value.longValue()); case DATE: return new java.sql.Date(value.longValue()); case TIME: return new java.sql.Time(value.longValue()); case TIMESTAMP: return new java.sql.Timestamp(value.longValue()); case POINT: case LINESTRING: case POLYGON: case MULTIPOINT: case MULTILINESTRING: case MULTIPOLYGON: case GEOMETRYCOLLECTION: case GEOMETRY: case UUID: case BIG_INTEGER: case BIG_DECIMAL: default: break; } throw new IOException(); } private Object valueFromBoolean(FieldType type, boolean value) throws Exception { switch (type) { case NULL: return null; case BOOLEAN: return new Boolean(value); case BYTE: return new Byte((byte) (value ? 1 : 0)); case SHORT: return new Short((short) (value ? 1 : 0)); case INTEGER: return new Integer((int) (value ? 1 : 0)); case LONG: return new Long((long) (value ? 1 : 0)); case FLOAT: return new Float((float) (value ? 1 : 0)); case DOUBLE: return new Double((double) (value ? 1 : 0)); case STRING: return Boolean.toString(value); case BOOLEAN_ARRAY: boolean boolArray[] = { value }; return boolArray; case BYTE_ARRAY: byte byteArray[] = { (byte) (value ? 1 : 0) }; return byteArray; case SHORT_ARRAY: short shortArray[] = { (short) (value ? 1 : 0) }; return shortArray; case INTEGER_ARRAY: int intArray[] = { (int) (value ? 1 : 0) }; return intArray; case LONG_ARRAY: long longArray[] = { (long) (value ? 1 : 0) }; return longArray; case FLOAT_ARRAY: float floatArray[] = { (float) (value ? 1 : 0) }; return floatArray; case DOUBLE_ARRAY: double doubleArray[] = { (double) (value ? 1 : 0) }; return doubleArray; case STRING_ARRAY: String stringArray[] = { Boolean.toString(value) }; return stringArray; case POINT: case LINESTRING: case POLYGON: case MULTIPOINT: case MULTILINESTRING: case MULTIPOLYGON: case GEOMETRYCOLLECTION: case GEOMETRY: case UUID: case BIG_INTEGER: case BIG_DECIMAL: case DATETIME: case DATE: case TIME: case TIMESTAMP: default: break; } throw new IOException(); } private Object valueFromString(FieldType type, String value) throws Exception { Geometry geom; switch (type) { case NULL: return null; case BOOLEAN: return new Boolean(value); case BYTE: return new Byte(value); case SHORT: return new Short(value); case INTEGER: return new Integer(value); case LONG: return new Long(value); case FLOAT: return new Float(value); case DOUBLE: return new Double(value); case STRING: return value; case BOOLEAN_ARRAY: boolean boolArray[] = { Boolean.parseBoolean(value) }; return boolArray; case BYTE_ARRAY: byte byteArray[] = { Byte.parseByte(value) }; return byteArray; case SHORT_ARRAY: short shortArray[] = { Short.parseShort(value) }; return shortArray; case INTEGER_ARRAY: int intArray[] = { Integer.parseInt(value) }; return intArray; case LONG_ARRAY: long longArray[] = { Long.parseLong(value) }; return longArray; case FLOAT_ARRAY: float floatArray[] = { Float.parseFloat(value) }; return floatArray; case DOUBLE_ARRAY: double doubleArray[] = { Double.parseDouble(value) }; return doubleArray; case STRING_ARRAY: String stringArray[] = { value }; return stringArray; case UUID: return UUID.fromString(value); case BIG_INTEGER: return new BigInteger(value); case BIG_DECIMAL: return new BigDecimal(value); case DATETIME: return new SimpleDateFormat().parse(value); case DATE: return java.sql.Date.valueOf(value); case TIME: return java.sql.Time.valueOf(value); case TIMESTAMP: return new java.sql.Timestamp(new SimpleDateFormat().parse(value).getTime()); case POINT: geom = new WKTReader2().read(value); if (geom instanceof Point) { return (Point) geom; } break; case LINESTRING: geom = new WKTReader2().read(value); if (geom instanceof LineString) { return (LineString) geom; } break; case POLYGON: geom = new WKTReader2().read(value); if (geom instanceof Polygon) { return (Polygon) geom; } break; case MULTIPOINT: geom = new WKTReader2().read(value); if (geom instanceof MultiPoint) { return (MultiPoint) geom; } break; case MULTILINESTRING: geom = new WKTReader2().read(value); if (geom instanceof MultiLineString) { return (MultiLineString) geom; } break; case MULTIPOLYGON: geom = new WKTReader2().read(value); if (geom instanceof MultiPolygon) { return (MultiPolygon) geom; } break; case GEOMETRYCOLLECTION: geom = new WKTReader2().read(value); if (geom instanceof GeometryCollection) { return (GeometryCollection) geom; } break; case GEOMETRY: return new WKTReader2().read(value); default: break; } throw new IOException(); } private int getDescriptorIndex(String key, ImmutableList<PropertyDescriptor> properties) { for (int i = 0; i < properties.size(); i++) { PropertyDescriptor prop = properties.get(i); if (prop.getName().toString().equals(key)) { return i; } } return -1; } }