io.vertx.jdbcclient.impl.actions.JDBCTxOp Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2011-2014 The original author or authors
 * ------------------------------------------------------
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * and Apache License v2.0 which accompanies this distribution.
 *
 *     The Eclipse Public License is available at
 *     http://www.eclipse.org/legal/epl-v10.html
 *
 *     The Apache License v2.0 is available at
 *     http://www.opensource.org/licenses/apache2.0.php
 *
 * You may elect to redistribute this code under either of these licenses.
 */

package io.vertx.jdbcclient.impl.actions;

import io.vertx.jdbcclient.SqlOptions;
import io.vertx.sqlclient.internal.command.TxCommand;

import java.sql.Connection;
import java.sql.SQLException;

/**
 * @author Julien Viet
 */
public class JDBCTxOp extends AbstractJDBCAction {

  private final TxCommand op;

  public JDBCTxOp(JDBCStatementHelper helper, TxCommand op, SqlOptions options) {
    super(helper, options);
    this.op = op;
  }

  @Override
  public R execute(Connection conn) throws SQLException {
    if (op.kind == TxCommand.Kind.BEGIN) {
      conn.setAutoCommit(false);
    } else {
      try {
        if (op.kind == TxCommand.Kind.COMMIT) {
          conn.commit();
        } else {
          conn.rollback();
        }
      } finally {
        conn.setAutoCommit(false);
      }
    }
    return op.result;
  }
}