Initial Data Model for Schema SQL Generation

pom.xml

@@ -5,7 +5,7 @@
 
 	<groupId>org.springframework.data</groupId>
 	<artifactId>spring-data-relational-parent</artifactId>
-	<version>3.1.0-SNAPSHOT</version>
+	<version>3.1.0-1478-SNAPSHOT</version>
 	<packaging>pom</packaging>
 
 	<name>Spring Data Relational Parent</name>

spring-data-jdbc-distribution/pom.xml

@@ -14,7 +14,7 @@
 	<parent>
 		<groupId>org.springframework.data</groupId>
 		<artifactId>spring-data-relational-parent</artifactId>
-		<version>3.1.0-SNAPSHOT</version>
+		<version>3.1.0-1478-SNAPSHOT</version>
 		<relativePath>../pom.xml</relativePath>
 	</parent>
 

spring-data-jdbc/pom.xml

@@ -6,7 +6,7 @@
 	<modelVersion>4.0.0</modelVersion>
 
 	<artifactId>spring-data-jdbc</artifactId>
-	<version>3.1.0-SNAPSHOT</version>
+	<version>3.1.0-1478-SNAPSHOT</version>
 
 	<name>Spring Data JDBC</name>
 	<description>Spring Data module for JDBC repositories.</description>
@@ -15,7 +15,7 @@
 	<parent>
 		<groupId>org.springframework.data</groupId>
 		<artifactId>spring-data-relational-parent</artifactId>
-		<version>3.1.0-SNAPSHOT</version>
+		<version>3.1.0-1478-SNAPSHOT</version>
 	</parent>
 
 	<properties>

spring-data-r2dbc/pom.xml

@@ -6,7 +6,7 @@
 	<modelVersion>4.0.0</modelVersion>
 
 	<artifactId>spring-data-r2dbc</artifactId>
-	<version>3.1.0-SNAPSHOT</version>
+	<version>3.1.0-1478-SNAPSHOT</version>
 
 	<name>Spring Data R2DBC</name>
 	<description>Spring Data module for R2DBC</description>
@@ -15,7 +15,7 @@
 	<parent>
 		<groupId>org.springframework.data</groupId>
 		<artifactId>spring-data-relational-parent</artifactId>
-		<version>3.1.0-SNAPSHOT</version>
+		<version>3.1.0-1478-SNAPSHOT</version>
 	</parent>
 
 	<properties>

spring-data-relational/pom.xml

@@ -6,15 +6,15 @@
 	<modelVersion>4.0.0</modelVersion>
 
 	<artifactId>spring-data-relational</artifactId>
-	<version>3.1.0-SNAPSHOT</version>
+	<version>3.1.0-1478-SNAPSHOT</version>
 
 	<name>Spring Data Relational</name>
 	<description>Spring Data Relational support</description>
 
 	<parent>
 		<groupId>org.springframework.data</groupId>
 		<artifactId>spring-data-relational-parent</artifactId>
-		<version>3.1.0-SNAPSHOT</version>
+		<version>3.1.0-1478-SNAPSHOT</version>
 	</parent>
 
 	<properties>

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/RelationalMappingContext.java

@@ -22,6 +22,8 @@
 import org.springframework.data.util.TypeInformation;
 import org.springframework.util.Assert;
 
+import java.util.Iterator;
+
 /**
  * {@link MappingContext} implementation.
  *
@@ -101,5 +103,4 @@ protected RelationalPersistentProperty createPersistentProperty(Property propert
 	public NamingStrategy getNamingStrategy() {
 		return this.namingStrategy;
 	}
-
 }

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/schemasqlgeneration/BaseTypeMapper.java

@@ -0,0 +1,34 @@
+/*
+ * Copyright 2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.relational.core.mapping.schemasqlgeneration;
+
+import java.util.HashMap;
+
+public class BaseTypeMapper {
+
+    final HashMap<Class<?>,String> mapClassToDatabaseType = new HashMap<Class<?>,String>();
+
+    public BaseTypeMapper() {
+        mapClassToDatabaseType.put(String.class, "VARCHAR(255)");
+        mapClassToDatabaseType.put(Boolean.class, "TINYINT");
+        mapClassToDatabaseType.put(Double.class, "DOUBLE");
+        mapClassToDatabaseType.put(Float.class, "FLOAT");
+        mapClassToDatabaseType.put(Integer.class, "INT");
+    }
+    public String databaseTypeFromClass(Class<?> type) {
+        return mapClassToDatabaseType.get(type);
+    }
+}

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/schemasqlgeneration/ColumnModel.java

@@ -0,0 +1,58 @@
+/*
+ * Copyright 2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.relational.core.mapping.schemasqlgeneration;
+
+import org.springframework.data.relational.core.sql.SqlIdentifier;
+
+import java.io.Serial;
+import java.io.Serializable;
+
+/**
+ * Class that models a Column for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class ColumnModel implements Serializable {
+    @Serial
+    private static final long serialVersionUID = 1L;
+    private final SqlIdentifier name;
+    private final String type;
+    private final boolean nullable;
+
+    public ColumnModel(SqlIdentifier name, String type, boolean nullable) {
+        this.name = name;
+        this.type = type;
+        this.nullable = nullable;
+    }
+
+    public ColumnModel(SqlIdentifier name, String type) {
+        this.name = name;
+        this.type = type;
+        this.nullable = false;
+    }
+
+    public SqlIdentifier getName() {
+        return name;
+    }
+
+    public String getType() {
+        return type;
+    }
+
+    public boolean isNullable() {
+        return nullable;
+    }
+}

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/schemasqlgeneration/ForeignKeyColumnModel.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright 2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.relational.core.mapping.schemasqlgeneration;
+
+import java.io.Serial;
+import java.io.Serializable;
+
+/**
+ * Class that models a Foreign Key relationship for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class ForeignKeyColumnModel implements Serializable {
+    @Serial
+    private static final long serialVersionUID = 1L;
+    private final TableModel foreignTable;
+    private final ColumnModel foreignColumn;
+    private final ColumnModel column;
+
+    public ForeignKeyColumnModel(TableModel foreignTable, ColumnModel foreignColumn, ColumnModel column) {
+        this.foreignTable = foreignTable;
+        this.foreignColumn = foreignColumn;
+        this.column = column;
+    }
+
+    public TableModel getForeignTable() {
+        return foreignTable;
+    }
+
+    public ColumnModel getForeignColumn() {
+        return foreignColumn;
+    }
+
+    public ColumnModel getColumn() {
+        return column;
+    }
+}

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/schemasqlgeneration/SchemaSQLGenerationDataModel.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright 2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.relational.core.mapping.schemasqlgeneration;
+
+import org.springframework.data.relational.core.mapping.BasicRelationalPersistentProperty;
+import org.springframework.data.relational.core.mapping.Column;
+import org.springframework.data.relational.core.mapping.RelationalMappingContext;
+import org.springframework.data.relational.core.mapping.RelationalPersistentEntity;
+
+import java.io.Serial;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+
+/**
+ * Model class that contains Table/Column information that can be used
+ * to generate SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class SchemaSQLGenerationDataModel implements Serializable {
+    @Serial
+    private static final long serialVersionUID = 1L;
+    private final List<TableModel> tableData = new ArrayList<TableModel>();
+    BaseTypeMapper typeMapper;
+
+    /**
+     * Default constructor so that we can deserialize a model
+     */
+    public SchemaSQLGenerationDataModel() {
+    }
+
+    /**
+     * Create model from a RelationalMappingContext
+     */
+    public SchemaSQLGenerationDataModel(RelationalMappingContext context) {
+
+        if (typeMapper == null) {
+            typeMapper = new BaseTypeMapper();
+        }
+
+        for (RelationalPersistentEntity entity : context.getPersistentEntities()) {
+            TableModel tableModel = new TableModel(entity.getTableName());
+
+            Iterator<BasicRelationalPersistentProperty> iter =
+                    entity.getPersistentProperties(Column.class).iterator();
+
+            while (iter.hasNext()) {
+                BasicRelationalPersistentProperty p = iter.next();
+                ColumnModel columnModel = new ColumnModel(p.getColumnName(),
+                        typeMapper.databaseTypeFromClass(p.getActualType()),
+                        true);
+                tableModel.getColumns().add(columnModel);
+            }
+            tableData.add(tableModel);
+        }
+    }
+
+    public List<TableModel> getTableData() {
+        return tableData;
+    }
+}

spring-data-relational/src/main/java/org/springframework/data/relational/core/mapping/schemasqlgeneration/TableModel.java

@@ -0,0 +1,66 @@
+/*
+ * Copyright 2023 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.relational.core.mapping.schemasqlgeneration;
+
+import org.springframework.data.relational.core.sql.SqlIdentifier;
+
+import java.io.Serial;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Class that models a Table for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class TableModel implements Serializable {
+    @Serial
+    private static final long serialVersionUID = 1L;
+    private final String schema;
+    private final SqlIdentifier name;
+    private final List<ColumnModel> columns = new ArrayList<ColumnModel>();
+    private final List<ColumnModel> keyColumns = new ArrayList<ColumnModel>();
+    private final List<ForeignKeyColumnModel> foreignKeyColumns = new ArrayList<ForeignKeyColumnModel>();
+
+    public TableModel(String schema, SqlIdentifier name) {
+        this.schema = schema;
+        this.name = name;
+    }
+    public TableModel(SqlIdentifier name) {
+        this(null, name);
+    }
+
+    public String getSchema() {
+        return schema;
+    }
+
+    public SqlIdentifier getName() {
+        return name;
+    }
+
+    public List<ColumnModel> getColumns() {
+        return columns;
+    }
+
+    public List<ColumnModel> getKeyColumns() {
+        return keyColumns;
+    }
+
+    public List<ForeignKeyColumnModel> getForeignKeyColumns() {
+        return foreignKeyColumns;
+    }
+}