TestBean.java

package test;

import javax.ejb.*;
import javax.naming.*;
import javax.sql.DataSource;

import java.sql.*;

import com.sun.jdo.api.persistence.support.PersistenceManagerFactory;

/**
 * This SessionBean is used to test Read-Only CMP beans by inserting
 * initial data into the table and updating column 'NAME' when requested.
 * This allows to use java2db for the actual beans.
 * This bean does not access CMP beans.
 */ 
public class TestBean implements SessionBean {

    private DataSource ds = null;

    // SessionBean methods
 
    public void ejbCreate() throws CreateException {
        System.out.println("TestBean ejbCreate");
        try {
            ds = lookupDataSource();
 
        } catch (NamingException ex) {
            throw new EJBException(ex.getMessage());
        }
    }    
 
    public void ejbActivate() {
        System.out.println("TestBean ejbActivate");
    }    

    public void ejbPassivate() {
            ds = null;
    }

    public void ejbRemove() {

    }
    
    public void setSessionContext(SessionContext sc) {

    }

    /** Look up a DataSource by JNDI name.
     * The JNDI name is expected to be 'jdo/pmf', but it can reference
     * either a PersistenceManagerFactory or a DataSource. In case of
     * a former, the DataSource name will be a ConnectionFactory's name.
     */
    private DataSource lookupDataSource() throws NamingException {
        Context initial = new InitialContext();
        Object objref = initial.lookup("jdo/pmf");

        if (objref instanceof PersistenceManagerFactory) {
            PersistenceManagerFactory pmf = (PersistenceManagerFactory)objref;
            String cfname = pmf.getConnectionFactoryName();

            System.out.println("DATASOURCE NAME: " + cfname);
            objref = initial.lookup(cfname);
        } 
        return (DataSource) objref;
    }    

    /** Insert values via jdbc call */
    public void insertValues (String table_name) {
        String st = "INSERT INTO " + table_name + " VALUES ('" + 
                table_name + "', '" + table_name + "', '" + table_name + "')";
        System.out.println("INSERT STATEMENT: " + st);
        executeStatement(st);

        // Insert another row
        st = "INSERT INTO " + table_name + " VALUES ('" + 
                table_name + "1', '" + table_name + "1', '" + table_name + "1')";
        System.out.println("INSERT STATEMENT: " + st);
        executeStatement(st);
    }

    /** Update values via jdbc call */
    public void updateValues (String table_name) {
        String st = "UPDATE " + table_name + " SET SHORTNAME = 'FOO' WHERE ID = '" + table_name + "'";
        System.out.println("UPDATE STATEMENT: " + st);
        executeStatement(st);
    }

    /** Execute SQL statement.
     * @param st the SQL statement as a String.
     * @throws EJBException to wrap a SQLException.
     */
    private void executeStatement (String st) {
        Connection conn = null;
        try {
            conn = ds.getConnection();
            PreparedStatement ps = conn.prepareStatement(st);
            ps.execute();
        } catch (SQLException e) {
            e.printStackTrace();
            throw new EJBException(e.toString());
        } finally {
            try {
                if (conn != null) conn.close();
            } catch (SQLException e1) {
                e1.printStackTrace();
            }
        }
    }

}