style: use constants for pg types

teamkeel · Jan 21, 2025 · bf079e1 · bf079e1
1 parent 07866c9
commit bf079e1
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/packages/functions-runtime/src/database.js b/packages/functions-runtime/src/database.js
@@ -147,12 +147,12 @@ function getDialect(connString) {
     case "pg":
       // Adding a custom type parser for numeric fields: see https://kysely.dev/docs/recipes/data-types#configuring-runtime-javascript-types
       // 1700 = type for NUMERIC
-      pg.types.setTypeParser(1700, function (val) {
+      pg.types.setTypeParser(pg.types.builtins.NUMERIC, function (val) {
         return parseFloat(val);
       });
       // Adding a custom type parser for interval fields: see https://kysely.dev/docs/recipes/data-types#configuring-runtime-javascript-types
       // 1186 = type for INTERVAL
-      pg.types.setTypeParser(1186, function (val) {
+      pg.types.setTypeParser(pg.types.builtins.INTERVAL, function (val) {
         return new Duration(val);
       });
 
@@ -185,14 +185,20 @@ function getDialect(connString) {
     case "neon":
       // Adding a custom type parser for numeric fields: see https://kysely.dev/docs/recipes/data-types#configuring-runtime-javascript-types
       // 1700 = type for NUMERIC
-      neonserverless.types.setTypeParser(1700, function (val) {
-        return parseFloat(val);
-      });
+      neonserverless.types.setTypeParser(
+        pg.types.builtins.NUMERIC,
+        function (val) {
+          return parseFloat(val);
+        }
+      );
       // Adding a custom type parser for interval fields: see https://kysely.dev/docs/recipes/data-types#configuring-runtime-javascript-types
       // 1186 = type for INTERVAL
-      neonserverless.types.setTypeParser(1186, function (val) {
-        return new Duration(val);
-      });
+      neonserverless.types.setTypeParser(
+        pg.types.builtins.INTERVAL,
+        function (val) {
+          return new Duration(val);
+        }
+      );
 
       neonserverless.neonConfig.webSocketConstructor = ws;