Add function add_type_for_try_convert to sql

open-gpdb · Nov 27, 2024 · 6d8305b · 6d8305b
1 parent 7e81389
commit 6d8305b
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/contrib/try_convert/try_convert--1.0.sql b/contrib/try_convert/try_convert--1.0.sql
@@ -14,17 +14,22 @@ RETURNS anyelement
 AS 'MODULE_PATHNAME', 'try_convert'
 LANGUAGE C STRICT;
 
-CREATE FUNCTION try_convert(int8, anyelement)
-RETURNS anyelement
-AS 'MODULE_PATHNAME', 'try_convert'
-LANGUAGE C STRICT;
 
-CREATE FUNCTION try_convert(int4, anyelement)
-RETURNS anyelement
-AS 'MODULE_PATHNAME', 'try_convert'
-LANGUAGE C STRICT;
+CREATE OR REPLACE FUNCTION add_type_for_try_convert(type regtype)
+  RETURNS void 
+  LANGUAGE plpgsql AS
+$func$
+BEGIN
+   EXECUTE 'CREATE FUNCTION try_convert(' || type || ', anyelement)
+            RETURNS anyelement
+            AS ''MODULE_PATHNAME'', ''try_convert''
+            LANGUAGE C STRICT;';
+END
+$func$;
 
-CREATE FUNCTION try_convert(int2, anyelement)
-RETURNS anyelement
-AS 'MODULE_PATHNAME', 'try_convert'
-LANGUAGE C STRICT;
+select add_type_for_try_convert('int2'::regtype);
+select add_type_for_try_convert('int4'::regtype);
+select add_type_for_try_convert('int8'::regtype);
+select add_type_for_try_convert('float4'::regtype);
+select add_type_for_try_convert('float8'::regtype);
+select add_type_for_try_convert('date'::regtype);