User: salborini
Date: 00/09/25 19:35:34
Added: src/java/org/jboss/zol/testbean/bean BMTStatefulBean.java
BMTStatelessBean.java
Log:
New tests for bean managed transaction demarcation.
Won't work for now, but I have to commit for Marc to finish testing.
Revision Changes Path
1.1 zola/src/java/org/jboss/zol/testbean/bean/BMTStatefulBean.java
Index: BMTStatefulBean.java
===================================================================
package org.jboss.zol.testbean.bean;
import java.rmi.*;
import javax.ejb.*;
import javax.naming.InitialContext;
import javax.naming.Context;
import org.jboss.zol.testbean.interfaces.BMTStateful;
import javax.transaction.UserTransaction;
import javax.transaction.Status;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.Statement;
import java.sql.ResultSet;
import javax.sql.DataSource;
import java.sql.SQLException;
public class BMTStatefulBean implements SessionBean {
private SessionContext sessionContext;
public void ejbCreate() throws RemoteException, CreateException {
System.out.println("BMTStatefulBean.ejbCreate() called");
}
public void ejbActivate() throws RemoteException {
System.out.println("BMTStatefulBean.ejbActivate() called");
}
public void ejbPassivate() throws RemoteException {
System.out.println("BMTStatefulBean.ejbPassivate() called");
}
public void ejbRemove() throws RemoteException {
System.out.println("BMTStatefulBean.ejbRemove() called");
}
public void setSessionContext(SessionContext context) throws RemoteException {
sessionContext = context;
}
public String txExists() throws RemoteException {
String result = "";
try {
UserTransaction ut1 = sessionContext.getUserTransaction();
result += "Got UserTransaction via
sessionContext.getUserTransaction(): "+ statusName(ut1.getStatus()) + "\n";
UserTransaction ut2 = (UserTransaction)new
InitialContext().lookup("java:comp/UserTransaction");
result += "Got UserTransaction via
lookup(java:comp/UserTransaction): "+ statusName(ut2.getStatus()) + "\n";
return result;
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String txCommit() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
String result = "Got transaction : " +
statusName(tx.getStatus()) + "\n";
tx.begin();
result += "tx.begin(): " + statusName(tx.getStatus()) + "\n";
tx.commit();
result += "tx.commit(): " + statusName(tx.getStatus()) + "\n";
return result;
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String txRollback() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
String result = "Got transaction : " +
statusName(tx.getStatus()) + "\n";
tx.begin();
result += "tx.begin(): " + statusName(tx.getStatus()) + "\n";
tx.rollback();
result += "tx.rollback(): " + statusName(tx.getStatus()) +
"\n";
return result;
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String txBegin() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
tx.begin();
return "status: " + statusName(tx.getStatus());
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String txEnd() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
tx.commit();
return "status: " + statusName(tx.getStatus());
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public void createTable() throws RemoteException {
try {
Connection connection = ((DataSource)new
InitialContext().lookup("java:comp/env/jdbc/myDatabase")).getConnection();
Statement stm = connection.createStatement();
try {
stm.executeQuery("CREATE TABLE bmttest (field
VARCHAR(256))");
} catch (SQLException e) {
// ignore, table probably already exists
}
stm.executeQuery("INSERT INTO bmttest VALUES ('initial
value')");
connection.close();
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public void dropTable() throws RemoteException {
try {
Connection connection = ((DataSource)new
InitialContext().lookup("java:comp/env/jdbc/myDatabase")).getConnection();
Statement stm = connection.createStatement();
stm.executeQuery("DROP TABLE bmttest");
connection.close();
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String dbCommit() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
tx.begin();
Connection connection = ((DataSource)new
InitialContext().lookup("java:comp/env/jdbc/myDatabase")).getConnection();
Statement stm = connection.createStatement();
stm.executeQuery("UPDATE bmttest SET field = 'updated via
dbCommit'");
connection.close();
tx.commit();
return statusName(tx.getStatus());
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String dbRollback() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
tx.begin();
Connection connection = ((DataSource)new
InitialContext().lookup("java:comp/env/jdbc/myDatabase")).getConnection();
Statement stm = connection.createStatement();
stm.executeQuery("UPDATE bmttest SET field = 'updated via
dbRollback'");
connection.close();
tx.rollback();
return statusName(tx.getStatus());
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String getDbField() throws RemoteException {
try {
Connection connection = ((DataSource)new
InitialContext().lookup("java:comp/env/jdbc/myDatabase")).getConnection();
Statement stm = connection.createStatement();
ResultSet rs = stm.executeQuery("SELECT field FROM bmttest ");
String result = "not found";
if (rs.next()) {
result = rs.getString(1);
}
connection.close();
return result;
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
private String statusName(int s) {
switch (s) {
case Status.STATUS_ACTIVE: return "STATUS_ACTIVE";
case Status.STATUS_COMMITTED: return "STATUS_COMMITED";
case Status.STATUS_COMMITTING: return "STATUS_COMMITTING";
case Status.STATUS_MARKED_ROLLBACK: return
"STATUS_MARKED_ROLLBACK";
case Status.STATUS_NO_TRANSACTION: return
"STATUS_NO_TRANSACTION";
case Status.STATUS_PREPARED: return "STATUS_PREPARED";
case Status.STATUS_PREPARING: return "STATUS_PREPARING";
case Status.STATUS_ROLLEDBACK: return "STATUS_ROLLEDBACK";
case Status.STATUS_ROLLING_BACK: return "STATUS_ROLLING_BACK";
case Status.STATUS_UNKNOWN: return "STATUS_UNKNOWN";
}
return "REALLY_UNKNOWN";
}
}
1.1 zola/src/java/org/jboss/zol/testbean/bean/BMTStatelessBean.java
Index: BMTStatelessBean.java
===================================================================
package org.jboss.zol.testbean.bean;
import java.rmi.*;
import javax.ejb.*;
import javax.naming.InitialContext;
import javax.naming.Context;
import org.jboss.zol.testbean.interfaces.BMTStateful;
import javax.transaction.UserTransaction;
import javax.transaction.Status;
public class BMTStatelessBean implements SessionBean {
private SessionContext sessionContext;
public void ejbCreate() throws RemoteException, CreateException {
System.out.println("BMTStatelessBean.ejbCreate() called");
}
public void ejbActivate() throws RemoteException {
System.out.println("BMTStatelessBean.ejbActivate() called");
}
public void ejbPassivate() throws RemoteException {
System.out.println("BMTStatelessBean.ejbPassivate() called");
}
public void ejbRemove() throws RemoteException {
System.out.println("BMTStatelessBean.ejbRemove() called");
}
public void setSessionContext(SessionContext context) throws RemoteException {
sessionContext = context;
}
public String txExists() throws RemoteException {
String result = "";
try {
UserTransaction ut1 = sessionContext.getUserTransaction();
result += "Got UserTransaction via
sessionContext.getUserTransaction()\n";
UserTransaction ut2 = (UserTransaction)new
InitialContext().lookup("java:comp/UserTransaction");
result += "Got UserTransaction via
lookup(java:comp/UserTransaction)\n";
return result;
} catch (Exception e) {
throw new RemoteException(e.getMessage());
}
}
public String txCommit() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
String result = "Got transaction : " +
statusName(tx.getStatus()) + "\n";
tx.begin();
result += "tx.begin(): " + statusName(tx.getStatus()) + "\n";
tx.commit();
result += "tx.commit(): " + statusName(tx.getStatus()) + "\n";
return result;
} catch (Exception e) {
e.printStackTrace();
throw new RemoteException(e.getMessage());
}
}
public String txRollback() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
String result = "Got transaction : " +
statusName(tx.getStatus()) + "\n";
tx.begin();
result += "tx.begin(): " + statusName(tx.getStatus()) + "\n";
tx.rollback();
result += "tx.rollback(): " + statusName(tx.getStatus()) +
"\n";
return result;
} catch (Exception e) {
throw new RemoteException(e.getMessage());
}
}
// this should not be allowed by the container
public String txBegin() throws RemoteException {
try {
UserTransaction tx = sessionContext.getUserTransaction();
tx.begin();
return "status: " + statusName(tx.getStatus());
} catch (Exception e) {
throw new RemoteException(e.getMessage());
}
}
private String statusName(int s) {
switch (s) {
case Status.STATUS_ACTIVE: return "STATUS_ACTIVE";
case Status.STATUS_COMMITTED: return "STATUS_COMMITED";
case Status.STATUS_COMMITTING: return "STATUS_COMMITTING";
case Status.STATUS_MARKED_ROLLBACK: return
"STATUS_MARKED_ROLLBACK";
case Status.STATUS_NO_TRANSACTION: return
"STATUS_NO_TRANSACTION";
case Status.STATUS_PREPARED: return "STATUS_PREPARED";
case Status.STATUS_PREPARING: return "STATUS_PREPARING";
case Status.STATUS_ROLLEDBACK: return "STATUS_ROLLEDBACK";
case Status.STATUS_ROLLING_BACK: return "STATUS_ROLLING_BACK";
case Status.STATUS_UNKNOWN: return "STATUS_UNKNOWN";
}
return "REALLY_UNKNOWN";
}
}
