private void loadTableData(
final File dbFile, final TableViewer guiTableViewer, final String dbTableName) {
SqlJetDb db = null;
try {
db = SqlJetDb.open(dbFile, true);
final ISqlJetTable dbTable = db.getTable(dbTableName);
final ArrayList<DataRow> data = new ArrayList<DataRow>();
final ISqlJetTableDef tableDef = dbTable.getDefinition();
final List<String> names = new ArrayList<String>();
for (final ISqlJetColumnDef column : tableDef.getColumns()) {
names.add(column.getName());
}
final String[] namesArray = names.toArray(new String[names.size()]);
final Table guiTable = guiTableViewer.getTable();
createGuiTableColumns(guiTable, namesArray);
dbTable
.getDataBase()
.runReadTransaction(
new ISqlJetTransaction() {
@Override
public Object run(final SqlJetDb db) throws SqlJetException {
final ISqlJetCursor cursor = dbTable.open();
try {
int count = 0;
while (!cursor.eof()) {
data.add(DataRow.read(cursor, count, namesArray));
cursor.next();
count++;
}
} finally {
cursor.close();
}
return null;
}
});
// Populate data and refresh table viewer
guiTableViewer.setInput(data);
guiTableViewer.refresh();
} catch (final Exception e) {
e.printStackTrace();
} finally {
if (db != null) {
try {
db.close();
} catch (final SqlJetException e) {
e.printStackTrace();
}
}
}
}
/**
* @param schemaTable
* @param tableDef
* @throws SqlJetException
*/
private void addConstraints(ISqlJetBtreeSchemaTable schemaTable, final SqlJetTableDef tableDef)
throws SqlJetException {
final String tableName = tableDef.getName();
final List<ISqlJetColumnDef> columns = tableDef.getColumns();
int i = 0;
for (final ISqlJetColumnDef column : columns) {
final List<ISqlJetColumnConstraint> constraints = column.getConstraints();
if (null == constraints) continue;
for (final ISqlJetColumnConstraint constraint : constraints) {
if (constraint instanceof ISqlJetColumnPrimaryKey) {
final ISqlJetColumnPrimaryKey pk = (ISqlJetColumnPrimaryKey) constraint;
if (!column.hasExactlyIntegerType()) {
if (pk.isAutoincremented()) {
throw new SqlJetException(
SqlJetErrorCode.ERROR,
"AUTOINCREMENT is allowed only for INTEGER PRIMARY KEY fields");
}
createAutoIndex(schemaTable, tableName, generateAutoIndexName(tableName, ++i));
} else if (pk.isAutoincremented()) {
checkSequenceTable();
}
} else if (constraint instanceof ISqlJetColumnUnique) {
createAutoIndex(schemaTable, tableName, generateAutoIndexName(tableName, ++i));
}
}
}
final List<ISqlJetTableConstraint> constraints = tableDef.getConstraints();
if (null != constraints) {
for (final ISqlJetTableConstraint constraint : constraints) {
if (constraint instanceof ISqlJetTablePrimaryKey) {
boolean b = false;
final ISqlJetTablePrimaryKey pk = (ISqlJetTablePrimaryKey) constraint;
if (pk.getColumns().size() == 1) {
final String n = pk.getColumns().get(0);
final ISqlJetColumnDef c = tableDef.getColumn(n);
b = c != null && c.hasExactlyIntegerType();
}
if (!b) {
createAutoIndex(schemaTable, tableName, generateAutoIndexName(tableName, ++i));
}
} else if (constraint instanceof ISqlJetTableUnique) {
createAutoIndex(schemaTable, tableName, generateAutoIndexName(tableName, ++i));
}
}
}
}
/**
* @param tableDef
* @throws SqlJetException
*/
private void checkFieldNamesRepeatsConflict(String tableName, List<ISqlJetColumnDef> columns)
throws SqlJetException {
final Set<String> names = new HashSet<String>();
for (ISqlJetColumnDef columnDef : columns) {
final String name = columnDef.getName();
if (names.contains(name)) {
throw new SqlJetException(
SqlJetErrorCode.ERROR,
String.format(
"Definition for table '%s' has conflict of repeating fields named '%s'",
tableName, name));
} else {
names.add(name);
}
}
}
public static TableModel createInstance(
final ISqlJetTable table, final long fromID, final int pageSize, final IProgress progress)
throws SqlJetException {
if (table == null) {
return new DefaultTableModel();
}
final ArrayList<DataRow> data = new ArrayList<DataRow>(pageSize);
final ISqlJetTableDef tableDef = table.getDefinition();
final List<String> names = new ArrayList<String>();
for (ISqlJetColumnDef column : tableDef.getColumns()) {
names.add(column.getName());
}
final String[] namesArray = (String[]) names.toArray(new String[names.size()]);
table
.getDataBase()
.runReadTransaction(
new ISqlJetTransaction() {
public Object run(SqlJetDb db) throws SqlJetException {
ISqlJetCursor cursor = table.open(); // order(table.getPrimaryKeyIndexName());
try {
for (long i = 0; i < fromID && !cursor.eof(); i++) {
cursor.next();
}
int count = 0;
while (!cursor.eof() && count < pageSize) {
data.add(DataRow.read(cursor, fromID + count, namesArray));
progress.current(count);
cursor.next();
count++;
}
} finally {
cursor.close();
}
return null;
}
});
return new DataTableModel(data, namesArray, fromID);
}
/**
* @param tableName
* @param newTableName
* @param newColumnDef
* @return
* @throws SqlJetException
*/
private ISqlJetTableDef alterTableSafe(final SqlJetAlterTableDef alterTableDef)
throws SqlJetException {
assert (null != alterTableDef);
String tableName = alterTableDef.getTableName();
String newTableName = alterTableDef.getNewTableName();
ISqlJetColumnDef newColumnDef = alterTableDef.getNewColumnDef();
if (null == tableName) {
throw new SqlJetException(SqlJetErrorCode.MISUSE, "Table name isn't defined");
}
if (null == newTableName && null == newColumnDef) {
throw new SqlJetException(SqlJetErrorCode.MISUSE, "Not defined any altering");
}
boolean renameTable = false;
if (null != newTableName) {
renameTable = true;
} else {
newTableName = tableName;
}
if (renameTable && tableDefs.containsKey(newTableName)) {
throw new SqlJetException(
SqlJetErrorCode.MISUSE, String.format("Table \"%s\" already exists", newTableName));
}
final SqlJetTableDef tableDef = (SqlJetTableDef) tableDefs.get(tableName);
if (null == tableDef) {
throw new SqlJetException(
SqlJetErrorCode.MISUSE, String.format("Table \"%s\" not found", tableName));
}
List<ISqlJetColumnDef> columns = tableDef.getColumns();
if (null != newColumnDef) {
final String fieldName = newColumnDef.getName();
if (tableDef.getColumn(fieldName) != null) {
throw new SqlJetException(
SqlJetErrorCode.MISUSE,
String.format("Field \"%s\" already exists in table \"%s\"", fieldName, tableName));
}
final List<ISqlJetColumnConstraint> constraints = newColumnDef.getConstraints();
if (null != constraints && 0 != constraints.size()) {
boolean notNull = false;
boolean defaultValue = false;
for (final ISqlJetColumnConstraint constraint : constraints) {
if (constraint instanceof ISqlJetColumnNotNull) {
notNull = true;
} else if (constraint instanceof ISqlJetColumnDefault) {
defaultValue = true;
} else {
throw new SqlJetException(
SqlJetErrorCode.MISUSE,
String.format("Invalid constraint: %s", constraint.toString()));
}
}
if (notNull && !defaultValue) {
throw new SqlJetException(
SqlJetErrorCode.MISUSE, "NOT NULL requires to have DEFAULT value");
}
}
columns = new ArrayList<ISqlJetColumnDef>(columns);
columns.add(newColumnDef);
}
final int page = tableDef.getPage();
final long rowId = tableDef.getRowId();
final SqlJetTableDef alterDef =
new SqlJetTableDef(
newTableName,
null,
tableDef.isTemporary(),
false,
columns,
tableDef.getConstraints(),
page,
rowId);
final ISqlJetBtreeSchemaTable schemaTable = openSchemaTable(true);
try {
schemaTable.lock();
try {
if (!schemaTable.goToRow(rowId)) {
throw new SqlJetException(SqlJetErrorCode.CORRUPT);
}
final String typeField = schemaTable.getTypeField();
final String nameField = schemaTable.getNameField();
final String tableField = schemaTable.getTableField();
final int pageField = schemaTable.getPageField();
if (null == typeField || !TABLE_TYPE.equals(typeField)) {
throw new SqlJetException(SqlJetErrorCode.CORRUPT);
}
if (null == nameField || !tableName.equals(nameField)) {
throw new SqlJetException(SqlJetErrorCode.CORRUPT);
}
if (null == tableField || !tableName.equals(tableField)) {
throw new SqlJetException(SqlJetErrorCode.CORRUPT);
}
if (0 == pageField || pageField != page) {
throw new SqlJetException(SqlJetErrorCode.CORRUPT);
}
final String alteredSql = getTableAlteredSql(schemaTable.getSqlField(), alterTableDef);
db.getOptions().changeSchemaVersion();
schemaTable.insertRecord(TABLE_TYPE, newTableName, newTableName, page, alteredSql);
if (renameTable && !tableName.equals(newTableName)) {
renameTablesIndices(
schemaTable, tableName, newTableName, getAlterTableName(alterTableDef));
}
tableDefs.remove(tableName);
tableDefs.put(newTableName, alterDef);
return alterDef;
} finally {
schemaTable.unlock();
}
} finally {
schemaTable.close();
}
}
