Limiting the upgrade script to required datatypes

Signed-off-by: Tanya Gupta <[email protected]>
amazon-aurora · Sep 25, 2024 · f3a8aa7 · f3a8aa7
1 parent 9b82c10
commit f3a8aa7
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/contrib/babelfishpg_tsql/sql/upgrades/babelfishpg_tsql--4.3.0--4.4.0.sql b/contrib/babelfishpg_tsql/sql/upgrades/babelfishpg_tsql--4.3.0--4.4.0.sql
@@ -1916,13 +1916,13 @@ SET probin = (
                 (
                     SELECT jsonb_agg(
                         CASE
-                            WHEN val::text = '-8000' THEN '-1'
-                            WHEN val::text = '-1' AND p2.proallargtypes[indx] = ('sys.smalldatetime'::regtype)::oid THEN '0'
-                            WHEN val::text = '-1' THEN '1'
-                            ELSE val::text
+                            WHEN a::text = '-8000' AND (p2.proallargtypes[b] = ('sys.varchar'::regtype)::oid OR p2.proallargtypes[b] = ('sys.nvarchar'::regtype)::oid OR p2.proallargtypes[b] = ('sys.varbinary'::regtype)::oid) THEN '-1'
+                            WHEN a::text = '-1' AND p2.proallargtypes[b] = ('sys.smalldatetime'::regtype)::oid THEN '0'
+                            WHEN a::text = '-1' AND (p2.proallargtypes[b] = ('sys.varchar'::regtype)::oid OR p2.proallargtypes[b] = ('sys.nvarchar'::regtype)::oid OR p2.proallargtypes[b] = ('sys.varbinary'::regtype)::oid OR p2.proallargtypes[b] = ('sys.nchar'::regtype)::oid OR p2.proallargtypes[b] = ('sys.binary'::regtype)::oid OR p2.proallargtypes[b] = ('sys.bpchar'::regtype)::oid) THEN '1'
+                            ELSE a::text
                         END
                     )
-                    FROM jsonb_array_elements_text(probin::jsonb->'typmod_array')  WITH ORDINALITY AS elem(val,indx)
+                    FROM jsonb_array_elements_text(probin::jsonb->'typmod_array')  WITH ORDINALITY AS elem(a,b)
                 )
             )
         )::text