appserver/tests/appserv-tests/devtests/jdbc/jdbc41-devtests/basic/ejb/SimpleSessionBean.java - glassfish - Git at Google

 /*
  * Copyright (c) 2017, 2018 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 com.sun.s1asdev.jdbc41.ejb;

 import javax.ejb.*;
 import javax.naming.*;
 import javax.sql.*;
 import java.rmi.*;
 import java.util.*;
 import java.sql.*;
 import javax.transaction.UserTransaction;
 import java.util.concurrent.*;
 import java.lang.reflect.Method;

 public class SimpleSessionBean implements SessionBean
 {

     private SessionContext ctxt_;
     private InitialContext ic_;
     private DataSource ds;
     PrintTask task1 = new PrintTask("thread1");
     PrintTask task2 = new PrintTask("thread2");
     ExecutorService threadExecutor = Executors.newCachedThreadPool(new ThreadFactory() {
 	public Thread newThread(Runnable r) {
 	    Thread th = new Thread(r);
 	    th.setDaemon(true);
 	    return th;
 	}
     });

     class PrintTask implements Runnable {
         private String threadName;

         public PrintTask(String name) {
             threadName = name;
         }

         public void run() {
             System.out.println("Thread : " + threadName + " executed");
         }
     }

     public void setSessionContext(SessionContext context) {
         ctxt_ = context;
 	try {
 	    ic_ = new InitialContext();
 	    ds = (DataSource) ic_.lookup("java:comp/env/DataSource");
 	} catch( Exception ne ) {
 	    ne.printStackTrace();
 	}
     }

     public void ejbCreate() throws CreateException {
     }

     /**
      * Testing connection object
      */
     public boolean test1() throws Exception {
 	Connection con = null;
 	Connection[] conns = new Connection[32];
 	Statement stmt = null;
 	boolean passed = true;
 	try {
 	    con = ds.getConnection("scott", "tiger");
 	    stmt = con.createStatement();

 	    //Try getting schema from con object
 	    System.out.println(">>> con.getSchema()=" + con.getSchema());

 	    //try getting network timeout from con object
 	    try {
 	        System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
 	    } catch(Exception ex) {
 		passed = true;
 		//Expected as this is not implemented in derby 10.8 jdbc41 driver
 	    }

 	    //Invoke abort on con object
 	    threadExecutor.submit(task1);
 	    threadExecutor.submit(task2);
 	    con.abort(threadExecutor);
 	    System.out.println("Marked connection for aborting...");

 	    //Try to use the aborted connection to create a statement. should fail
 	    try {
 	        con.createStatement();
 	    } catch(SQLException ex) {
 		passed = true;
 		System.out.println("Creating statement from con after tx abort : Failed : Expected");
 	    }
 	} catch(Exception ex) {
 	    passed = false;
 	} finally {
 	    if ( stmt != null ) {
 	        try { stmt.close(); } catch( Exception e1) {}
 	    }
 	    if ( con != null ) {
 	        try { con.close(); } catch( Exception e1) {}
 	    }
 	    for(int i=0; i<32; i++) {
 		if(conns[i] != null) {
 		    try { conns[i].close(); } catch(Exception ex) {}
 		}
 	    }
 	}
 	return passed;
     }

     /**
      * Testing connection object in a User transaction
      */
     public boolean test2() throws Exception {
 	Connection con = null;
 	Connection[] conns = new Connection[32];
 	Connection[] conns1 = new Connection[32];
 	Statement[] stmts1 = new Statement[32];
 	Statement stmt = null;
 	boolean passed = true;
 	try {
 	    UserTransaction usertx = (UserTransaction) ctxt_.getUserTransaction();
             usertx.begin();
 	    con = ds.getConnection("scott", "tiger");
 	    //System.out.println("!!!! TEST2 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(con)).hashCode());
 	    stmt = con.createStatement();

 	    //Invoke abort on con object
 	    threadExecutor.submit(task1);
 	    threadExecutor.submit(task2);
 	    con.abort(threadExecutor);
 	    System.out.println("Marked connection for aborting...");

 	    //Operations on con object after marking for abort in a tx
 	    //before tx commit, all operations on con pass
             //Try getting schema from con object
 	    System.out.println(">>> con.getSchema()=" + con.getSchema());

 	    //try getting network timeout from con object
 	    try {
 	        System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
 	    } catch(Exception ex) {
 		passed = true;
 		//Expected as this is not implemented in derby 10.8 jdbc41 driver
 	    }

 	    usertx.commit();

 	    //Try to use the aborted connection to create a statement. should fail
 	    try {
 	        con.createStatement();
 	    } catch(SQLException ex) {
 		passed = true;
 		System.out.println("Creating statement from con after tx abort : Failed : Expected");
 	    }
 	} catch(Exception ex) {
 	    passed = false;
 	} finally {
 	    if ( stmt != null ) {
 	        try { stmt.close(); } catch( Exception e1) {}
 	    }
 	    if ( con != null ) {
 	        try { con.close(); } catch( Exception e1) {}
 	    }
 	    //Test to get (max pool size-1) no. of connections after a previous abort.
 	    //should pass as the aborted connection is destroyed
 	    for(int i=0; i<32; i++) {
 		if(i == 31) {
 		    //Last connection should not be got
 		    try {
 			conns1[i] = ds.getConnection("scott", "tiger");
 		    } catch(Exception ex) {
 			passed = true;
 		    }
 		} else {
                     conns1[i] = ds.getConnection("scott", "tiger");
 		    stmts1[i] = conns1[i].createStatement();
 		}
 	    }
 	    for(int i=0; i<32; i++) {
 		if(stmts1[i] != null) {
 		    try { stmts1[i].close(); } catch(Exception ex) {}
 		}
 	    }
 	    for(int i=0; i<32; i++) {
 		if(conns1[i] != null) {
 		    try { conns1[i].close(); } catch(Exception ex) {}
 		}
 	    }
 	}
 	return passed;
     }

     //Tests DatabaseMetaData
     public boolean test3() throws Exception {
 	Connection con = null;
 	DatabaseMetaData dmd = null;
 	boolean passed = true;
 	try {
 	    con = ds.getConnection("scott", "tiger");
 	    dmd = con.getMetaData();
 	    //Testing DatabaseMetaData
 	    System.out.println(">>> dmd.generatedKeyAlwaysReturned()=" + dmd.generatedKeyAlwaysReturned());
 	} catch(Exception ex) {
 	    passed = false;
 	} finally {
 	    if(con!= null) {
 		try { con.close(); } catch(Exception ex) {}
 	    }
 	}
 	return passed;
     }

     /**
      * Testing Statement object within a transaction
      */
     public boolean test4() throws Exception {
 	Connection conn1 = null;
 	Statement stmt1 = null;
 	ResultSet rs1 = null;
 	boolean passed = true;

 	//Now try getting a connection again, but within a transaction
 	int size = 0;
 	try {
 	    createTable();
 	    UserTransaction tx = (UserTransaction) ctxt_.getUserTransaction();
 	    tx.begin();
 	    conn1 = ds.getConnection("scott", "tiger");
 	    System.out.println("!!!! TEST4 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(conn1)).hashCode());
 	    stmt1 = conn1.createStatement();
 	    insertRow(stmt1);

 	    threadExecutor.submit(task1);
             threadExecutor.submit(task2);

 	    //Testing Statement object
 	    stmt1.closeOnCompletion();
 	    System.out.println(">>> stmt1.isCloseOnCompletion()=" + stmt1.isCloseOnCompletion());

             conn1.abort(threadExecutor);

 	    System.out.println("Marked connection for aborting...");
 	    insertRow(stmt1);

 	    //After con abort (within the tx)
 	    size = getCount(stmt1);

 	    if(size != 2) {
 		passed = false;
 		System.out.println("FAILED as size!= 2");
 	    }

 	    tx.commit();

 	    try {
 	        //After tx commit, insert data. Should fail because of no current connection
 		insertRow(stmt1);
 	    } catch(SQLException ex) {
 		passed = true;
 		System.out.println("Trying to insert data after a tx commit (prev. conn abort): Failed with : " + ex + " Expected.");
 	    }
 	} catch (Exception e) {
 	   passed = false;
 	} finally {
 	    if (rs1 != null ) {
 	        try { rs1.close(); } catch( Exception e1 ) {}
 	    }
 	    if ( stmt1 != null ) {
 	        try { stmt1.close(); } catch( Exception e1) {}
 	    }
 	    if ( conn1 != null ) {
 	        try { conn1.close(); } catch( Exception e1) {}
 	    }
 	}
 	return passed;
     }

     private void createTable() throws SQLException {
         String tableName = "JDBC41";

         Connection con = null;
         Statement stmt = null;
         con = ds.getConnection("scott", "tiger");
         stmt = con.createStatement();

         try{
             stmt.executeUpdate("drop table " + tableName);
         }catch(Exception ex){
         }
         stmt.executeUpdate("create table " + tableName + " (qty integer, name varchar(32))");
         stmt.close();
         con.close();
     }

     private void insertRow(Statement stmt) throws SQLException {
         String tableName = "JDBC41";
         stmt.executeUpdate("INSERT INTO " + tableName + " values (100, 'COFFEE')");
     }

     private int getCount(Statement stmt) throws SQLException {
         String tableName = "JDBC41";
         ResultSet rs = null;
         rs = stmt.executeQuery( "SELECT count(*) FROM " + tableName);
         rs.next();
         int count = rs.getInt(1);
         rs.close();
         return count;
     }

     public void ejbLoad() {}
     public void ejbStore() {}
     public void ejbRemove() {}
     public void ejbActivate() {}
     public void ejbPassivate() {}
     public void unsetEntityContext() {}
     public void ejbPostCreate() {}
 }
	/*
	* Copyright (c) 2017, 2018 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 com.sun.s1asdev.jdbc41.ejb;

	import javax.ejb.*;
	import javax.naming.*;
	import javax.sql.*;
	import java.rmi.*;
	import java.util.*;
	import java.sql.*;
	import javax.transaction.UserTransaction;
	import java.util.concurrent.*;
	import java.lang.reflect.Method;

	public class SimpleSessionBean implements SessionBean
	{

	private SessionContext ctxt_;
	private InitialContext ic_;
	private DataSource ds;
	PrintTask task1 = new PrintTask("thread1");
	PrintTask task2 = new PrintTask("thread2");
	ExecutorService threadExecutor = Executors.newCachedThreadPool(new ThreadFactory() {
	public Thread newThread(Runnable r) {
	Thread th = new Thread(r);
	th.setDaemon(true);
	return th;
	}
	});

	class PrintTask implements Runnable {
	private String threadName;

	public PrintTask(String name) {
	threadName = name;
	}

	public void run() {
	System.out.println("Thread : " + threadName + " executed");
	}
	}

	public void setSessionContext(SessionContext context) {
	ctxt_ = context;
	try {
	ic_ = new InitialContext();
	ds = (DataSource) ic_.lookup("java:comp/env/DataSource");
	} catch( Exception ne ) {
	ne.printStackTrace();
	}
	}

	public void ejbCreate() throws CreateException {
	}

	/**
	* Testing connection object
	*/
	public boolean test1() throws Exception {
	Connection con = null;
	Connection[] conns = new Connection[32];
	Statement stmt = null;
	boolean passed = true;
	try {
	con = ds.getConnection("scott", "tiger");
	stmt = con.createStatement();

	//Try getting schema from con object
	System.out.println(">>> con.getSchema()=" + con.getSchema());

	//try getting network timeout from con object
	try {
	System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
	} catch(Exception ex) {
	passed = true;
	//Expected as this is not implemented in derby 10.8 jdbc41 driver
	}

	//Invoke abort on con object
	threadExecutor.submit(task1);
	threadExecutor.submit(task2);
	con.abort(threadExecutor);
	System.out.println("Marked connection for aborting...");

	//Try to use the aborted connection to create a statement. should fail
	try {
	con.createStatement();
	} catch(SQLException ex) {
	passed = true;
	System.out.println("Creating statement from con after tx abort : Failed : Expected");
	}
	} catch(Exception ex) {
	passed = false;
	} finally {
	if ( stmt != null ) {
	try { stmt.close(); } catch( Exception e1) {}
	}
	if ( con != null ) {
	try { con.close(); } catch( Exception e1) {}
	}
	for(int i=0; i<32; i++) {
	if(conns[i] != null) {
	try { conns[i].close(); } catch(Exception ex) {}
	}
	}
	}
	return passed;
	}

	/**
	* Testing connection object in a User transaction
	*/
	public boolean test2() throws Exception {
	Connection con = null;
	Connection[] conns = new Connection[32];
	Connection[] conns1 = new Connection[32];
	Statement[] stmts1 = new Statement[32];
	Statement stmt = null;
	boolean passed = true;
	try {
	UserTransaction usertx = (UserTransaction) ctxt_.getUserTransaction();
	usertx.begin();
	con = ds.getConnection("scott", "tiger");
	//System.out.println("!!!! TEST2 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(con)).hashCode());
	stmt = con.createStatement();

	//Invoke abort on con object
	threadExecutor.submit(task1);
	threadExecutor.submit(task2);
	con.abort(threadExecutor);
	System.out.println("Marked connection for aborting...");

	//Operations on con object after marking for abort in a tx
	//before tx commit, all operations on con pass
	//Try getting schema from con object
	System.out.println(">>> con.getSchema()=" + con.getSchema());

	//try getting network timeout from con object
	try {
	System.out.println(">>> con.getNetworkTimeout=" + con.getNetworkTimeout());
	} catch(Exception ex) {
	passed = true;
	//Expected as this is not implemented in derby 10.8 jdbc41 driver
	}

	usertx.commit();

	//Try to use the aborted connection to create a statement. should fail
	try {
	con.createStatement();
	} catch(SQLException ex) {
	passed = true;
	System.out.println("Creating statement from con after tx abort : Failed : Expected");
	}
	} catch(Exception ex) {
	passed = false;
	} finally {
	if ( stmt != null ) {
	try { stmt.close(); } catch( Exception e1) {}
	}
	if ( con != null ) {
	try { con.close(); } catch( Exception e1) {}
	}
	//Test to get (max pool size-1) no. of connections after a previous abort.
	//should pass as the aborted connection is destroyed
	for(int i=0; i<32; i++) {
	if(i == 31) {
	//Last connection should not be got
	try {
	conns1[i] = ds.getConnection("scott", "tiger");
	} catch(Exception ex) {
	passed = true;
	}
	} else {
	conns1[i] = ds.getConnection("scott", "tiger");
	stmts1[i] = conns1[i].createStatement();
	}
	}
	for(int i=0; i<32; i++) {
	if(stmts1[i] != null) {
	try { stmts1[i].close(); } catch(Exception ex) {}
	}
	}
	for(int i=0; i<32; i++) {
	if(conns1[i] != null) {
	try { conns1[i].close(); } catch(Exception ex) {}
	}
	}
	}
	return passed;
	}

	//Tests DatabaseMetaData
	public boolean test3() throws Exception {
	Connection con = null;
	DatabaseMetaData dmd = null;
	boolean passed = true;
	try {
	con = ds.getConnection("scott", "tiger");
	dmd = con.getMetaData();
	//Testing DatabaseMetaData
	System.out.println(">>> dmd.generatedKeyAlwaysReturned()=" + dmd.generatedKeyAlwaysReturned());
	} catch(Exception ex) {
	passed = false;
	} finally {
	if(con!= null) {
	try { con.close(); } catch(Exception ex) {}
	}
	}
	return passed;
	}

	/**
	* Testing Statement object within a transaction
	*/
	public boolean test4() throws Exception {
	Connection conn1 = null;
	Statement stmt1 = null;
	ResultSet rs1 = null;
	boolean passed = true;

	//Now try getting a connection again, but within a transaction
	int size = 0;
	try {
	createTable();
	UserTransaction tx = (UserTransaction) ctxt_.getUserTransaction();
	tx.begin();
	conn1 = ds.getConnection("scott", "tiger");
	System.out.println("!!!! TEST4 ConHashcode=" + (((com.sun.appserv.jdbc.DataSource)ds).getConnection(conn1)).hashCode());
	stmt1 = conn1.createStatement();
	insertRow(stmt1);

	threadExecutor.submit(task1);
	threadExecutor.submit(task2);

	//Testing Statement object
	stmt1.closeOnCompletion();
	System.out.println(">>> stmt1.isCloseOnCompletion()=" + stmt1.isCloseOnCompletion());

	conn1.abort(threadExecutor);

	System.out.println("Marked connection for aborting...");
	insertRow(stmt1);

	//After con abort (within the tx)
	size = getCount(stmt1);

	if(size != 2) {
	passed = false;
	System.out.println("FAILED as size!= 2");
	}

	tx.commit();

	try {
	//After tx commit, insert data. Should fail because of no current connection
	insertRow(stmt1);
	} catch(SQLException ex) {
	passed = true;
	System.out.println("Trying to insert data after a tx commit (prev. conn abort): Failed with : " + ex + " Expected.");
	}
	} catch (Exception e) {
	passed = false;
	} finally {
	if (rs1 != null ) {
	try { rs1.close(); } catch( Exception e1 ) {}
	}
	if ( stmt1 != null ) {
	try { stmt1.close(); } catch( Exception e1) {}
	}
	if ( conn1 != null ) {
	try { conn1.close(); } catch( Exception e1) {}
	}
	}
	return passed;
	}

	private void createTable() throws SQLException {
	String tableName = "JDBC41";

	Connection con = null;
	Statement stmt = null;
	con = ds.getConnection("scott", "tiger");
	stmt = con.createStatement();

	try{
	stmt.executeUpdate("drop table " + tableName);
	}catch(Exception ex){
	}
	stmt.executeUpdate("create table " + tableName + " (qty integer, name varchar(32))");
	stmt.close();
	con.close();
	}

	private void insertRow(Statement stmt) throws SQLException {
	String tableName = "JDBC41";
	stmt.executeUpdate("INSERT INTO " + tableName + " values (100, 'COFFEE')");
	}

	private int getCount(Statement stmt) throws SQLException {
	String tableName = "JDBC41";
	ResultSet rs = null;
	rs = stmt.executeQuery( "SELECT count(*) FROM " + tableName);
	rs.next();
	int count = rs.getInt(1);
	rs.close();
	return count;
	}

	public void ejbLoad() {}
	public void ejbStore() {}
	public void ejbRemove() {}
	public void ejbActivate() {}
	public void ejbPassivate() {}
	public void unsetEntityContext() {}
	public void ejbPostCreate() {}
	}