Add support for postgresql data types

ballerina-platform · Nov 29, 2023 · 264bdff · 264bdff
1 parent d501b4f
commit 264bdff
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 0 deletions.
diff --git a/compiler-plugin/src/main/java/io/ballerina/stdlib/persist/compiler/Constants.java b/compiler-plugin/src/main/java/io/ballerina/stdlib/persist/compiler/Constants.java
@@ -71,6 +71,7 @@ public static final class Datastores {
 
         public static final String MYSQL = "mysql";
         public static final String MSSQL = "mssql";
+        public static final String POSTGRESQL = "postgresql";
         public static final String IN_MEMORY = "inmemory";
         public static final String GOOGLE_SHEETS = "googlesheets";
 

diff --git a/...lugin/src/main/java/io/ballerina/stdlib/persist/compiler/utils/ValidatorsByDatastore.java b/...lugin/src/main/java/io/ballerina/stdlib/persist/compiler/utils/ValidatorsByDatastore.java
@@ -92,6 +92,8 @@ public static boolean isValidSimpleType(String type, String datastore) {
                 return isValidMysqlType(type);
             case Constants.Datastores.MSSQL:
                 return isValidMssqlType(type);
+            case Constants.Datastores.POSTGRESQL:
+                return isValidPostgresqlType(type);
             case Constants.Datastores.IN_MEMORY:
                 return isValidInMemoryType(type);
             case Constants.Datastores.GOOGLE_SHEETS:
@@ -107,6 +109,8 @@ public static boolean isValidArrayType(String type, String datastore) {
                 return isValidMysqlArrayType(type);
             case Constants.Datastores.MSSQL:
                 return isValidMssqlArrayType(type);
+            case Constants.Datastores.POSTGRESQL:
+                return isValidPostgresqlArrayType(type);
             case Constants.Datastores.IN_MEMORY:
                 return isValidInMemoryArrayType(type);
             case Constants.Datastores.GOOGLE_SHEETS:
@@ -122,6 +126,8 @@ public static boolean isValidImportedType(String modulePrefix, String identifier
                 return isValidMysqlImportedType(modulePrefix, identifier);
             case Constants.Datastores.MSSQL:
                 return isValidMssqlImportedType(modulePrefix, identifier);
+            case Constants.Datastores.POSTGRESQL:
+                return isValidPostgresqlImportedType(modulePrefix, identifier);
             case Constants.Datastores.IN_MEMORY:
                 return isValidInMemoryImportedType(modulePrefix, identifier);
             case Constants.Datastores.GOOGLE_SHEETS:
@@ -159,6 +165,21 @@ public static boolean isValidMssqlType(String type) {
         }
     }
 
+    public static boolean isValidPostgresqlType(String type) {
+        switch (type) {
+            case INT:
+            case BOOLEAN:
+            case DECIMAL:
+            case FLOAT:
+            case STRING:
+            case ENUM:
+                return true;
+            default:
+                return false;
+        }
+    }
+
+
     public static boolean isValidInMemoryType(String type) {
         return true;
     }
@@ -195,6 +216,16 @@ public static boolean isValidMssqlArrayType(String type) {
         }
     }
 
+    public static boolean isValidPostgresqlArrayType(String type) {
+        switch (type) {
+            case BYTE:
+                return true;
+            default:
+                return false;
+        }
+    }
+
+
     public static boolean isValidInMemoryArrayType(String type) {
        return true;
     }
@@ -233,6 +264,22 @@ public static boolean isValidMssqlImportedType(String modulePrefix, String ident
         }
     }
 
+    public static boolean isValidPostgresqlImportedType(String modulePrefix, String identifier) {
+        if (!modulePrefix.equals(TIME_MODULE)) {
+            return false;
+        }
+        switch (identifier) {
+            case DATE:
+            case TIME_OF_DAY:
+            case UTC:
+            case CIVIL:
+                return true;
+            default:
+                return false;
+        }
+    }
+
+
     public static boolean isValidInMemoryImportedType(String modulePrefix, String identifier) {
         return true;
     }