SimpleBMPBean.java

package com.sun.s1asdev.jdbc.simpleora.ejb;

import javax.ejb.*;
import javax.naming.*;
import javax.sql.*;
import java.rmi.*;
import java.util.*;
import java.sql.*;

public class SimpleBMPBean
    implements EntityBean
{

    protected DataSource ds;
    protected DataSource ds1;

    public void setEntityContext(EntityContext entityContext) {
	Context context = null;
	try {
	    context    = new InitialContext();
	    ds = (DataSource) context.lookup("java:comp/env/DataSource");
	    ds1 = (DataSource) context.lookup("java:comp/env/DataSource1");
	} catch (NamingException e) {
	    throw new EJBException("cant find datasource");
	}
    }

    public Integer ejbCreate() throws CreateException {
	return new Integer(1);
    }

    /* Get 10 connections in a bunch and then close them*/
    public boolean test1( int numRuns ) {
	boolean passed = true;
	Connection[] conns = new Connection[10];
        for( int i = 0; i < numRuns; i++ ) {
	    try {
	        conns[i] = ds.getConnection();
	    } catch (Exception e) {
	        passed = false;
	    } 
        }
        
	for (int i = 0 ; i < numRuns;i++ ) {
	    try {
	        conns[i].close();
	    } catch( Exception e) {
	        passed = false;
	    }
	}
	return passed;
    }

    /* Get a single connection and close it */
    public boolean test2() {
        Connection conn = null;
	Statement stmt = null;
	ResultSet rs = null;
        boolean passed = true;
	try {
	    conn = ds.getConnection();
	    conn.setTransactionIsolation( conn.getTransactionIsolation() );
	    stmt = conn.createStatement();
	    rs = stmt.executeQuery( "SELECT * FROM O_CUSTOMER");
	    while( rs.next() ) {
	        System.out.println(rs.getString(1));
	    }
	    stmt.executeUpdate("INSERT INTO O_CUSTOMER values (100, 'new phone')");
            rs = stmt.executeQuery( "SELECT * FROM O_CUSTOMER");
	    while( rs.next() ) {
	        System.out.println(rs.getString(1));
	    }
            stmt.executeUpdate("DELETE FROM O_CUSTOMER WHERE c_id=100");
	} catch (Exception e) {
	    passed = false;
	    e.printStackTrace();
	} finally {
	    if (rs != null ) { 
	        try { rs.close(); } catch( Exception e1 ) {}
	    }	
	    if (stmt != null ) { 
	        try { stmt.close(); } catch( Exception e1 ) {}
	    }	
	    if ( conn != null ) {
	        try {
	            conn.close();
	        } catch( Exception e1) {}    
	    }
	    
        } 
        
	return passed;
    }

    /** Application auth */
    public boolean test3() {
        Connection conn = null;
	Statement stmt = null;
	ResultSet rs = null;

        boolean passed = true;
	try {
	    conn = ds1.getConnection("scott", "tiger");
	    stmt = conn.createStatement();
	    stmt.executeUpdate("INSERT INTO O_CUSTOMER values (100, 'new phone')");
            rs = stmt.executeQuery( "SELECT * FROM O_CUSTOMER");
	    while( rs.next() ) {
	        System.out.println(rs.getString(1));
	    }
            stmt.executeUpdate("DELETE FROM O_CUSTOMER WHERE c_id=100");
            rs = stmt.executeQuery( "SELECT * FROM O_CUSTOMER");
	    System.out.println( rs );
	} catch (Exception e) {
	    passed = false;
	    e.printStackTrace();
	} finally {
	    if (rs != null ) { 
	        try { rs.close(); } catch( Exception e1 ) {}
	    }	
	    if (stmt != null ) { 
	        try { stmt.close(); } catch( Exception e1 ) {}
	    }	
	    if ( conn != null ) {
	        try {
	            conn.close();
	        } catch( Exception e1) {}    
	    }
        } 
        
	return passed;
    }
  

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