KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > openejb > ri > sp > PseudoTransactionService


1 /**
2  * Redistribution and use of this software and associated documentation
3  * ("Software"), with or without modification, are permitted provided
4  * that the following conditions are met:
5  *
6  * 1. Redistributions of source code must retain copyright
7  * statements and notices. Redistributions must also contain a
8  * copy of this document.
9  *
10  * 2. Redistributions in binary form must reproduce the
11  * above copyright notice, this list of conditions and the
12  * following disclaimer in the documentation and/or other
13  * materials provided with the distribution.
14  *
15  * 3. The name "Exolab" must not be used to endorse or promote
16  * products derived from this Software without prior written
17  * permission of Exoffice Technologies. For written permission,
18  * please contact info@exolab.org.
19  *
20  * 4. Products derived from this Software may not be called "Exolab"
21  * nor may "Exolab" appear in their names without prior written
22  * permission of Exoffice Technologies. Exolab is a registered
23  * trademark of Exoffice Technologies.
24  *
25  * 5. Due credit should be given to the Exolab Project
26  * (http://www.exolab.org/).
27  *
28  * THIS SOFTWARE IS PROVIDED BY EXOFFICE TECHNOLOGIES AND CONTRIBUTORS
29  * ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
30  * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
31  * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
32  * EXOFFICE TECHNOLOGIES OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
33  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
34  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
35  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
36  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
37  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
38  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
39  * OF THE POSSIBILITY OF SUCH DAMAGE.
40  *
41  * Copyright 1999 (C) Exoffice Technologies Inc. All Rights Reserved.
42  *
43  * $Id: PseudoTransactionService.java 2487 2006-02-22 22:05:03Z dblevins $
44  */

45 package org.openejb.ri.sp;
46
47 import java.util.Enumeration JavaDoc;
48 import java.util.Hashtable JavaDoc;
49 import java.util.Vector JavaDoc;
50 import javax.transaction.Status JavaDoc;
51 import javax.transaction.Synchronization JavaDoc;
52 import javax.transaction.Synchronization JavaDoc;
53 import javax.transaction.Transaction JavaDoc;
54 import javax.transaction.TransactionManager JavaDoc;
55 import javax.transaction.RollbackException JavaDoc;
56 import javax.transaction.UserTransaction JavaDoc;
57 import javax.transaction.xa.XAResource JavaDoc;
58 import org.openejb.spi.TransactionService;
59
60 public class PseudoTransactionService implements TransactionService {
61     TransactionManager JavaDoc txm = new MyTransactionManager();
62     Hashtable JavaDoc map = new Hashtable JavaDoc();
63     
64     public void init(java.util.Properties JavaDoc props ) {
65         props = props;
66     }
67     
68     public TransactionManager JavaDoc getTransactionManager( ){
69         return txm;
70     }
71     public class MyTransactionManager implements TransactionManager JavaDoc{
72         public UserTransaction JavaDoc getUserTransaction(Object JavaDoc txID){
73             return new UserTransaction JavaDoc( ){
74                 public void begin() {MyTransactionManager.this.begin();}
75                 public void commit() throws RollbackException JavaDoc {MyTransactionManager.this.commit();}
76                 public int getStatus()throws javax.transaction.SystemException JavaDoc{
77                     return MyTransactionManager.this.getStatus();
78                 }
79                 public void rollback() {MyTransactionManager.this.rollback();}
80                 public void setRollbackOnly() {MyTransactionManager.this.setRollbackOnly();}
81                 public void setTransactionTimeout(int seconds) {MyTransactionManager.this.setTransactionTimeout(seconds);}
82             };
83         }
84         public void begin( ){
85             Transaction JavaDoc tx = new MyTransaction( );
86             map.put(Thread.currentThread(), tx);
87         }
88         public void commit() throws RollbackException JavaDoc {
89             MyTransaction tx = (MyTransaction)map.remove(Thread.currentThread());
90             if(tx!=null)
91                 tx.commit();
92             else
93                 throw new IllegalStateException JavaDoc();
94         }
95         public int getStatus()throws javax.transaction.SystemException JavaDoc{
96             Transaction JavaDoc tx = (Transaction JavaDoc)map.get(Thread.currentThread());
97             if(tx==null)return Status.STATUS_NO_TRANSACTION;
98             return tx.getStatus();
99         }
100         public Transaction JavaDoc getTransaction( ){
101             return (Transaction JavaDoc)map.get(Thread.currentThread());
102         }
103         public void resume(Transaction JavaDoc tx)
104         throws javax.transaction.SystemException JavaDoc, javax.transaction.InvalidTransactionException JavaDoc{
105             Transaction JavaDoc ctx = (Transaction JavaDoc)map.get(Thread.currentThread());
106             int status = tx.getStatus();
107             // allow to resume a tx that has been marked for rollback.
108
if(ctx!= null || tx == null || (status != Status.STATUS_ACTIVE && status != Status.STATUS_MARKED_ROLLBACK))
109                 throw new javax.transaction.InvalidTransactionException JavaDoc();
110             map.put(Thread.currentThread(),tx);
111         }
112         public Transaction JavaDoc suspend( ){
113             return (Transaction JavaDoc)map.remove(Thread.currentThread());
114         }
115         public void rollback(){
116             MyTransaction tx = (MyTransaction)map.remove(Thread.currentThread());
117             if(tx==null) throw new IllegalStateException JavaDoc();
118             tx.rollback();
119         }
120         public void setRollbackOnly( ){
121             MyTransaction tx = (MyTransaction)map.get(Thread.currentThread());
122             if(tx==null) throw new IllegalStateException JavaDoc();
123             tx.setRollbackOnly();
124         }
125         public void setTransactionTimeout(int x){}
126     
127     }
128     public class MyTransaction implements Transaction JavaDoc {
129         Vector JavaDoc registeredSynchronizations = new Vector JavaDoc();
130         Vector JavaDoc xaResources = new Vector JavaDoc();
131         int status = Status.STATUS_ACTIVE;
132         public void commit() throws RollbackException JavaDoc {
133         if ( status == Status.STATUS_MARKED_ROLLBACK ) {
134         rollback();
135         throw new RollbackException JavaDoc();
136         }
137             doBeforeCompletion();
138             doXAResources(Status.STATUS_COMMITTED);
139             status = Status.STATUS_COMMITTED;
140             doAfterCompletion(Status.STATUS_COMMITTED);
141             registeredSynchronizations = new Vector JavaDoc();
142             map.remove(Thread.currentThread());
143         }
144         public boolean delistResource(XAResource JavaDoc xaRes, int flag) {
145             xaResources.remove(xaRes); return true;
146         }
147         public boolean enlistResource(XAResource JavaDoc xaRes){
148             xaResources.addElement(xaRes);return true;
149         }
150         public int getStatus() {return status;}
151         
152         public void registerSynchronization(Synchronization JavaDoc sync){
153             registeredSynchronizations.addElement(sync);
154         }
155         public void rollback() {
156             doXAResources(Status.STATUS_ROLLEDBACK);
157             doAfterCompletion(Status.STATUS_ROLLEDBACK);
158             status = Status.STATUS_ROLLEDBACK;
159             registeredSynchronizations = new Vector JavaDoc();
160             map.remove(Thread.currentThread());
161         }
162         public void setRollbackOnly() {status = Status.STATUS_MARKED_ROLLBACK;}
163         
164         // the transaciton must be NOT be rolleback for this method to execute.
165
private void doBeforeCompletion(){
166             Enumeration JavaDoc enumeration = registeredSynchronizations.elements();
167             while(enumeration.hasMoreElements()){
168                 try{
169                 Synchronization JavaDoc sync = (Synchronization JavaDoc)enumeration.nextElement();
170                 sync.beforeCompletion();
171                 }catch(RuntimeException JavaDoc re){
172                     re.printStackTrace();
173                 }
174             }
175         }
176         private void doAfterCompletion(int status){
177             Enumeration JavaDoc enumeration = registeredSynchronizations.elements();
178             while(enumeration.hasMoreElements()){
179                 try{
180                 Synchronization JavaDoc sync = (Synchronization JavaDoc)enumeration.nextElement();
181                 sync.afterCompletion(status);
182                 }catch(RuntimeException JavaDoc re){
183                     re.printStackTrace();
184                 }
185             }
186         }
187         
188         private void doXAResources(int status){
189             Object JavaDoc [] resources = xaResources.toArray();
190             for(int i = 0; i < resources.length; i++){
191                 XAResource JavaDoc xaRes = (XAResource JavaDoc)resources[i];
192                 if(status == Status.STATUS_COMMITTED){
193                     try{
194                     xaRes.commit(null,true);
195                     }catch(javax.transaction.xa.XAException JavaDoc xae){
196                         // do nothing.
197
}
198                     try{
199                     xaRes.end(null,XAResource.TMSUCCESS);
200                     }catch(javax.transaction.xa.XAException JavaDoc xae){
201                         // do nothing.
202
}
203                 }else{
204                     try{
205                     xaRes.rollback(null);
206                     }catch(javax.transaction.xa.XAException JavaDoc xae){
207                         // do nothing.
208
}
209                     try{
210                     xaRes.end(null,XAResource.TMFAIL);
211                     }catch(javax.transaction.xa.XAException JavaDoc xae){
212                         // do nothing.
213
}
214                 }
215             }
216             xaResources = new Vector JavaDoc();
217             
218         }
219         
220     }
221     
222 }
223
Popular Tags