Add compiler plugin validations for PostgreSQL data types

ballerina/Ballerina.toml

@@ -1,7 +1,7 @@
 [package]
 org = "ballerina"
 name = "persist"
-version = "1.2.0"
+version = "1.2.1"
 authors = ["Ballerina"]
 keywords = ["persist"]
 repository = "https://github.com/ballerina-platform/module-ballerina-persist"
@@ -15,5 +15,5 @@ graalvmCompatible = true
 [[platform.java17.dependency]]
 groupId = "io.ballerina.persist"
 artifactId = "persist-native"
-version = "1.2.0"
-path = "../native/build/libs/persist-native-1.2.0.jar"
+version = "1.2.1"
+path = "../native/build/libs/persist-native-1.2.1-SNAPSHOT.jar"

ballerina/CompilerPlugin.toml

@@ -3,4 +3,4 @@ id = "persist-compiler-plugin"
 class = "io.ballerina.stdlib.persist.compiler.PersistCompilerPlugin"
 
 [[dependency]]
-path = "../compiler-plugin/build/libs/persist-compiler-plugin-1.2.0.jar"
+path = "../compiler-plugin/build/libs/persist-compiler-plugin-1.2.1-SNAPSHOT.jar"

ballerina/Dependencies.toml

@@ -18,7 +18,7 @@ modules = [
 [[package]]
 org = "ballerina"
 name = "persist"
-version = "1.2.0"
+version = "1.2.1"
 dependencies = [
  {org = "ballerina", name = "jballerina.java"}
 ]

changelog.md

@@ -7,6 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [Unreleased]
 
 ### Added
+- [Added compiler plugin validations for Postgresql as a datasource](https://github.com/ballerina-platform/ballerina-library/issues/5829)
 
 ### Changed
 

compiler-plugin-test/src/test/java/io/ballerina/stdlib/persist/compiler/CompilerPluginTest.java

@@ -245,6 +245,50 @@ public void validateEntityFieldTypeForMssql() {
  );
  }
 
+ @Test
+ public void validateEntityFieldTypeForPostgresql() {
+ List<Diagnostic> diagnostics = getErrorDiagnostics("project_6", "field-types.bal", 10);
+ testDiagnostic(
+ diagnostics,
+ new String[]{
+ PERSIST_306.getCode(),
+ PERSIST_305.getCode(),
+ PERSIST_306.getCode(),
+ PERSIST_306.getCode(),
+ PERSIST_305.getCode(),
+ PERSIST_305.getCode(),
+ PERSIST_306.getCode(),
+ PERSIST_305.getCode(),
+ PERSIST_306.getCode(),
+ PERSIST_306.getCode(),
+ },
+ new String[]{
+ "an entity does not support boolean array field type",
+ "an entity does not support json-typed field",
+ "an entity does not support json array field type",
+ "an entity does not support time:Civil array field type",
+ "an entity does not support union-typed field",
+ "an entity does not support error-typed field",
+ "an entity does not support error array field type",
+ "an entity does not support mysql:Client-typed field",
+ "an entity does not support mysql:Client array field type",
+ "an entity does not support enum array field type"
+ },
+ new String[]{
+ "(18:4,18:13)",
+ "(20:4,20:8)",
+ "(21:4,21:10)",
+ "(24:4,24:16)",
+ "(25:4,25:21)",
+ "(27:4,27:9)",
+ "(28:4,28:11)",
+ "(30:4,30:16)",
+ "(31:4,31:18)",
+ "(34:4,34:12)"
+ }
+ );
+ }
+
  @Test
  public void validateEntityFieldTypeForGoogleSheets() {
  List<Diagnostic> diagnostics = getErrorDiagnostics("project_3", "field-types.bal", 12);

compiler-plugin-test/src/test/resources/project_6/Ballerina.toml

@@ -0,0 +1,7 @@
+[package]
+org = "root"
+name = "project_2"
+version = "0.1.0"
+
+[persist]
+datastore = "postgresql"

compiler-plugin-test/src/test/resources/project_6/persist/field-types.bal

@@ -0,0 +1,36 @@
+import ballerina/time;
+import ballerinax/mysql;
+import ballerina/persist as _;
+
+public enum Gender {
+ M,
+ F
+}
+
+public type MedicalNeed record {|
+ readonly int needId;
+ boolean booleanTest;
+ int itemId;
+ float floatTest;
+ decimal decimalTest;
+ string beneficiaryId;
+ byte[] beneficiaryIdByteArray;
+
+ boolean[] booleanArray;
+
+ json jsonTest;
+ json[] jsonArray;
+
+ time:Civil period;
+ time:Civil[] periodArray;
+ time:Civil|string unionType;
+
+ error errorType;
+ error[] errorArrayType;
+
+ mysql:Client clientType;
+ mysql:Client[] clientArrayType;
+
+ Gender gender;
+ Gender[] genderArray;
+|};

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";
 

compiler-plugin/src/main/java/io/ballerina/stdlib/persist/compiler/utils/ValidatorsByDatastore.java

@@ -92,6 +92,8 @@
  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 @@
  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 @@
  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,20 @@
  }
  }
 
+ 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 +215,15 @@
  }
  }
 
+ 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 +262,21 @@
  }
  }
 
+ 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;
  }