ee.jakarta.tck.concurrent.spec.ManagedExecutorService.tx.CancelledTransactedTask 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.ManagedExecutorService.tx;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.time.Duration;

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

public class CancelledTransactedTask implements Runnable {
	private volatile boolean runFlag;

	private volatile boolean transactionBegan;

	private volatile boolean cancelled;

	private final String username, password, sqlTemplate;

	public CancelledTransactedTask(String username, String password, String sqlTemplate) {
		this.username = username;
		this.password = password;
		this.sqlTemplate = sqlTemplate;
	}

	public void resume() {
		runFlag = true;
	}

	public void cancelTask() {
		cancelled = true;
	}

	public boolean transactionBegin() {
		return transactionBegan;
	}

	@Override
	public void run() {
		Connection conn = Util.getConnection(false, username, password);
		UserTransaction ut = TestUtil.lookup(TestConstants.UserTransaction);
		if (ut == null) {
			// error if no transaction can be obtained in task.
			throw new RuntimeException("didn't get user transaction inside the submitted task.");
		} else {
			PreparedStatement pStmt = null;
			try {
				ut.begin();
				transactionBegan = true;
				while (!runFlag) {
					TestUtil.sleep(Duration.ofMillis(500));
				}
				pStmt = conn.prepareStatement(sqlTemplate);
				String sTypeDesc = "Type-Cancelled-99";
				int newType = 991;
				pStmt.setInt(1, newType);
				pStmt.setString(2, sTypeDesc);
				pStmt.executeUpdate();
				// check if it is cancelled here
				if (cancelled) {
					ut.rollback();
					return;
				}
				ut.commit();
			} catch (Exception e) {
				try {
					ut.rollback();
				} catch (Exception e1) {
					e1.printStackTrace();
				}
				e.printStackTrace();
			} finally {
				try {
					pStmt.close();
					conn.close();
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
	}

}