CodeCow/querydsl
1
0
Fork 0
mirror of https://github.com/querydsl/querydsl.git synced 2026-07-03 21:07:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added support for JTS Geometries

Browse Source
This commit is contained in:
Steven Thomson 2014-10-24 10:29:58 -05:00
parent d895d15611
commit cd8ee7409b
2 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
querydsl-apt/src/main/java/com/mysema/query/apt/DefaultConfiguration.java
View File

@ -183,6 +183,14 @@ public class DefaultConfiguration implements Configuration {
} catch (Exception e) {
// do nothing
}
try {
// register additional mappings, if querydsl-spatial and JTS is on the classpath
Class.forName("com.mysema.query.spatial.jts.JTSGeometryExpression");
SpatialSupport.addJTSSupport(module);
} catch (Exception e) {
// do nothing
}
defaultSerializerConfig = new SimpleSerializerConfig(entityAccessors, listAccessors,
mapAccessors, createDefaultVariable, "");

28
querydsl-apt/src/main/java/com/mysema/query/apt/SpatialSupport.java
View File

@ -31,10 +31,27 @@ public final class SpatialSupport {
new SimpleType("com.mysema.query.spatial.path." + entry.getValue()));
}
}
private static void registerJTSTypes(TypeMappings typeMappings) {
Map<String, String> additions = Maps.newHashMap();
additions.put("Geometry", "GeometryPath");
additions.put("GeometryCollection", "GeometryCollectionPath");
additions.put("LinearRing", "LinearRingPath");
additions.put("LineString", "LineStringPath");
additions.put("MultiLineString", "MultiLineStringPath");
additions.put("MultiPoint", "MultiPointPath");
additions.put("MultiPolygon", "MultiPolygonPath");
additions.put("Point", "PointPath");
additions.put("Polygon", "PolygonPath");
for (Map.Entry<String, String> entry : additions.entrySet()) {
typeMappings.register(
new SimpleType("com.vividsolutions.jts.geom."+ entry.getKey()),
new SimpleType("com.mysema.query.spatial.jts.path." + entry.getValue()));
}
}
private static void addImports(AbstractModule module) {
private static void addImports(AbstractModule module, String packageName) {
Set<String> imports = module.get(Set.class, CodegenModule.IMPORTS);
String packageName = "com.mysema.query.spatial.path";
if (imports.isEmpty()) {
imports = ImmutableSet.of(packageName);
} else {
@ -48,7 +65,12 @@ public final class SpatialSupport {
public static void addSupport(AbstractModule module) {
registerTypes(module.get(TypeMappings.class));
addImports(module);
addImports(module,"com.mysema.query.spatial.path");
}
public static void addJTSSupport(AbstractModule module) {
registerJTSTypes(module.get(TypeMappings.class));
addImports(module,"com.mysema.query.spatial.jts.path");
}
private SpatialSupport() {}