void initDatabase() {
try {
// Connect to the database.
if (conn == null) {
dbbindingLogger.debug("DBConnector: Connecting to " + dbName);
if (dbType.equals(DBType.MEMORY)) {
// In-memory embedded database.
Class.forName("org.apache.derby.jdbc.EmbeddedDriver").newInstance();
conn = DriverManager.getConnection("jdbc:derby:memory:" + dbName + ";create=true");
} else {
// External database.
if (jdbcDriver != null) Class.forName(jdbcDriver).newInstance();
conn = DriverManager.getConnection(makeURL(), username, password);
}
}
// Create a JDBC statement.
if (stmt == null) {
stmt = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);
}
// Create missing tables.
DatabaseMetaData meta = conn.getMetaData();
for (Entry<String, String> e : createTableStatements.entrySet()) {
String table = e.getKey();
// Check if table exists
ResultSet rs = meta.getTables(null, null, table.toUpperCase(), null);
if (!rs.next()) stmt.execute(e.getValue()); // Table doesn't exist so create it.
}
} catch (IllegalAccessException ej) {
dbbindingLogger.error("Failed to start up the database, IllgealAccessException: " + ej);
exceptionLogger.error("Failed to start up the database, IllgealAccessException: " + ej);
} catch (InstantiationException ek) {
dbbindingLogger.error("Failed to start up the database, InstantiationException: " + ek);
exceptionLogger.error("Failed to start up the database, InstantiationException: " + ek);
} catch (ClassNotFoundException el) {
dbbindingLogger.error("Failed to start up the database, ClassNotFoundException: " + el);
exceptionLogger.error("Failed to start up the database, ClassNotFoundException: " + el);
} catch (SQLException em) {
dbbindingLogger.error("Failed to start up the database: " + em);
exceptionLogger.error("Failed to start up the database: " + em);
}
}
@Override
public void write(DataOutput out) throws IOException {
int numCol = colType.length;
boolean[] nullBits = new boolean[numCol];
int[] colLength = new int[numCol];
byte[] enumType = new byte[numCol];
int[] padLength = new int[numCol];
byte[] padbytes = new byte[8];
/**
* Compute the total payload and header length header = total length (4 byte), Version (2 byte),
* Error (1 byte), #col (2 byte) col type array = #col * 1 byte null bit array = ceil(#col/8)
*/
int datlen = 4 + 2 + 1 + 2;
datlen += numCol;
datlen += getNullByteArraySize(numCol);
for (int i = 0; i < numCol; i++) {
/* Get the enum type */
DBType coldbtype;
switch (DataType.get(colType[i])) {
case BIGINT:
coldbtype = DBType.BIGINT;
break;
case BOOLEAN:
coldbtype = DBType.BOOLEAN;
break;
case FLOAT8:
coldbtype = DBType.FLOAT8;
break;
case INTEGER:
coldbtype = DBType.INTEGER;
break;
case REAL:
coldbtype = DBType.REAL;
break;
case SMALLINT:
coldbtype = DBType.SMALLINT;
break;
case BYTEA:
coldbtype = DBType.BYTEA;
break;
default:
coldbtype = DBType.TEXT;
}
enumType[i] = (byte) (coldbtype.ordinal());
/* Get the actual value, and set the null bit */
if (colValue[i] == null) {
nullBits[i] = true;
colLength[i] = 0;
} else {
nullBits[i] = false;
/*
* For fixed length type, we get the fixed length.
* For var len binary format, the length is in the col value.
* For text format, we must convert encoding first.
*/
if (!coldbtype.isVarLength()) {
colLength[i] = coldbtype.getTypeLength();
} else if (!isTextForm(colType[i])) {
colLength[i] = ((byte[]) colValue[i]).length;
} else {
colLength[i] = ((String) colValue[i]).getBytes(CHARSET).length;
}
/* calculate and add the type alignment padding */
padLength[i] = roundUpAlignment(datlen, coldbtype.getAlignment()) - datlen;
datlen += padLength[i];
/* for variable length type, we add a 4 byte length header */
if (coldbtype.isVarLength()) {
datlen += 4;
}
}
datlen += colLength[i];
}
/*
* Add the final alignment padding for the next record
*/
int endpadding = roundUpAlignment(datlen, 8) - datlen;
datlen += endpadding;
/* Construct the packet header */
out.writeInt(datlen);
out.writeShort(VERSION);
out.writeByte(errorFlag);
out.writeShort(numCol);
/* Write col type */
for (int i = 0; i < numCol; i++) {
out.writeByte(enumType[i]);
}
/* Nullness */
byte[] nullBytes = boolArrayToByteArray(nullBits);
out.write(nullBytes);
/* Column Value */
for (int i = 0; i < numCol; i++) {
if (!nullBits[i]) {
/* Pad the alignment byte first */
if (padLength[i] > 0) {
out.write(padbytes, 0, padLength[i]);
}
/* Now, write the actual column value */
switch (DataType.get(colType[i])) {
case BIGINT:
out.writeLong(((Long) colValue[i]));
break;
case BOOLEAN:
out.writeBoolean(((Boolean) colValue[i]));
break;
case FLOAT8:
out.writeDouble(((Double) colValue[i]));
break;
case INTEGER:
out.writeInt(((Integer) colValue[i]));
break;
case REAL:
out.writeFloat(((Float) colValue[i]));
break;
case SMALLINT:
out.writeShort(((Short) colValue[i]));
break;
/* For BYTEA format, add 4byte length header at the beginning */
case BYTEA:
out.writeInt(colLength[i]);
out.write((byte[]) colValue[i]);
break;
/* For text format, add 4byte length header. string is already '\0' terminated */
default:
{
out.writeInt(colLength[i]);
byte[] data = ((String) colValue[i]).getBytes(CHARSET);
out.write(data);
break;
}
}
}
}
/* End padding */
out.write(padbytes, 0, endpadding);
}
/**
* Starts up the database. This must be called before calling methods to get connection of execute
* queries or updates.
*/
public void startup() {
dbbindingLogger.debug("Begin to start up the database.");
Properties prop = new Properties();
try {
prop.load(new FileInputStream(dbpropfile));
try {
dbType = DBType.MEMORY; // default value.
dbType = DBType.valueOf(prop.getProperty("type"));
} catch (IllegalArgumentException e) {
dbbindingLogger.warn("Bad database type in the property file.");
exceptionLogger.warn(
"Here is an exception : ", new Exception("Bad database type in the property file."));
} catch (NullPointerException e) {
dbbindingLogger.warn("Missing database type in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database type in the property file."));
}
database = prop.getProperty("database");
if (database == null) {
dbbindingLogger.warn("Missing database key in the property file.");
exceptionLogger.warn(
"Here is an exception : ", new Exception("Missing database key in the property file."));
}
jdbcDriver = prop.getProperty("jdbc.driver");
if (jdbcDriver == null) {
dbbindingLogger.warn("Missing jdbcDriver key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing jdbcDriver key in the property file."));
}
jdbcURL = prop.getProperty("database.url");
if (jdbcURL == null) {
dbbindingLogger.warn("Missing database URL key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database URL key in the property file."));
}
dbHost = prop.getProperty("database.host");
if (dbHost == null) {
dbbindingLogger.warn("Missing database host key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database host key in the property file."));
}
dbPort = prop.getProperty("database.port");
if (dbPort == null) {
dbbindingLogger.warn("Missing database port key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database port key in the property file."));
}
dbName = prop.getProperty("database.name");
if (dbName == null) {
dbbindingLogger.warn("Missing database name key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database name key in the property file."));
}
username = prop.getProperty("username");
if (username == null) {
dbbindingLogger.warn("Missing database username key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database username key in the property file."));
}
password = prop.getProperty("password");
if (password == null) {
dbbindingLogger.warn("Missing database password key in the property file.");
exceptionLogger.warn(
"Here is an exception : ",
new Exception("Missing database password key in the property file."));
}
if (dbbindingLogger.isDebugEnabled()) {
dbbindingLogger.debug(
"Parse: "
+ database
+ ","
+ jdbcDriver
+ ","
+ jdbcURL
+ dbHost
+ ","
+ dbPort
+ ","
+ dbName
+ ","
+ username
+ ","
+ password);
}
// Make connection to database and initialize the database.
initDatabase();
// TODO: support prepared statement if needed
} catch (IOException ei) {
dbbindingLogger.error("Failed to start up the database, IOException: " + ei);
exceptionLogger.error("Failed to start up the database, IOException: " + ei);
}
}