package org.hsqldb;
import org.hsqldb.error.Error;
import org.hsqldb.error.ErrorCode;
import org.hsqldb.lib.HsqlDeque;
import org.hsqldb.lib.LongDeque;
import org.hsqldb.persist.PersistentStore;
public class TransactionManagerMVCC extends TransactionManagerCommon
implements TransactionManager {
HsqlDeque committedTransactions = new HsqlDeque();
LongDeque committedTransactionTimestamps = new LongDeque();
boolean isLockedMode;
Session catalogWriteSession;
long lockTxTs;
long lockSessionId;
long unlockTxTs;
long unlockSessionId;
int redoCount = 0;
public TransactionManagerMVCC(Database db) {
super(db);
lobSession = database.sessionManager.getSysLobSession();
txModel = MVCC;
}
public long getGlobalChangeTimestamp() {
return globalChangeTimestamp.get();
}
public void setGlobalChangeTimestamp(long ts) {
globalChangeTimestamp.set(ts);
}
public boolean isMVRows() {
return true;
}
public boolean isMVCC() {
return true;
}
public boolean is2PL() {
return false;
}
public int getTransactionControl() {
return MVCC;
}
public void setTransactionControl(Session session, int mode) {
super.setTransactionControl(session, mode);
}
public void completeActions(Session session) {}
public boolean prepareCommitActions(Session session) {
if (session.abortTransaction) {
return false;
}
writeLock.lock();
try {
int limit = session.rowActionList.size();
for (int i = 0; i < limit; i++) {
RowAction action = (RowAction) session.rowActionList.get(i);
if (!action.canCommit(session)) {
return false;
}
}
session.actionTimestamp = getNextGlobalChangeTimestamp();
for (int i = 0; i < limit; i++) {
RowAction action = (RowAction) session.rowActionList.get(i);
action.prepareCommit(session);
}
for (int i = 0; i < session.actionSet.size(); i++) {
Session current =
((RowActionBase) session.actionSet.get(i)).session;
current.abortTransaction = true;
}
return true;
} finally {
writeLock.unlock();
session.actionSet.clear();
}
}
public boolean commitTransaction(Session session) {
if (session.abortTransaction) {
return false;
}
writeLock.lock();
try {
int limit = session.rowActionList.size();
for (int i = 0; i < limit; i++) {
RowAction action = (RowAction) session.rowActionList.get(i);
if (!action.canCommit(session)) {
return false;
}
}
session.actionTimestamp = getNextGlobalChangeTimestamp();
session.transactionEndTimestamp = session.actionTimestamp;
endTransaction(session);
for (int i = 0; i < limit; i++) {
RowAction action = (RowAction) session.rowActionList.get(i);
action.commit(session);
}
for (int i = 0; i < session.actionSet.size(); i++) {
Session current =
((RowActionBase) session.actionSet.get(i)).session;
current.abortTransaction = true;
}
adjustLobUsage(session);
persistCommit(session);
int newLimit = session.rowActionList.size();
if (newLimit > limit) {
Object[] list = session.rowActionList.getArray();
mergeTransaction(list, limit, newLimit,
session.actionTimestamp);
finaliseRows(session, list, limit, newLimit);
session.rowActionList.setSize(limit);
}
if (session == lobSession
|| getFirstLiveTransactionTimestamp()
> session.actionTimestamp) {
Object[] list = session.rowActionList.getArray();
mergeTransaction(list, 0, limit, session.actionTimestamp);
finaliseRows(session, list, 0, limit);
} else {
if (session.rowActionList.size() > 0) {
Object[] list = session.rowActionList.toArray();
addToCommittedQueue(session, list);
}
}
endTransactionTPL(session);
session.isTransaction = false;
countDownLatches(session);
} finally {
session.actionSet.clear();
writeLock.unlock();
}
return true;
}
public void rollback(Session session) {
writeLock.lock();
try {
session.abortTransaction = false;
session.actionTimestamp = getNextGlobalChangeTimestamp();
session.transactionEndTimestamp = session.actionTimestamp;
rollbackPartial(session, 0, session.transactionTimestamp);
endTransaction(session);
session.logSequences();
endTransactionTPL(session);
session.isTransaction = false;
countDownLatches(session);
} finally {
writeLock.unlock();
}
}
public void rollbackSavepoint(Session session, int index) {
long timestamp = session.sessionContext.savepointTimestamps.get(index);
Integer oi = (Integer) session.sessionContext.savepoints.get(index);
int start = oi.intValue();
while (session.sessionContext.savepoints.size() > index + 1) {
session.sessionContext.savepoints.remove(
session.sessionContext.savepoints.size() - 1);
session.sessionContext.savepointTimestamps.removeLast();
}
rollbackPartial(session, start, timestamp);
}
public void rollbackAction(Session session) {
rollbackPartial(session, session.actionIndex,
session.actionStartTimestamp);
}
public void rollbackPartial(Session session, int start, long timestamp) {
int limit = session.rowActionList.size();
if (start == limit) {
return;
}
for (int i = limit - 1; i >= start; i--) {
RowAction action = (RowAction) session.rowActionList.get(i);
if (action == null || action.type == RowActionBase.ACTION_NONE
|| action.type == RowActionBase.ACTION_DELETE_FINAL) {
continue;
}
Row row = action.memoryRow;
if (row == null) {
row = (Row) action.store.get(action.getPos(), false);
}
if (row == null) {
continue;
}
writeLock.lock();
try {
action.rollback(session, timestamp);
int type = action.mergeRollback(session, timestamp, row);
if (action.type == RowActionBase.ACTION_DELETE_FINAL) {
if (action.deleteComplete) {
continue;
}
action.deleteComplete = true;
}
action.store.rollbackRow(session, row, type, txModel);
} finally {
writeLock.unlock();
}
}
session.rowActionList.setSize(start);
}
public RowAction addDeleteAction(Session session, Table table,
PersistentStore store, Row row,
int[] colMap) {
RowAction action = addDeleteActionToRow(session, table, store, row,
colMap);
Session actionSession = null;
boolean redoAction = true;
if (action == null) {
writeLock.lock();
try {
rollbackAction(session);
if (session.isolationLevel == SessionInterface
.TX_REPEATABLE_READ || session
.isolationLevel == SessionInterface.TX_SERIALIZABLE) {
session.actionSet.clear();
session.redoAction = false;
session.abortTransaction = session.txConflictRollback;
throw Error.error(ErrorCode.X_40501);
}
if (row.rowAction != null && row.rowAction.isDeleted()) {
session.actionSet.clear();
session.redoAction = true;
redoCount++;
throw Error.error(ErrorCode.X_40501);
}
redoAction = !session.actionSet.isEmpty();
if (redoAction) {
actionSession =
((RowActionBase) session.actionSet.get(0)).session;
session.actionSet.clear();
if (actionSession != null) {
redoAction = checkDeadlock(session, actionSession);
}
}
if (redoAction) {
session.redoAction = true;
if (actionSession != null) {
actionSession.waitingSessions.add(session);
session.waitedSessions.add(actionSession);
session.latch.setCount(session.waitedSessions.size());
}
redoCount++;
} else {
session.redoAction = false;
session.abortTransaction = session.txConflictRollback;
}
throw Error.error(ErrorCode.X_40501);
} finally {
writeLock.unlock();
}
}
session.rowActionList.add(action);
return action;
}
public void addInsertAction(Session session, Table table,
PersistentStore store, Row row,
int[] changedColumns) {
RowAction action = row.rowAction;
Session actionSession = null;
boolean redoAction = false;
boolean redoWait = true;
HsqlException cause = null;
if (action == null) {
throw Error.runtimeError(ErrorCode.GENERAL_ERROR,
"TXManager - null insert action ");
}
try {
store.indexRow(session, row);
} catch (HsqlException e) {
if (session.actionSet.isEmpty()) {
throw e;
}
redoAction = true;
cause = e;
}
if (!redoAction) {
if (table.persistenceScope == Table.SCOPE_ROUTINE) {
row.rowAction = null;
return;
}
session.rowActionList.add(action);
return;
}
writeLock.lock();
try {
rollbackAction(session);
RowActionBase otherAction =
(RowActionBase) session.actionSet.get(0);
actionSession = otherAction.session;
session.actionSet.clear();
if (otherAction.commitTimestamp != 0) {
redoWait = false;
}
switch (session.isolationLevel) {
case SessionInterface.TX_REPEATABLE_READ :
case SessionInterface.TX_SERIALIZABLE :
redoAction = false;
break;
default :
redoAction = checkDeadlock(session, actionSession);
}
if (redoAction) {
session.redoAction = true;
if (redoWait) {
actionSession.waitingSessions.add(session);
session.waitedSessions.add(actionSession);
session.latch.setCount(session.waitedSessions.size());
}
redoCount++;
} else {
session.abortTransaction = session.txConflictRollback;
session.redoAction = false;
}
throw Error.error(cause, ErrorCode.X_40501, null);
} finally {
writeLock.unlock();
}
}
public void addInsertAction(Session session, PersistentStore store,
Row row) {
RowAction action = row.rowAction;
if (action == null) {
throw Error.runtimeError(ErrorCode.GENERAL_ERROR,
"null insert action ");
}
store.indexRow(session, row);
session.rowActionList.add(action);
}
public boolean canRead(Session session, PersistentStore store, Row row,
int mode, int[] colMap) {
RowAction action = row.rowAction;
if (action == null) {
return true;
} else if (action.table.tableType == TableBase.TEMP_TABLE) {
return true;
}
if (mode == TransactionManager.ACTION_READ) {
return action.canRead(session, TransactionManager.ACTION_READ);
}
if (mode == ACTION_REF) {
return action.canRead(session, TransactionManager.ACTION_READ);
}
return action.canRead(session, mode);
}
void addToCommittedQueue(Session session, Object[] list) {
synchronized (committedTransactionTimestamps) {
committedTransactions.addLast(list);
committedTransactionTimestamps.addLast(session.actionTimestamp);
}
}
void mergeExpiredTransactions(Session session) {
long timestamp = getFirstLiveTransactionTimestamp();
while (true) {
long commitTimestamp;
Object[] actions;
synchronized (committedTransactionTimestamps) {
if (committedTransactionTimestamps.isEmpty()) {
break;
}
commitTimestamp = committedTransactionTimestamps.getFirst();
if (commitTimestamp < timestamp) {
committedTransactionTimestamps.removeFirst();
actions = (Object[]) committedTransactions.removeFirst();
} else {
break;
}
}
mergeTransaction(actions, 0, actions.length, commitTimestamp);
finaliseRows(session, actions, 0, actions.length);
}
}
public void beginTransaction(Session session) {
writeLock.lock();
try {
if (!session.isTransaction) {
beginTransactionCommon(session);
liveTransactionTimestamps.addLast(
session.transactionTimestamp);
}
} finally {
writeLock.unlock();
}
}
public void beginAction(Session session, Statement cs) {
if (session.isTransaction) {
return;
}
if (cs == null) {
return;
}
writeLock.lock();
try {
if (hasExpired) {
session.redoAction = true;
return;
}
cs = updateCurrentStatement(session, cs);
if (cs == null) {
return;
}
if (session.abortTransaction) {
return;
}
session.isPreTransaction = true;
if (!isLockedMode && !cs.isCatalogLock(txModel)) {
return;
}
beginActionTPL(session, cs);
} finally {
writeLock.unlock();
}
}
public void beginActionResume(Session session) {
writeLock.lock();
try {
if (session.isTransaction) {
session.actionTimestamp = getNextGlobalChangeTimestamp();
session.actionStartTimestamp = session.actionTimestamp;
} else {
beginTransactionCommon(session);
liveTransactionTimestamps.addLast(
session.transactionTimestamp);
}
} finally {
writeLock.unlock();
}
}
RowAction addDeleteActionToRow(Session session, Table table,
PersistentStore store, Row row,
int[] colMap) {
RowAction action = store.addDeleteActionToRow(session, row, colMap,
true);
if (table.tableType == TableBase.TEMP_TABLE) {
store.delete(session, row);
row.rowAction = null;
}
return action;
}
void endTransaction(Session session) {
long timestamp = session.transactionTimestamp;
int index = liveTransactionTimestamps.indexOf(timestamp);
if (index >= 0) {
transactionCount.decrementAndGet();
liveTransactionTimestamps.remove(index);
mergeExpiredTransactions(session);
}
}
private void countDownLatches(Session session) {
for (int i = 0; i < session.waitingSessions.size(); i++) {
Session current = (Session) session.waitingSessions.get(i);
current.waitedSessions.remove(session);
current.latch.setCount(current.waitedSessions.size());
}
session.waitedSessions.clear();
session.waitingSessions.clear();
}
void endTransactionTPL(Session session) {
if (catalogWriteSession != session) {
return;
}
Session nextSession = null;
for (int i = 0; i < session.waitingSessions.size(); i++) {
Session current = (Session) session.waitingSessions.get(i);
Statement st = current.sessionContext.currentStatement;
if (st != null && st.isCatalogLock(txModel)) {
nextSession = current;
break;
}
}
if (nextSession == null) {
catalogWriteSession = null;
isLockedMode = false;
} else {
for (int i = 0; i < session.waitingSessions.size(); i++) {
Session current = (Session) session.waitingSessions.get(i);
if (current != nextSession) {
current.waitedSessions.add(nextSession);
nextSession.waitingSessions.add(current);
current.latch.setCount(current.waitedSessions.size());
}
}
catalogWriteSession = nextSession;
}
unlockTxTs = session.actionTimestamp;
unlockSessionId = session.getId();
}
boolean beginActionTPL(Session session, Statement cs) {
if (session == catalogWriteSession) {
return true;
}
session.tempSet.clear();
if (cs.isCatalogLock(txModel)) {
if (catalogWriteSession == null) {
catalogWriteSession = session;
isLockedMode = true;
lockTxTs = session.actionTimestamp;
lockSessionId = session.getId();
getTransactionAndPreSessions(session);
if (!session.tempSet.isEmpty()) {
session.waitedSessions.addAll(session.tempSet);
setWaitingSessionTPL(session);
}
return true;
}
}
if (!isLockedMode) {
return true;
}
if (cs.getTableNamesForWrite().length > 0) {
if (cs.getTableNamesForWrite()[0].schema
== SqlInvariants.LOBS_SCHEMA_HSQLNAME) {
return true;
}
} else if (cs.getTableNamesForRead().length > 0) {
if (cs.getTableNamesForRead()[0].schema
== SqlInvariants.LOBS_SCHEMA_HSQLNAME) {
return true;
}
}
if (session.waitingSessions.contains(catalogWriteSession)) {
return true;
}
if (catalogWriteSession.waitingSessions.add(session)) {
session.waitedSessions.add(catalogWriteSession);
session.latch.setCount(session.waitedSessions.size());
}
return true;
}
public void resetSession(Session session, Session targetSession,
int mode) {
super.resetSession(session, targetSession, mode);
}
}