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

@@ -19,9 +19,14 @@
 import org.springframework.data.mapping.context.MappingContext;
 import org.springframework.data.mapping.model.Property;
 import org.springframework.data.mapping.model.SimpleTypeHolder;
+import org.springframework.data.relational.core.mapping.schemasqlgeneration.ColumnModel;
+import org.springframework.data.relational.core.mapping.schemasqlgeneration.SchemaSQLGenerationDataModel;
+import org.springframework.data.relational.core.mapping.schemasqlgeneration.TableModel;
 import org.springframework.data.util.TypeInformation;
 import org.springframework.util.Assert;
 
+import java.util.Iterator;
+
 /**
  * {@link MappingContext} implementation.
  *
@@ -102,4 +107,31 @@ public NamingStrategy getNamingStrategy() {
 		return this.namingStrategy;
 	}
 
+	/**
+	 * Method to build a data model of Tables/Columns and their relationships
+	 * that will be used in SQL Generation.
+	 */
+	public SchemaSQLGenerationDataModel generateSchemaSQL() {
+
+		SchemaSQLGenerationDataModel model = new SchemaSQLGenerationDataModel();
+
+		for (RelationalPersistentEntity entity : getPersistentEntities()) {
+			TableModel tableModel = new TableModel();
+			tableModel.setName(entity.getTableName().getReference());
+
+			Iterator<BasicRelationalPersistentProperty> iter =
+					entity.getPersistentProperties(Column.class).iterator();
+
+			while (iter.hasNext()) {
+				BasicRelationalPersistentProperty p = iter.next();
+				ColumnModel columnModel = new ColumnModel();
+				columnModel.setName(p.getColumnName().getReference());
+				columnModel.setType(p.getActualType());
+				tableModel.getColumns().add(columnModel);
+			}
+			model.getTableData().add(tableModel);
+		}
+
+		return model;
+	}
 }

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

@@ -0,0 +1,51 @@
+/*
+ * 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;
+
+/**
+ * Class that models a Column for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class ColumnModel {
+    private String name;
+    private Class<?> type;
+    private boolean nullable;
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public Class<?> getType() {
+        return type;
+    }
+
+    public void setType(Class<?> type) {
+        this.type = type;
+    }
+
+    public boolean isNullable() {
+        return nullable;
+    }
+
+    public void setNullable(boolean nullable) {
+        this.nullable = nullable;
+    }
+}

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

@@ -0,0 +1,45 @@
+/*
+ * 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;
+
+/**
+ * Class that models a Foreign Key relationship for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class ForeignKeyColumnModel {
+    private TableModel foreignTable;
+    private ColumnModel foreignColumn;
+    private 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,37 @@
+/*
+ * 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.ArrayList;
+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 {
+    private List<TableModel> tableData = new ArrayList<TableModel>();
+
+    public List<TableModel> getTableData() {
+        return tableData;
+    }
+
+    public void setTableData(List<TableModel> tableData) {
+        this.tableData = tableData;
+    }
+}

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

@@ -0,0 +1,72 @@
+/*
+ * 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.ArrayList;
+import java.util.List;
+
+/**
+ * Class that models a Table for generating SQL for Schema generation.
+ *
+ * @author Kurt Niemi
+ */
+public class TableModel {
+    private String schema;
+    private String name;
+    private List<ColumnModel> columns = new ArrayList<ColumnModel>();
+    private List<ColumnModel> keyColumns = new ArrayList<ColumnModel>();
+    private List<ForeignKeyColumnModel> foreignKeyColumns = new ArrayList<ForeignKeyColumnModel>();
+
+    public String getSchema() {
+        return schema;
+    }
+
+    public void setSchema(String schema) {
+        this.schema = schema;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public List<ColumnModel> getColumns() {
+        return columns;
+    }
+
+    public void setColumns(List<ColumnModel> columns) {
+        this.columns = columns;
+    }
+
+    public List<ColumnModel> getKeyColumns() {
+        return keyColumns;
+    }
+
+    public void setKeyColumns(List<ColumnModel> keyColumns) {
+        this.keyColumns = keyColumns;
+    }
+
+    public List<ForeignKeyColumnModel> getForeignKeyColumns() {
+        return foreignKeyColumns;
+    }
+
+    public void setForeignKeyColumns(List<ForeignKeyColumnModel> foreignKeyColumns) {
+        this.foreignKeyColumns = foreignKeyColumns;
+    }
+}