仿GeometryJson 转换GeoJSON Jackson序列化类 v2

仿GeometryJson 转换GeoJSON Jackson序列化类 v2

核心序列化类

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import org.locationtech.jts.geom.*;

import java.io.IOException;

public class ToJsonGeometrySerializer extends JsonSerializer<Geometry> {

    @Override
    public Class<Geometry> handledType() {
        return Geometry.class;
    }

    @Override
    public void serialize(Geometry geometry, JsonGenerator gen, SerializerProvider provider) throws IOException {
        if (geometry == null) {
            gen.writeNull();
            return;
        }

        writeGeometry(gen, geometry);
    }

    private void writeGeometry(JsonGenerator gen, Geometry geometry) throws IOException {
        if (geometry instanceof Point) {
            writeGeometry(gen, (Point) geometry);
        } else if (geometry instanceof LineString) {
            writeGeometry(gen, (LineString) geometry);
        } else if (geometry instanceof Polygon) {
            writeGeometry(gen, (Polygon) geometry);
        } else if (geometry instanceof MultiPoint) {
            writeGeometry(gen, (MultiPoint) geometry);
        } else if (geometry instanceof MultiLineString) {
            writeGeometry(gen, (MultiLineString) geometry);
        } else if (geometry instanceof MultiPolygon) {
            writeGeometry(gen, (MultiPolygon) geometry);
        } else if (geometry instanceof GeometryCollection) {
            writeGeometry(gen, (GeometryCollection) geometry);
        } else {
            String message = String.format("不支持该Geometry类型:%s", geometry.getGeometryType());
            throw new IllegalArgumentException(message);
        }
    }

    private void writeGeometry(JsonGenerator gen, GeometryCollection geometry)
            throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeArrayFieldStart(GeomConstants.GEOMETRIES);

        for (int i = 0; i != geometry.getNumGeometries(); ++i) {
            writeGeometry(gen, geometry.getGeometryN(i));
        }
        gen.writeEndArray();
        gen.writeEndObject();
    }


    private void writeGeometry(JsonGenerator gen, MultiPolygon geometry)
            throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeArrayFieldStart(GeomConstants.COORDINATES);
        for (int i = 0; i != geometry.getNumGeometries(); ++i) {
            writeCoordinates(gen, (Polygon) geometry.getGeometryN(i));
        }
        gen.writeEndArray();
        gen.writeEndObject();
    }

    private void writeGeometry(JsonGenerator gen, MultiLineString geometry)
            throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeArrayFieldStart(GeomConstants.COORDINATES);
        for (int i = 0; i != geometry.getNumGeometries(); ++i) {
            writeCoordinates(gen, (LineString) geometry.getGeometryN(i));
        }
        gen.writeEndArray();
        gen.writeEndObject();
    }


    private void writeGeometry(JsonGenerator gen, MultiPoint geometry)
            throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeArrayFieldStart(GeomConstants.COORDINATES);
        for (int i = 0; i != geometry.getNumGeometries(); ++i) {
            writeCoordinates(gen, (Point) geometry.getGeometryN(i));
        }
        gen.writeEndArray();
        gen.writeEndObject();
    }


    private void writeGeometry(JsonGenerator gen, Polygon geometry) throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeFieldName(GeomConstants.COORDINATES);
        writeCoordinates(gen, geometry);
        gen.writeEndObject();
    }

    private void writeCoordinates(JsonGenerator gen, Polygon polygon)
            throws IOException {
        gen.writeStartArray();
        // 画外环
        writeCoordinates(gen, polygon.getExteriorRing());

        for (int i = 0; i < polygon.getNumInteriorRing(); ++i) {
            // 画内环
            writeCoordinates(gen, polygon.getInteriorRingN(i));
        }
        gen.writeEndArray();
    }

    private void writeGeometry(JsonGenerator gen, LineString geometry) throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeFieldName(GeomConstants.COORDINATES);
        writeCoordinates(gen, geometry);
        gen.writeEndObject();
    }

    private void writeCoordinates(JsonGenerator gen, LineString lineString)
            throws IOException {
        gen.writeStartArray();
        for (int i = 0; i != lineString.getNumPoints(); ++i) {
            Point p = lineString.getPointN(i);
            writeCoordinates(gen, p);
        }
        gen.writeEndArray();
    }


    private void writeGeometry(JsonGenerator gen, Point geometry) throws IOException {
        gen.writeStartObject();
        gen.writeStringField(GeomConstants.TYPE, geometry.getGeometryType());
        gen.writeFieldName(GeomConstants.COORDINATES);
        writeCoordinates(gen, geometry);
        gen.writeEndObject();
    }

    private void writeCoordinates(JsonGenerator gen, Point point)
            throws IOException {
        gen.writeStartArray();

        final Coordinate coordinate = point.getCoordinate();
        if (null != coordinate) {
            gen.writeNumber(coordinate.x);
            gen.writeNumber(coordinate.y);

            if (!Double.isNaN(coordinate.z)) {
                gen.writeNumber(coordinate.z);
            }
        }

        gen.writeEndArray();
    }
}