From 5d974ea790397d0646f013e91c8d3b0829ffbd54 Mon Sep 17 00:00:00 2001
From: Anton Tananaev <anton.tananaev@gmail.com>
Date: Fri, 18 Dec 2015 16:32:46 +1300
Subject: Finish Liquibase migration implementation

---
 src/org/traccar/database/DataManager.java | 64 +++++++------------------------
 1 file changed, 13 insertions(+), 51 deletions(-)

(limited to 'src')

diff --git a/src/org/traccar/database/DataManager.java b/src/org/traccar/database/DataManager.java
index f905b4ded..37cbac35d 100644
--- a/src/org/traccar/database/DataManager.java
+++ b/src/org/traccar/database/DataManager.java
@@ -19,12 +19,8 @@ import com.mchange.v2.c3p0.ComboPooledDataSource;
 import java.io.File;
 import java.net.URL;
 import java.net.URLClassLoader;
-import java.nio.file.Path;
-import java.nio.file.Paths;
-import java.sql.Connection;
 import java.sql.Driver;
 import java.sql.DriverManager;
-import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.Collection;
 import java.util.Date;
@@ -34,10 +30,11 @@ import javax.naming.InitialContext;
 import javax.sql.DataSource;
 
 import liquibase.Contexts;
+import liquibase.LabelExpression;
 import liquibase.Liquibase;
+import liquibase.changelog.ChangeSetStatus;
 import liquibase.database.Database;
 import liquibase.database.DatabaseFactory;
-import liquibase.exception.DatabaseException;
 import liquibase.exception.LiquibaseException;
 import liquibase.resource.FileSystemResourceAccessor;
 import liquibase.resource.ResourceAccessor;
@@ -48,7 +45,6 @@ import org.traccar.model.Device;
 import org.traccar.model.MiscFormatter;
 import org.traccar.model.Permission;
 import org.traccar.model.Position;
-import org.traccar.model.Schema;
 import org.traccar.model.Server;
 import org.traccar.model.User;
 import org.traccar.web.AsyncServlet;
@@ -160,9 +156,9 @@ public class DataManager implements IdentityManager {
 
     private void initDatabaseSchema() throws SQLException, LiquibaseException {
 
-        if (config.getString("web.type", "new").equals("new") || config.getString("web.type", "new").equals("api")) {
+        if (config.hasKey("database.changelog")) {
 
-            /*ResourceAccessor resourceAccessor = new FileSystemResourceAccessor();
+            ResourceAccessor resourceAccessor = new FileSystemResourceAccessor();
 
             Database database = DatabaseFactory.getInstance().openDatabase(
                     config.getString("database.url"),
@@ -173,46 +169,17 @@ public class DataManager implements IdentityManager {
             Liquibase liquibase = new Liquibase(
                     config.getString("database.changelog"), resourceAccessor, database);
 
-            liquibase.update(new Contexts());*/
-
-
-            boolean exist = false;
-
-            try (Connection connection = dataSource.getConnection();
-                 ResultSet result = connection.getMetaData().getTables(connection.getCatalog(), null, null, null)) {
-
-                String checkTable = config.getString("database.checkTable");
-                while (result.next()) {
-                    if (result.getString("TABLE_NAME").equalsIgnoreCase(checkTable)) {
-                        exist = true;
-                        break;
-                    }
+            boolean first = true;
+            for (ChangeSetStatus status : liquibase.getChangeSetStatuses(null, new LabelExpression())) {
+                if (!status.getWillRun()) {
+                    first = false;
+                    break;
                 }
             }
 
-            if (exist) {
-
-                String schemaVersionQuery = getQuery("database.selectSchemaVersion");
-                if (schemaVersionQuery != null) {
-
-                    Schema schema = QueryBuilder.create(dataSource, schemaVersionQuery)
-                            .executeQuerySingle(Schema.class);
-
-                    int version = 0;
-                    if (schema != null) {
-                        version = schema.getVersion();
-                    }
-
-                    if (version != 302) {
-                        Log.error("Wrong database schema version (" + version + ")");
-                        throw new RuntimeException();
-                    }
-                }
-
-            } else {
-
-                QueryBuilder.create(dataSource, getQuery("database.createSchema")).executeUpdate();
+            liquibase.update(new Contexts());
 
+            if (first) {
                 User admin = new User();
                 admin.setName("admin");
                 admin.setEmail("admin");
@@ -220,14 +187,7 @@ public class DataManager implements IdentityManager {
                 admin.setPassword("admin");
                 addUser(admin);
 
-                Server server = new Server();
-                server.setRegistration(true);
-                QueryBuilder.create(dataSource, getQuery("database.insertServer"))
-                        .setObject(server)
-                        .executeUpdate();
-
                 mockData(admin.getId());
-
             }
         }
     }
@@ -395,6 +355,7 @@ public class DataManager implements IdentityManager {
 
     public void addPosition(Position position) throws SQLException {
         position.setId(QueryBuilder.create(dataSource, getQuery("database.insertPosition"), true)
+                .setDate("now", new Date())
                 .setObject(position)
                 .setDate("time", position.getFixTime()) // tmp
                 .setLong("device_id", position.getDeviceId()) // tmp
@@ -406,6 +367,7 @@ public class DataManager implements IdentityManager {
 
     public void updateLatestPosition(Position position) throws SQLException {
         QueryBuilder.create(dataSource, getQuery("database.updateLatestPosition"))
+                .setDate("now", new Date())
                 .setObject(position)
                 .setDate("time", position.getFixTime()) // tmp
                 .setLong("device_id", position.getDeviceId()) // tmp
-- 
cgit v1.2.3