[WIP]: Database Templates

common/src/main/java/com/adaptivescale/rosetta/common/TranslationMatrix.java

@@ -1,5 +1,6 @@
 package com.adaptivescale.rosetta.common;
 
+import com.adaptivescale.rosetta.common.models.DatabaseTemplateModel;
 import com.adaptivescale.rosetta.common.models.TranslationAttributeModel;
 import com.adaptivescale.rosetta.common.models.TranslationModel;
 import lombok.extern.slf4j.Slf4j;
@@ -20,6 +21,11 @@ public class TranslationMatrix {
     private static final String TRANSLATION_TABLE_NAME = "TRANSLATION";
     private static final String TRANSLATION_ATTRIBUTE_TABLE_NAME = "TRANSLATION_ATTRIBUTE";
 
+    private static final String DATABASE_TEMPLATE_TABLE_NAME = "DATABASE_TEMPLATE";
+
+    private static final String DEFAULT_DATABASE_TEMPLATE_FILE = "translation_matrix/database_template.csv";
+
+
     private static TranslationMatrix instance = null;
 
     public TranslationMatrix() {
@@ -58,6 +64,14 @@ void initTables() throws IOException {
                 "ON "+ TRANSLATION_TABLE_NAME +" (source_type, source_column_type, target_type)";
         execute(index);
         execute(uniqueIndex);
+
+        String databaseTemplateTable = "CREATE TABLE "+ DATABASE_TEMPLATE_TABLE_NAME +"(id INT PRIMARY KEY AUTO_INCREMENT, " +
+                "database_type VARCHAR(255) not null, " +
+                "template_name VARCHAR(255) not null, " +
+                "template_file VARCHAR(255) not null " +
+                ");";
+        execute(databaseTemplateTable);
+
         loadCSVData();
     }
 
@@ -66,7 +80,9 @@ void loadCSVData() throws IOException {
 
         StringBuilder dataInsertQuery = new StringBuilder();
         StringBuilder attributesInsertQuery = new StringBuilder();
+        StringBuilder databaseTemplateInsertQuery = new StringBuilder();
 
+        readDatabaseTemplate(databaseTemplateInsertQuery);
         Map<Integer, List<TranslationAttributeModel>> translationAttributesMappedByTranslationId = readTranslationAttributes(attributesInsertQuery);
         BufferedReader br = readTranslationMatrixFile();
 
@@ -86,6 +102,7 @@ void loadCSVData() throws IOException {
         }
         execute(dataInsertQuery.toString());
         execute(attributesInsertQuery.toString());
+        execute(databaseTemplateInsertQuery.toString());
     }
 
     void execute(String sql) {
@@ -144,6 +161,36 @@ private List<TranslationAttributeModel> getTranslationAttributeRecords(String qu
         }
     }
 
+    private Map<String, Map<String, String>> getDatabaseTemplates(String query) {
+        try {
+            Connection connection = DriverManager.getConnection(URL);
+            Statement statement = connection.createStatement();
+            ResultSet resultSet = statement.executeQuery(query);
+            Map<String, Map<String, String>> result = new HashMap<>();
+
+            while (resultSet.next()) {
+                DatabaseTemplateModel databaseTemplateModel = new DatabaseTemplateModel();
+                databaseTemplateModel.setId(resultSet.getInt("id"));
+                databaseTemplateModel.setDatabaseType(resultSet.getString("database_type"));
+                databaseTemplateModel.setTemplateName(resultSet.getString("template_name"));
+                databaseTemplateModel.setTemplateFile(resultSet.getString("template_file"));
+
+                if (!result.containsKey(databaseTemplateModel.getDatabaseType())) {
+                    result.put(databaseTemplateModel.getDatabaseType(), new HashMap<>());
+                }
+
+                result.get(databaseTemplateModel.getDatabaseType())
+                        .put(databaseTemplateModel.getTemplateName(), databaseTemplateModel.getTemplateFile());
+            }
+
+            connection.close();
+
+            return result;
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
     public TranslationModel findById(Integer id) {
         String query = String.format("SELECT TOP 1 * from %s where id=%s", TRANSLATION_TABLE_NAME, id);
         return getSingleRecord(query);
@@ -167,6 +214,13 @@ public List<TranslationAttributeModel> findByTranslationAttributesByTranslationI
         return getTranslationAttributeRecords(query);
     }
 
+    public Map<String, Map<String, String>> findDatabaseTemplates() {
+        String query = String.format("SELECT * from %s",
+                DATABASE_TEMPLATE_TABLE_NAME);
+
+        return getDatabaseTemplates(query);
+    }
+
     private Map<Integer, List<TranslationAttributeModel>> readTranslationAttributes(StringBuilder attributesInsertQuery) throws IOException {
         Map<Integer, List<TranslationAttributeModel>> translationAttributesMappedByTranslationId = new HashMap<>();
         BufferedReader br = readTranslationAttributesFile();
@@ -190,6 +244,23 @@ private Map<Integer, List<TranslationAttributeModel>> readTranslationAttributes(
         return translationAttributesMappedByTranslationId;
     }
 
+    private void readDatabaseTemplate(StringBuilder databaseTemplateInsertQuery) throws IOException {
+        BufferedReader br = readDatabaseTemplateFile();
+        String line = "";
+        while ((line = br.readLine()) != null)
+        {
+            String[] translation = line.split(DELIMITER);
+            DatabaseTemplateModel databaseTemplateModel = new DatabaseTemplateModel();
+            databaseTemplateModel.setId(Integer.valueOf(translation[0]));
+            databaseTemplateModel.setDatabaseType(translation[1]);
+            databaseTemplateModel.setTemplateName(translation[2]);
+            databaseTemplateModel.setTemplateFile(translation[3]);
+
+            String insertStatement = databaseTemplateModel.generateInsertStatement(DATABASE_TEMPLATE_TABLE_NAME);
+            databaseTemplateInsertQuery.append(insertStatement);
+        }
+    }
+
     private BufferedReader readTranslationMatrixFile() throws FileNotFoundException {
         //Check for the translation file from the ENV variable EXTERNAL_TRANSLATION_FILE
         String externalTranslationFile = System.getenv(EXTERNAL_TRANSLATION_FILE_ENV);
@@ -217,4 +288,10 @@ private BufferedReader readTranslationAttributesFile() throws FileNotFoundExcept
         InputStream resourceAsStream = TranslationMatrix.class.getClassLoader().getResourceAsStream(DEFAULT_TRANSLATION_ATTRIBUTE_FILE);
         return new BufferedReader(new InputStreamReader(resourceAsStream));
     }
+
+    private BufferedReader readDatabaseTemplateFile() throws FileNotFoundException {
+        //If the file is not provided in the project read it from the resources
+        InputStream resourceAsStream = TranslationMatrix.class.getClassLoader().getResourceAsStream(DEFAULT_DATABASE_TEMPLATE_FILE);
+        return new BufferedReader(new InputStreamReader(resourceAsStream));
+    }
 }

common/src/main/java/com/adaptivescale/rosetta/common/models/DatabaseTemplateModel.java

@@ -0,0 +1,66 @@
+package com.adaptivescale.rosetta.common.models;
+
+public class DatabaseTemplateModel {
+
+    private Integer id;
+
+    private String databaseType;
+
+    private String templateName;
+
+    private String templateFile;
+
+    public Integer getId() {
+        return id;
+    }
+
+    public void setId(Integer id) {
+        this.id = id;
+    }
+
+    public String getDatabaseType() {
+        return databaseType;
+    }
+
+    public void setDatabaseType(String databaseType) {
+        this.databaseType = databaseType;
+    }
+
+    public String getTemplateName() {
+        return templateName;
+    }
+
+    public void setTemplateName(String templateName) {
+        this.templateName = templateName;
+    }
+
+    public String getTemplateFile() {
+        return templateFile;
+    }
+
+    public void setTemplateFile(String templateFile) {
+        this.templateFile = templateFile;
+    }
+
+    public String generateInsertStatement(String tableName) {
+        StringBuilder builder = new StringBuilder();
+        builder.append("insert into ").append(tableName)
+                .append(" values (")
+                .append(id)
+                .append(", '").append(databaseType).append("' ")
+                .append(", '").append(templateName).append("' ")
+                .append(", '").append(templateFile).append("' ")
+                .append(");");
+        return builder.toString();
+    }
+
+    @Override
+    public String toString() {
+        return "DatabaseTemplateModel{" +
+                "id=" + id +
+                ", databaseType='" + databaseType + '\'' +
+                ", templateName='" + templateName + '\'' +
+                ", templateFile='" + templateFile + '\'' +
+                '}';
+    }
+}

ddl/src/main/java/com/adaptivescale/rosetta/ddl/DatabaseTemplateEnum.java

@@ -0,0 +1,27 @@
+package com.adaptivescale.rosetta.ddl;
+
+public enum DatabaseTemplateEnum {
+
+    SCHEMA_CREATE("schema_create"),
+    TABLE_CREATE("table_create"),
+    TABLE_ALTER("table_alter"),
+    TABLE_ALTER_DROP_PRIMARY_KEY("table_alter_drop_primary_key"),
+    TABLE_ALTER_ADD_PRIMARY_KEY("table_alter_add_primary_key"),
+    TABLE_DROP("table_drop"),
+    FOREIGNKEY_CREATE("foreignkey_create"),
+    FOREIGNKEY_DROP("foreignkey_drop"),
+    COLUMN_ADD("column_add"),
+    COLUMN_ALTER_TYPE("column_alter_type"),
+    COLUMN_ALTER_NULL("column_alter_null"),
+    COLUMN_DROP("column_drop");
+
+    private String name;
+
+    DatabaseTemplateEnum(String name) {
+        this.name = name;
+    }
+
+    public String getName() {
+        return name;
+    }
+}