SQLiteColumnQueryBuilder

Author: ZorTik

api/src/main/java/me/zort/sqllib/api/model/TableSchema.java

@@ -35,6 +35,14 @@ public String[] getDefinitions() {
         return definitions;
     }
 
+    public String[] getDefinitionNames() {
+        String[] definitions = new String[this.definitions.length];
+        for (int i = 0; i < definitions.length; i++) {
+            definitions[i] = getDefinitionName(i);
+        }
+        return definitions;
+    }
+
     public String getTable() {
         return table;
     }

core/src/main/java/me/zort/sqllib/SQLConnectionBuilder.java

@@ -98,7 +98,7 @@ public SQLConnectionBuilder(@Nullable SQLEndpoint endpoint) {
         if (driver == null) driver = SQLDatabaseConnectionImpl.DEFAULT_DRIVER;
         SQLConnectionFactory connectionFactory = new LocalConnectionFactory(driver);
         return jdbc.contains("jdbc:sqlite")
-                ? new SQLiteDatabaseConnectionImpl(connectionFactory, options)
+                ? new SQLiteDatabaseConnection(connectionFactory, options)
                 : new SQLDatabaseConnectionImpl(connectionFactory, options);
     }
 

core/src/main/java/me/zort/sqllib/SQLDatabaseConnectionImpl.java

@@ -191,7 +191,7 @@ public boolean synchronizeModel() {
         return mappingRegistry.getProxyInstances()
                 .stream().flatMap(i -> i.getTableSchemas(
                         getOptions().getNamingStrategy(),
-                        this instanceof SQLiteDatabaseConnectionImpl).stream())
+                        this instanceof SQLiteDatabaseConnection).stream())
                 .anyMatch(schema -> synchronizeModel(schema, schema.getTable()));
     }
 
@@ -227,7 +227,7 @@ public boolean synchronizeModel(TableSchema entitySchema, String table) {
     public boolean synchronizeModel(Class<?> entity, String table) {
         return synchronizeModel(new EntitySchemaBuilder(table, entity,
                 getOptions().getNamingStrategy(),
-                this instanceof SQLiteDatabaseConnectionImpl).buildTableSchema(), table);
+                this instanceof SQLiteDatabaseConnection).buildTableSchema(), table);
     }
 
     /**
@@ -332,7 +332,7 @@ public final <T> T createProxy(final @NotNull Class<T> mappingInterface, final @
     public final boolean buildEntitySchema(final @NotNull String tableName, final @NotNull Class<?> entityClass) {
         Objects.requireNonNull(entityClass, "Entity class cannot be null!");
 
-        EntitySchemaBuilder converter = new EntitySchemaBuilder(tableName, entityClass, options.getNamingStrategy(), this instanceof SQLiteDatabaseConnectionImpl);
+        EntitySchemaBuilder converter = new EntitySchemaBuilder(tableName, entityClass, options.getNamingStrategy(), this instanceof SQLiteDatabaseConnection);
         String query = converter.buildTableQuery();
 
         return exec(() -> query).isSuccessful();

…sqllib/SQLiteDatabaseConnectionImpl.java …ort/sqllib/SQLiteDatabaseConnection.javacore/src/main/java/me/zort/sqllib/SQLiteDatabaseConnectionImpl.java renamed to core/src/main/java/me/zort/sqllib/SQLiteDatabaseConnection.java core/src/main/java/me/zort/sqllib/SQLiteDatabaseConnectionImpl.java renamed to core/src/main/java/me/zort/sqllib/SQLiteDatabaseConnection.java

@@ -8,6 +8,8 @@
 import me.zort.sqllib.internal.factory.SQLConnectionFactory;
 import me.zort.sqllib.internal.query.*;
 import me.zort.sqllib.internal.query.part.SetStatement;
+import me.zort.sqllib.model.column.SQLiteColumnQueryBuilder;
+import me.zort.sqllib.model.schema.SQLSchemaSynchronizer;
 import me.zort.sqllib.util.PrimaryKey;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -22,16 +24,24 @@
  *
  * @author ZorTik
  */
-public class SQLiteDatabaseConnectionImpl extends SQLDatabaseConnectionImpl {
-    private final SQLiteDatabaseConnectionImpl identity = this;
+public class SQLiteDatabaseConnection extends SQLDatabaseConnectionImpl {
+    private final SQLiteDatabaseConnection identity = this;
 
     @SuppressWarnings("unused")
-    public SQLiteDatabaseConnectionImpl(final @NotNull SQLConnectionFactory connectionFactory) {
+    public SQLiteDatabaseConnection(final @NotNull SQLConnectionFactory connectionFactory) {
         super(connectionFactory);
+        changeSQLiteColumnQueryBuilder();
     }
 
-    public SQLiteDatabaseConnectionImpl(final @NotNull SQLConnectionFactory connectionFactory, @Nullable ISQLDatabaseOptions options) {
+    public SQLiteDatabaseConnection(final @NotNull SQLConnectionFactory connectionFactory, @Nullable ISQLDatabaseOptions options) {
         super(connectionFactory, options);
+        changeSQLiteColumnQueryBuilder();
+    }
+
+    private void changeSQLiteColumnQueryBuilder() {
+        if (getSchemaSynchronizer() instanceof SQLSchemaSynchronizer) {
+            ((SQLSchemaSynchronizer) getSchemaSynchronizer()).setColumnQueryBuilder(new SQLiteColumnQueryBuilder(this));
+        }
     }
 
     /**

core/src/main/java/me/zort/sqllib/model/column/SQLiteColumnQueryBuilder.java

@@ -0,0 +1,23 @@
+package me.zort.sqllib.model.column;
+
+import lombok.AllArgsConstructor;
+import me.zort.sqllib.SQLiteDatabaseConnection;
+import me.zort.sqllib.api.model.ColumnDefinition;
+import me.zort.sqllib.api.model.TableSchema;
+
+@AllArgsConstructor
+public class SQLiteColumnQueryBuilder extends InnoColumnQueryBuilder {
+    private final SQLiteDatabaseConnection connection;
+    @Override
+    public String buildActionQuery(ColumnAction action, String table, ColumnDefinition from, ColumnDefinition to) {
+        if (action.equals(ColumnAction.MODIFY)) {
+            TableSchema schema = connection.getSchemaBuilder(table).buildTableSchema();
+            String queries = "ALTER TABLE" + table + " RENAME TO " + table + "_old;";
+            queries += "CREATE TABLE " + table + "(" + String.join(", ", schema.getDefinitions()) + ");";
+            queries += "INSERT INTO " + table + "(" + String.join(", ", schema.getDefinitionNames()) + ") SELECT " + String.join(", ", schema.getDefinitionNames()) + " FROM " + table + "_old;";
+            queries += "DROP TABLE " + table + "_old;";
+            return queries;
+        }
+        return super.buildActionQuery(action, table, from, to);
+    }
+}