org.h2.command.dml.TransactionCommand Maven / Gradle / Ivy
/*
* Copyright 2004-2019 H2 Group. Multiple-Licensed under the MPL 2.0,
* and the EPL 1.0 (https://h2database.com/html/license.html).
* Initial Developer: H2 Group
*/
package org.h2.command.dml;
import org.h2.command.CommandInterface;
import org.h2.command.Prepared;
import org.h2.engine.Database;
import org.h2.engine.Session;
import org.h2.message.DbException;
import org.h2.result.ResultInterface;
/**
* Represents a transactional statement.
*/
public class TransactionCommand extends Prepared {
private final int type;
private String savepointName;
private String transactionName;
public TransactionCommand(Session session, int type) {
super(session);
this.type = type;
}
public void setSavepointName(String name) {
this.savepointName = name;
}
@Override
public int update() {
switch (type) {
case CommandInterface.SET_AUTOCOMMIT_TRUE:
session.setAutoCommit(true);
break;
case CommandInterface.SET_AUTOCOMMIT_FALSE:
session.setAutoCommit(false);
break;
case CommandInterface.BEGIN:
session.begin();
break;
case CommandInterface.COMMIT:
session.commit(false);
break;
case CommandInterface.ROLLBACK:
session.rollback();
break;
case CommandInterface.CHECKPOINT:
session.getUser().checkAdmin();
session.getDatabase().checkpoint();
break;
case CommandInterface.SAVEPOINT:
session.addSavepoint(savepointName);
break;
case CommandInterface.ROLLBACK_TO_SAVEPOINT:
session.rollbackToSavepoint(savepointName);
break;
case CommandInterface.CHECKPOINT_SYNC:
session.getUser().checkAdmin();
session.getDatabase().sync();
break;
case CommandInterface.PREPARE_COMMIT:
session.prepareCommit(transactionName);
break;
case CommandInterface.COMMIT_TRANSACTION:
session.getUser().checkAdmin();
session.setPreparedTransaction(transactionName, true);
break;
case CommandInterface.ROLLBACK_TRANSACTION:
session.getUser().checkAdmin();
session.setPreparedTransaction(transactionName, false);
break;
case CommandInterface.SHUTDOWN_IMMEDIATELY:
session.getUser().checkAdmin();
session.getDatabase().shutdownImmediately();
break;
case CommandInterface.SHUTDOWN:
case CommandInterface.SHUTDOWN_COMPACT:
case CommandInterface.SHUTDOWN_DEFRAG: {
session.getUser().checkAdmin();
session.commit(false);
// throttle, to allow testing concurrent
// execution of shutdown and query
session.throttle();
Database db = session.getDatabase();
if (db.setExclusiveSession(session, true)) {
if (type == CommandInterface.SHUTDOWN_COMPACT ||
type == CommandInterface.SHUTDOWN_DEFRAG) {
db.setCompactMode(type);
}
// close the database, but don't update the persistent setting
db.setCloseDelay(0);
session.close();
}
break;
}
default:
DbException.throwInternalError("type=" + type);
}
return 0;
}
@Override
public boolean isTransactional() {
return true;
}
@Override
public boolean needRecompile() {
return false;
}
public void setTransactionName(String string) {
this.transactionName = string;
}
@Override
public ResultInterface queryMeta() {
return null;
}
@Override
public int getType() {
return type;
}
@Override
public boolean isCacheable() {
return true;
}
}