(no commit message)

[utils] / test / enterprise / src / main / java / org / wamblee / test / transactions / SimpleTransactionManager.java

diff --git a/test/enterprise/src/main/java/org/wamblee/test/transactions/SimpleTransactionManager.java b/test/enterprise/src/main/java/org/wamblee/test/transactions/SimpleTransactionManager.java
index ee8eb77dc08fb3d8a25ad5efb9bb154a86372750..7cb51431c12942a1a3edc1dcec707ec885461a63 100644 (file)
--- a/test/enterprise/src/main/java/org/wamblee/test/transactions/SimpleTransactionManager.java
+++ b/test/enterprise/src/main/java/org/wamblee/test/transactions/SimpleTransactionManager.java
@@ -20,9 +20,47 @@ import java.util.List;
 
 import javax.transaction.UserTransaction;
 
+import org.wamblee.general.ThreadSpecificProxyFactory;
+import org.wamblee.general.ThreadSpecificProxyFactory.CreationCallback;
+import org.wamblee.test.persistence.JpaBuilder;
+
 /**
- * Simple transaction manager provides a simple mechanism to manage transaction
- * in test code through the {@link UserTransaction} object.
+ * 
+ * <p>
+ * Simple transaction manager provides a simple mechanism to manage transactions
+ * in test code through the {@link UserTransaction} object. To construct the
+ * transaction manager use:
+ * </p>
+ * 
+ * <pre>
+ * SimpleTransactionManager manager = new SimpleTransactionManager(
+ *     new DefaultUserTransactionFactory());
+ * </pre>
+ * 
+ * <p>
+ * Next, add resources to manage using {@link #addResource(TransactionResource)}:
+ * 
+ * <pre>
+ * manager.addResource(jpaTester.getJpaBuilder());
+ * </pre>
+ * <p>
+ * As you can see from the above, {@link JpaBuilder} is a resource that can be
+ * used so this mechanism can be used with JPA testing.
+ * </p>
+ * 
+ * <p>
+ * The next step is to manage transactions using the standard
+ * <code>UserTransaction</code> APIs:
+ * </p>
+ * 
+ * <pre>
+ *      UserTransaction transaction = manager.getTransaction();
+ *      transaction.begin();
+ *     
+ *      ... do work...
+ *  
+ *      transaction.commit();
+ * </pre>
  * 
  * @author Erik Brakkee
  * 
@@ -30,9 +68,9 @@ import javax.transaction.UserTransaction;
 public class SimpleTransactionManager {
 
     private UserTransactionFactory factory;
-    private ThreadLocal<UserTransaction> current;
-    private UserTransactionCallback callback;
     private List<TransactionResource> resources;
+    private UserTransactionCallback transactionFInishedCallback;
+    private ThreadSpecificProxyFactory<UserTransaction> transaction;
 
     /**
      * Constructs the transaction manager.
@@ -42,15 +80,23 @@ public class SimpleTransactionManager {
      */
     public SimpleTransactionManager(UserTransactionFactory aFactory) {
         factory = aFactory;
-        current = new ThreadLocal<UserTransaction>();
-        callback = new UserTransactionCallback() {
 
+        resources = new ArrayList<TransactionResource>();
+
+        transactionFInishedCallback = new UserTransactionCallback() {
             @Override
             public void transactionFinished() {
-                current.set(null);
+                transaction.set(factory.create(this, resources));
             }
         };
-        resources = new ArrayList<TransactionResource>();
+        transaction = new ThreadSpecificProxyFactory<UserTransaction>(
+            UserTransaction.class, new CreationCallback<UserTransaction>() {
+                @Override
+                public UserTransaction create() {
+                    return factory.create(transactionFInishedCallback,
+                        resources);
+                }
+            });
     }
 
     /**
@@ -65,16 +111,23 @@ public class SimpleTransactionManager {
     }
 
     /**
-     * Gets a transaction associated with the current thread.
+     * Gets the user transaction. This is a contextual reference, meaning that
+     * it will delegate to the appropriate thread-specific user transaction. It
+     * is also safe to store in a JNDI tree and for caching by applications.
      * 
      * @return User transaction.
      */
     public UserTransaction getTransaction() {
-        UserTransaction transaction = current.get();
-        if (transaction == null) {
-            transaction = factory.create(callback, resources);
-            current.set(transaction);
-        }
-        return transaction;
+        return transaction.getProxy();
+    }
+
+    /**
+     * Gets the thread-specific transaction object.
+     * 
+     * @return Transaction object.
+     */
+    UserTransaction getThreadSpecificTransaction() {
+        getTransaction(); // create tx if needed
+        return transaction.get();
     }
 }