feat(query): implement ST_SRID (#15377)

ST_SRID

Signed-off-by: Fan Yang <yangfanlinux@gmail.com>

Author: kkk25641463

src/query/functions/src/scalars/geometry.rs

@@ -865,6 +865,33 @@ pub fn register(registry: &mut FunctionRegistry) {
         ),
     );
 
+    registry.register_combine_nullable_1_arg::<GeometryType, Int32Type, _, _>(
+        "st_srid",
+        |_, _| FunctionDomain::MayThrow,
+        vectorize_with_builder_1_arg::<GeometryType, NullableType<Int32Type>>(
+            |geometry, output, ctx| {
+                if let Some(validity) = &ctx.validity {
+                    if !validity.get_bit(output.len()) {
+                        output.push_null();
+                        return;
+                    }
+                }
+                match read_ewkb_srid(&mut io::Cursor::new(&geometry)) {
+                    Ok(srid) => {
+                        output.push(srid.unwrap_or(4326));
+                    }
+                    Err(e) => {
+                        ctx.set_error(
+                            output.len(),
+                            ErrorCode::GeometryError(e.to_string()).to_string(),
+                        );
+                        output.push(0);
+                    }
+                };
+            },
+        ),
+    );
+
     registry.register_passthrough_nullable_1_arg::<GeometryType, StringType, _, _>(
         "to_string",
         |_, _| FunctionDomain::MayThrow,

src/query/functions/tests/it/scalars/geometry.rs

@@ -40,6 +40,7 @@ fn test_geometry() {
     test_st_makepoint(file);
     test_st_makepolygon(file);
     test_st_pointn(file);
+    test_st_srid(file);
     test_to_geometry(file);
     test_to_string(file);
     test_try_to_geometry(file);
@@ -241,6 +242,17 @@ fn test_st_pointn(file: &mut impl Write) {
     );
 }
 
+fn test_st_srid(file: &mut impl Write) {
+    run_ast(
+        file,
+        "st_srid(to_geometry('POINT(-122.306100 37.554162)', 1234))",
+        &[],
+    );
+    run_ast(file, "st_srid(st_makegeompoint(37.5, 45.5))", &[]);
+    run_ast(file, "st_srid(st_makegeompoint(NULL, NULL))", &[]);
+    run_ast(file, "st_srid(NULL)", &[]);
+}
+
 fn test_to_geometry(file: &mut impl Write) {
     run_ast(file, "to_geometry('POINT(1820.12 890.56)')", &[]);
     run_ast(file, "to_geometry('SRID=4326;POINT(1820.12 890.56)')", &[]);

src/query/functions/tests/it/scalars/testdata/function_list.txt

@@ -3548,6 +3548,8 @@ Functions overloads:
 1 st_makepolygon(Geometry NULL) :: Geometry NULL
 0 st_pointn(Geometry, Int32) :: Geometry
 1 st_pointn(Geometry NULL, Int32 NULL) :: Geometry NULL
+0 st_srid(Geometry) :: Int32 NULL
+1 st_srid(Geometry NULL) :: Int32 NULL
 0 strcmp(String, String) :: Int8
 1 strcmp(String NULL, String NULL) :: Int8 NULL
 0 string_to_h3(String) :: UInt64

src/query/functions/tests/it/scalars/testdata/geometry.txt

@@ -348,6 +348,42 @@ output domain  : Undefined
 output         : 'POINT(3 3)'
 
 
+ast            : st_srid(to_geometry('POINT(-122.306100 37.554162)', 1234))
+raw expr       : st_srid(to_geometry('POINT(-122.306100 37.554162)', 1234))
+checked expr   : st_srid<Geometry>(to_geometry<String, Int32>("POINT(-122.306100 37.554162)", to_int32<UInt16>(1234_u16)))
+optimized expr : 1234_i32
+output type    : Int32 NULL
+output domain  : {1234..=1234}
+output         : 1234
+
+
+ast            : st_srid(st_makegeompoint(37.5, 45.5))
+raw expr       : st_srid(st_makegeompoint(37.5, 45.5))
+checked expr   : st_srid<Geometry>(st_makegeompoint<Float64, Float64>(to_float64<Decimal(3, 1)>(37.5_d128(3,1)), to_float64<Decimal(3, 1)>(45.5_d128(3,1))))
+optimized expr : 4326_i32
+output type    : Int32 NULL
+output domain  : {4326..=4326}
+output         : 4326
+
+
+ast            : st_srid(st_makegeompoint(NULL, NULL))
+raw expr       : st_srid(st_makegeompoint(NULL, NULL))
+checked expr   : st_srid<Geometry NULL>(st_makegeompoint<Float64 NULL, Float64 NULL>(CAST(NULL AS Float64 NULL), CAST(NULL AS Float64 NULL)))
+optimized expr : NULL
+output type    : Int32 NULL
+output domain  : {NULL}
+output         : NULL
+
+
+ast            : st_srid(NULL)
+raw expr       : st_srid(NULL)
+checked expr   : st_srid<Geometry NULL>(CAST(NULL AS Geometry NULL))
+optimized expr : NULL
+output type    : Int32 NULL
+output domain  : {NULL}
+output         : NULL
+
+
 ast            : to_geometry('POINT(1820.12 890.56)')
 raw expr       : to_geometry('POINT(1820.12 890.56)')
 checked expr   : to_geometry<String>("POINT(1820.12 890.56)")

tests/sqllogictests/suites/query/functions/02_0060_function_geometry.test

@@ -423,6 +423,26 @@ SELECT TO_STRING(ST_POINTN(TO_GEOMETRY('LINESTRING(1 1, 2 2, 3 3, 4 4)'), -2))
 POINT(3 3)
 
 
+query T
+SELECT ST_SRID(TO_GEOMETRY('POINT(-122.306100 37.554162)', 1234))
+----
+1234
+
+query T
+SELECT ST_SRID(ST_MAKEGEOMPOINT(37.5, 45.5))
+----
+4326
+
+query T
+SELECT ST_SRID(ST_MAKEGEOMPOINT(NULL, NULL))
+----
+NULL
+
+query T
+SELECT ST_SRID(NULL)
+----
+NULL
+
 statement ok
 SET enable_geo_create_table=0