-
Notifications
You must be signed in to change notification settings - Fork 74
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
15ff9dd
commit 132bb85
Showing
7 changed files
with
165 additions
and
44 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
22 changes: 0 additions & 22 deletions
22
nop-orm-geo/src/main/java/io/nop/orm/geo/binder/GeometryDataParameterBinder.java
This file was deleted.
Oops, something went wrong.
138 changes: 138 additions & 0 deletions
138
nop-orm-geo/src/main/java/io/nop/orm/geo/type/GeometryTypeHandler.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,138 @@ | ||
| package io.nop.orm.geo.type; | ||
|
|
||
| import io.nop.api.core.exceptions.NopException; | ||
| import io.nop.commons.type.StdSqlType; | ||
| import io.nop.commons.util.IoHelper; | ||
| import io.nop.dao.dialect.IDataTypeHandler; | ||
| import io.nop.dao.dialect.IDialect; | ||
| import io.nop.dataset.binder.IDataParameters; | ||
| import org.geolatte.geom.ByteBuffer; | ||
| import org.geolatte.geom.ByteOrder; | ||
| import org.geolatte.geom.C2D; | ||
| import org.geolatte.geom.Envelope; | ||
| import org.geolatte.geom.Geometry; | ||
| import org.geolatte.geom.Polygon; | ||
| import org.geolatte.geom.PositionSequence; | ||
| import org.geolatte.geom.PositionSequenceBuilders; | ||
| import org.geolatte.geom.codec.Wkb; | ||
| import org.geolatte.geom.codec.WkbDecoder; | ||
| import org.geolatte.geom.codec.WkbEncoder; | ||
| import org.geolatte.geom.codec.Wkt; | ||
| import org.geolatte.geom.crs.CoordinateReferenceSystems; | ||
| import org.geolatte.geom.jts.JTS; | ||
|
|
||
| import java.io.InputStream; | ||
| import java.sql.Blob; | ||
|
|
||
| public class GeometryTypeHandler implements IDataTypeHandler { | ||
|
|
||
| @Override | ||
| public StdSqlType getStdSqlType() { | ||
| return StdSqlType.GEOMETRY; | ||
| } | ||
|
|
||
| protected String getGeomFromTextFunc() { | ||
| return "ST_GeomFromText"; | ||
| } | ||
|
|
||
| protected Wkt.Dialect getWktDialect() { | ||
| return Wkt.Dialect.SFA_1_1_0; | ||
| } | ||
|
|
||
| protected Wkb.Dialect getWkbDialect() { | ||
| return Wkb.Dialect.POSTGIS_EWKB_1; | ||
| } | ||
|
|
||
| protected Geometry toGeometry(Object value) { | ||
| return (Geometry) value; | ||
| } | ||
|
|
||
| protected ByteOrder getByteOrder() { | ||
| return ByteOrder.NDR; | ||
| } | ||
|
|
||
| @Override | ||
| public String toLiteral(Object value, IDialect dialect) { | ||
| Geometry geom = toGeometry(value); | ||
| StringBuilder sb = new StringBuilder(); | ||
| sb.append(getGeomFromTextFunc()); | ||
| sb.append("('").append(Wkt.toWkt(geom, getWktDialect())); | ||
| sb.append("',").append(Math.max(geom.getSRID(), 0)); | ||
| sb.append(')'); | ||
| return sb.toString(); | ||
| } | ||
|
|
||
|
|
||
| @Override | ||
| public Object fromLiteral(String text, IDialect dialect) { | ||
| return null; | ||
| } | ||
|
|
||
| @Override | ||
| public boolean isJavaType(Object value) { | ||
| return value instanceof Geometry; | ||
| } | ||
|
|
||
| @Override | ||
| public Object getValue(IDataParameters params, int index) { | ||
| Object value = params.getObject(index); | ||
| return fromValue(value); | ||
| } | ||
|
|
||
| @Override | ||
| public void setValue(IDataParameters params, int index, Object value) { | ||
| final Geometry geom = (Geometry) value; | ||
| params.setBytes(index, toBytes(geom, getWkbDialect(), getByteOrder())); | ||
| } | ||
|
|
||
| protected byte[] toBytes(Geometry geom, Wkb.Dialect dialect, ByteOrder byteOrder) { | ||
| final WkbEncoder encoder = Wkb.newEncoder(dialect); | ||
| final ByteBuffer buffer = encoder.encode(geom, byteOrder); | ||
| return (buffer == null ? null : buffer.toByteArray()); | ||
| } | ||
|
|
||
| protected Geometry fromValue(Object object) { | ||
| if (object == null) { | ||
| return null; | ||
| } | ||
| try { | ||
| if (object instanceof org.locationtech.jts.geom.Geometry) { | ||
| return JTS.from((org.locationtech.jts.geom.Geometry) object); | ||
| } | ||
| final WkbDecoder decoder = Wkb.newDecoder(Wkb.Dialect.POSTGIS_EWKB_1); | ||
| if (object instanceof Blob) { | ||
| return decoder.decode(toByteBuffer((Blob) object)); | ||
| } else if (object instanceof byte[]) { | ||
| return decoder.decode(ByteBuffer.from((byte[]) object)); | ||
| } else if (object instanceof org.locationtech.jts.geom.Envelope) { | ||
| return toPolygon(JTS.from((org.locationtech.jts.geom.Envelope) object)); | ||
| } else { | ||
| throw new IllegalArgumentException(); | ||
| } | ||
| } catch (Exception e) { | ||
| throw NopException.adapt(e); | ||
| } | ||
| } | ||
|
|
||
| private static Geometry<C2D> toPolygon(Envelope env) { | ||
| final PositionSequence<C2D> ps = PositionSequenceBuilders.fixedSized(4, C2D.class) | ||
| .add(env.lowerLeft().getCoordinate(0), env.lowerLeft().getCoordinate(1)) | ||
| .add(env.lowerLeft().getCoordinate(0), env.upperRight().getCoordinate(1)) | ||
| .add(env.upperRight().getCoordinate(0), env.upperRight().getCoordinate(1)) | ||
| .add(env.lowerLeft().getCoordinate(0), env.lowerLeft().getCoordinate(1)) | ||
| .toPositionSequence(); | ||
| return new Polygon<C2D>(ps, CoordinateReferenceSystems.PROJECTED_2D_METER); | ||
| } | ||
|
|
||
| private static ByteBuffer toByteBuffer(Blob blob) { | ||
| InputStream is = null; | ||
| try { | ||
| is = blob.getBinaryStream(); | ||
| return ByteBuffer.from(IoHelper.readBytes(is)); | ||
| } catch (Exception e) { | ||
| throw NopException.adapt(e); | ||
| } finally { | ||
| IoHelper.safeCloseObject(is); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters