@Override
public void add(Session session, Row row) {
StatementBuilder sql = new StatementBuilder("insert into ");
sql.append(getName()).append("(_gui_row_id_");
for (Column c : getColumns()) {
sql.append(",");
sql.append(c.getName());
}
sql.append(") values (");
sql.append(row.getKey());
for (Column c : getColumns()) {
sql.append(",");
Value v = row.getValue(c.getColumnId());
if (v == null) {
sql.append("DEFAULT");
} else {
sql.append(v.getSQL());
}
}
sql.append(")");
Prepared prepared = session.prepare(sql.toString(), true);
prepared.setLocal(false);
prepared.update();
}
public GlobalUniqueIndex(
Session session,
MVTable table,
int id,
String indexName,
IndexColumn[] columns,
IndexType indexType) {
initIndexBase(table, id, indexName, columns, indexType);
if (!database.isStarting()) {
checkIndexColumnTypes(columns);
}
StatementBuilder sql = new StatementBuilder("create ");
if (table.isGlobalTemporary()) sql.append("global temporary ");
else if (table.isTemporary()) sql.append("local temporary ");
sql.append("table if not exists ").append(getName()).append("(_gui_row_id_ long,");
for (Column c : getColumns()) {
sql.append(c.getCreateSQL()).append(",");
}
sql.append("primary key(");
for (Column c : getColumns()) {
sql.appendExceptFirst(",");
sql.append(c.getName());
}
sql.append("))");
Prepared prepared = session.prepare(sql.toString(), true);
prepared.setLocal(true);
prepared.update();
}
@Override
public void rename(String newName) {
StatementBuilder sql = new StatementBuilder("alter table ");
sql.append(getName()).append(" rename to").append(newName);
Prepared prepared = getDatabase().getSystemSession().prepare(sql.toString(), true);
prepared.setLocal(true);
prepared.update();
}
@Override
public String getSQL(boolean isDistributed) {
StatementBuilder buff = new StatementBuilder(getName());
buff.append('(');
int i = 0;
for (Expression e : args) {
buff.appendExceptFirst(", ");
buff.append(columnList[i++].getCreateSQL()).append('=').append(e.getSQL(isDistributed));
}
return buff.append(')').toString();
}
@Override
public long getRowCount(Session session) {
StatementBuilder sql = new StatementBuilder("select count(*) from ");
sql.append(getName());
if (session == null) session = getDatabase().getSystemSession();
Prepared prepared = session.prepare(sql.toString(), true);
prepared.setLocal(false);
ResultInterface result = prepared.query(0);
return result.getRowCount();
}
public void prepare() {
if (columns == null) {
if (list.size() > 0 && list.get(0).length == 0) {
// special case where table is used as a sequence
columns = new Column[0];
} else {
columns = table.getColumns();
}
}
if (list.size() > 0) {
for (Expression[] expr : list) {
if (expr.length != columns.length) {
throw DbException.get(ErrorCode.COLUMN_COUNT_DOES_NOT_MATCH);
}
for (int i = 0; i < expr.length; i++) {
Expression e = expr[i];
if (e != null) {
expr[i] = e.optimize(session);
}
}
}
} else {
query.prepare();
if (query.getColumnCount() != columns.length) {
throw DbException.get(ErrorCode.COLUMN_COUNT_DOES_NOT_MATCH);
}
}
if (keys == null) {
Index idx = table.getPrimaryKey();
if (idx == null) {
throw DbException.get(ErrorCode.CONSTRAINT_NOT_FOUND_1, "PRIMARY KEY");
}
keys = idx.getColumns();
}
StatementBuilder buff = new StatementBuilder("UPDATE ");
buff.append(table.getSQL()).append(" SET ");
for (Column c : columns) {
buff.appendExceptFirst(", ");
buff.append(c.getSQL()).append("=?");
}
buff.append(" WHERE ");
buff.resetCount();
for (Column c : keys) {
buff.appendExceptFirst(" AND ");
buff.append(c.getSQL()).append("=?");
}
String sql = buff.toString();
update = session.prepare(sql);
}
@Override
public void remove(Session session, Row row) {
StatementBuilder sql = new StatementBuilder("delete from ");
sql.append(getName());
if (row != null) {
sql.append(" where ");
for (Column c : getColumns()) {
sql.appendExceptFirst(" and ");
sql.append(c.getName()).append("=");
Value v = row.getValue(c.getColumnId());
if (v != null) {
sql.append(v.getSQL());
}
}
}
Prepared prepared = session.prepare(sql.toString(), true);
prepared.setLocal(false);
prepared.update();
}
private int generateInsertValues(int count, Table table) throws IOException {
PlanItem plan = table.getBestPlanItem(session, null, null);
Index index = plan.getIndex();
Cursor cursor = index.find(session, null, null);
Column[] columns = table.getColumns();
StatementBuilder buff = new StatementBuilder("INSERT INTO ");
buff.append(table.getSQL()).append('(');
for (Column col : columns) {
buff.appendExceptFirst(", ");
buff.append(Parser.quoteIdentifier(col.getName()));
}
buff.append(") VALUES");
if (!simple) {
buff.append('\n');
}
buff.append('(');
String ins = buff.toString();
buff = null;
while (cursor.next()) {
Row row = cursor.get();
if (buff == null) {
buff = new StatementBuilder(ins);
} else {
buff.append(",\n(");
}
for (int j = 0; j < row.getColumnCount(); j++) {
if (j > 0) {
buff.append(", ");
}
Value v = row.getValue(j);
if (v.getPrecision() > lobBlockSize) {
int id;
if (v.getType() == Value.CLOB) {
id = writeLobStream(v);
buff.append("SYSTEM_COMBINE_CLOB(" + id + ")");
} else if (v.getType() == Value.BLOB) {
id = writeLobStream(v);
buff.append("SYSTEM_COMBINE_BLOB(" + id + ")");
} else {
buff.append(v.getSQL());
}
} else {
buff.append(v.getSQL());
}
}
buff.append(')');
count++;
if ((count & 127) == 0) {
checkCanceled();
}
if (simple || buff.length() > Constants.IO_BUFFER_SIZE) {
add(buff.toString(), true);
buff = null;
}
}
if (buff != null) {
add(buff.toString(), true);
}
return count;
}
public String getPlanSQL() {
StatementBuilder buff = new StatementBuilder("MERGE INTO ");
buff.append(table.getSQL()).append('(');
for (Column c : columns) {
buff.appendExceptFirst(", ");
buff.append(c.getSQL());
}
buff.append(')');
if (keys != null) {
buff.append(" KEY(");
buff.resetCount();
for (Column c : keys) {
buff.appendExceptFirst(", ");
buff.append(c.getSQL());
}
buff.append(')');
}
buff.append('\n');
if (list.size() > 0) {
buff.append("VALUES ");
int row = 0;
for (Expression[] expr : list) {
if (row++ > 0) {
buff.append(", ");
}
buff.append('(');
buff.resetCount();
for (Expression e : expr) {
buff.appendExceptFirst(", ");
if (e == null) {
buff.append("DEFAULT");
} else {
buff.append(e.getSQL());
}
}
buff.append(')');
}
} else {
buff.append(query.getPlanSQL());
}
return buff.toString();
}
private Cursor find(Session session, SearchRow first, boolean bigger, SearchRow last) {
StatementBuilder sql = new StatementBuilder("select _gui_row_id_");
for (Column c : getColumns()) {
sql.append(",");
sql.append(c.getName());
}
sql.append(" from ").append(getName());
if (first != null || last != null) {
sql.append(" where ");
for (Column c : getColumns()) {
sql.appendExceptFirst(" and ");
if (first != null) {
sql.append(c.getName()).append(">=");
Value v = first.getValue(c.getColumnId());
if (v != null) {
sql.append(v.getSQL());
}
}
if (last != null) {
sql.append(c.getName()).append("<=");
Value v = last.getValue(c.getColumnId());
if (v != null) {
sql.append(v.getSQL());
}
}
}
}
Prepared prepared = session.prepare(sql.toString(), true);
prepared.setLocal(false);
ResultInterface result = prepared.query(0);
if (bigger) result.next();
return new GlobalUniqueIndexTableCursor(result);
}
