support geo for reader (#11)

vesoft-inc · Oct 29, 2021 · 2632f1d · 2632f1d
1 parent 02919c9
commit 2632f1d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...k-connector/src/main/scala/com/vesoft/nebula/connector/reader/NebulaPartitionReader.scala b/...k-connector/src/main/scala/com/vesoft/nebula/connector/reader/NebulaPartitionReader.scala
@@ -106,6 +106,9 @@ abstract class NebulaPartitionReader extends InputPartitionReader[InternalRow] {
       if (value.isDouble) {
         getters(i).apply(value.asDouble(), mutableRow, i)
       }
+      if (value.isGeography) {
+        getters(i).apply(value.asGeography(), mutableRow, i)
+      }
     }
     mutableRow
   }