support Geometry too

trinodb · Apr 29, 2024 · fd368cb · fd368cb
1 parent 5a52680
commit fd368cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/trino/trino.go b/trino/trino.go
@@ -1469,7 +1469,7 @@ func (c *typeConverter) ConvertValue(v interface{}) (driver.Value, error) {
 			return nil, err
 		}
 		return vv.Bool, err
-	case "json", "char", "varchar", "varbinary", "interval year to month", "interval day to second", "decimal", "ipaddress", "uuid", "SphericalGeography", "unknown":
+	case "json", "char", "varchar", "varbinary", "interval year to month", "interval day to second", "decimal", "ipaddress", "uuid", "Geometry", "SphericalGeography", "unknown":
 		vv, err := scanNullString(v)
 		if !vv.Valid {
 			return nil, err

diff --git a/trino/trino_test.go b/trino/trino_test.go
@@ -1419,6 +1419,13 @@ func TestTypeConversion(t *testing.T) {
 				[]interface{}{"b"},
 			},
 		},
+		{
+			DataType:                   "Geometry",
+			RawType:                    "Geometry",
+			ResponseUnmarshalledSample: "Point (0 0)",
+			ExpectedGoValue:            "Point (0 0)",
+		},
+
 		{
 			DataType:                   "SphericalGeography",
 			RawType:                    "SphericalGeography",