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

import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.fail;

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);
		
		assertNotNull(ut, "didn't get user transaction inside the submitted task.");
		
		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) {
				//ignore
			}
			fail(e.getMessage());
		} finally {
			try {
				pStmt.close();
				conn.close();
			} catch (Exception e) {
				//ignore
			}
		}
	}
}