ee.jakarta.tck.concurrent.spec.ContextService.tx.TestTransactionWork Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2013, 2022 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package ee.jakarta.tck.concurrent.spec.ContextService.tx;

import java.io.Serializable;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

import ee.jakarta.tck.concurrent.framework.TestUtil;
import jakarta.transaction.UserTransaction;

@SuppressWarnings("serial")
public class TestTransactionWork implements TestWorkInterface, Serializable {

	protected String result;

	protected Connection con;

	protected String sqlTemplate;

	protected boolean beginTran = false;

	protected boolean commit = false;

	protected boolean rollback = false;

	protected String userName;

	protected String password;

	@Override
	public void doSomeWork() {
		UserTransaction ut = null;

		if (beginTran) {
			try {
				ut = TestUtil.lookup(Constants.UT_JNDI_NAME);
				ut.begin();
			} catch (Exception e) {
				throw new RuntimeException(e);
			}
		}
		PreparedStatement pStmt = null;
		try {
			con = Util.getConnection(false, userName, password);
			pStmt = con.prepareStatement(sqlTemplate);
			String sTypeDesc = "Type-98";
			int newType = 98;
			pStmt.setInt(1, newType);
			pStmt.setString(2, sTypeDesc);
			pStmt.executeUpdate();

			if (commit) {
				ut.commit();
			}

			if (rollback) {
				ut.rollback();
			}
		} catch (Exception e) {
			throw new RuntimeException(e);
		} finally {
			try {
				if (pStmt != null)
					pStmt.close();
				if (con != null)
					con.close();
			} catch (SQLException se) {
			}
		}
	}

	@Override
	public String getResult() {
		return result;
	}

	@Override
	public void setConnection(Connection con) {
		this.con = con;
	}

	@Override
	public void setSQLTemplate(String sqlTemplate) {
		this.sqlTemplate = sqlTemplate;
	}

	@Override
	public void needBeginTx(boolean beginTx) {
		this.beginTran = beginTx;
	}

	@Override
	public void needCommit(boolean commit) {
		this.commit = commit;
	}

	@Override
	public void needRollback(boolean rollback) {
		this.rollback = rollback;
	}

	@Override
	public void run() {
		doSomeWork();
	}

	@Override
	public void setUserName(String name) {
		this.userName = name;
	}

	@Override
	public void setPassword(String pass) {
		this.password = pass;
	}
}