hal/src/se/koc/hal/HalContext.java
Ziver Koc 3c763dbc2b DB upgrade implemented
Former-commit-id: 351a040593e4ddab880d517d9e0854d8c8815fb5
2015-12-08 22:25:14 +01:00

115 lines
4 KiB
Java
Executable file

package se.koc.hal;
import zutil.db.DBConnection;
import zutil.db.DBUpgradeHandler;
import zutil.db.handler.PropertiesSQLResult;
import zutil.io.file.FileUtil;
import zutil.log.LogUtil;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Properties;
import java.util.logging.Logger;
public class HalContext {
private static final Logger logger = LogUtil.getLogger();
// Constants
private static final String PROPERTY_DB_VERSION = "db_version";
private static final String CONF_FILE = "hal.conf";
private static final String DB_FILE = "hal.db";
private static final String DEFAULT_DB_FILE = "hal-default.db";
// Variables
private static DBConnection db;
private static Properties defaultFileConf;
private static Properties fileConf;
private static Properties dbConf;
static {
defaultFileConf = new Properties();
defaultFileConf.setProperty("http_port", ""+8080);
defaultFileConf.setProperty("sync_port", ""+6666);
}
public static void initialize(){
try {
// Read conf
fileConf = new Properties(defaultFileConf);
FileReader in = new FileReader(CONF_FILE);
fileConf.load(in);
in.close();
// Init DB
File dbFile = FileUtil.find(DB_FILE);
if(dbFile == null){
logger.info("Creating new DB...");
FileUtil.copy(dbFile, FileUtil.find(DEFAULT_DB_FILE));
}
db = new DBConnection(DBConnection.DBMS.SQLite, DB_FILE);
// Read DB conf
dbConf = db.exec("SELECT * FROM conf", new PropertiesSQLResult());
// Upgrade DB needed?
DBConnection referenceDB = new DBConnection(DBConnection.DBMS.SQLite, DEFAULT_DB_FILE);
Properties defaultDBConf =
referenceDB.exec("SELECT * FROM conf", new PropertiesSQLResult());
// Check DB version
logger.fine("DB version: "+ dbConf.getProperty(PROPERTY_DB_VERSION));
if(dbConf.getProperty(PROPERTY_DB_VERSION) == null ||
defaultDBConf.getProperty(PROPERTY_DB_VERSION).compareTo(dbConf.getProperty(PROPERTY_DB_VERSION)) > 0) {
logger.info("Starting DB upgrade...");
File backupDB = FileUtil.getNextFile(dbFile);
logger.fine("Backing up DB to: "+ backupDB);
FileUtil.copy(dbFile, backupDB);
logger.fine("Upgrading DB (from: v"+dbConf.getProperty(PROPERTY_DB_VERSION)
+", to: v"+defaultDBConf.getProperty(PROPERTY_DB_VERSION) +")...");
DBUpgradeHandler handler = new DBUpgradeHandler(referenceDB);
handler.setTargetDB(db);
handler.upgrade();
logger.info("DB upgrade done");
dbConf.setProperty(PROPERTY_DB_VERSION, defaultDBConf.getProperty(PROPERTY_DB_VERSION));
storeProperties();
}
referenceDB.close();
} catch (Exception e){
throw new RuntimeException(e);
}
}
public static String getStringProperty(String key){
String value = fileConf.getProperty(key);
if(value == null)
value = dbConf.getProperty(key);
return value;
}
public static int getIntegerProperty(String key){
return Integer.parseInt(getStringProperty(key));
}
public synchronized static void storeProperties() throws SQLException {
logger.fine("Saving conf to DB...");
PreparedStatement stmt = db.getPreparedStatement("REPLACE INTO conf (key, value) VALUES (?, ?)");
for(Object key : dbConf.keySet()){
stmt.setObject(1, key);
stmt.setObject(2, dbConf.get(key));
stmt.addBatch();
}
DBConnection.execBatch(stmt);
}
public static DBConnection getDB(){
return db;
}
}