add DSE Types support

datastax · Nov 7, 2023 · 1397781 · 1397781
1 parent b3d4c45
commit 1397781
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/com/datastax/cdm/data/CqlData.java b/src/main/java/com/datastax/cdm/data/CqlData.java
@@ -1,5 +1,6 @@
 package com.datastax.cdm.data;
 
+import com.datastax.dse.driver.api.core.type.DseDataTypes;
 import com.datastax.oss.driver.api.core.data.UdtValue;
 import com.datastax.oss.driver.api.core.type.*;
 
@@ -38,6 +39,9 @@ public enum Type {
         primitiveDataTypeToJavaClassMap.put(DataTypes.TIMEUUID, java.util.UUID.class);
         primitiveDataTypeToJavaClassMap.put(DataTypes.COUNTER, Long.class);
         primitiveDataTypeToJavaClassMap.put(DataTypes.DURATION, com.datastax.oss.driver.api.core.data.CqlDuration.class);
+        primitiveDataTypeToJavaClassMap.put(DseDataTypes.POLYGON, com.datastax.dse.driver.api.core.data.geometry.Polygon.class);
+        primitiveDataTypeToJavaClassMap.put(DseDataTypes.POINT, com.datastax.dse.driver.api.core.data.geometry.Point.class);
+        primitiveDataTypeToJavaClassMap.put(DseDataTypes.LINE_STRING, com.datastax.dse.driver.api.core.data.geometry.LineString.class);
     }
 
     public static Type toType(DataType dataType) {