KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > aspects > tx > TransactionManagerInjector


1 /*
2   * JBoss, Home of Professional Open Source
3   * Copyright 2005, JBoss Inc., and individual contributors as indicated
4   * by the @authors tag. See the copyright.txt in the distribution for a
5   * full listing of individual contributors.
6   *
7   * This is free software; you can redistribute it and/or modify it
8   * under the terms of the GNU Lesser General Public License as
9   * published by the Free Software Foundation; either version 2.1 of
10   * the License, or (at your option) any later version.
11   *
12   * This software is distributed in the hope that it will be useful,
13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15   * Lesser General Public License for more details.
16   *
17   * You should have received a copy of the GNU Lesser General Public
18   * License along with this software; if not, write to the Free
19   * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20   * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21   */
22 package org.jboss.aspects.tx;
23
24 import org.jboss.aop.annotation.AnnotationElement;
25 import org.jboss.aop.joinpoint.ConstructorInvocation;
26 import org.jboss.aop.joinpoint.FieldReadInvocation;
27 import org.jboss.aop.joinpoint.FieldWriteInvocation;
28 import org.jboss.aspects.Injected;
29 import org.jboss.tm.TransactionManagerLocator;
30
31 import javax.transaction.TransactionManager JavaDoc;
32 import java.lang.reflect.Method JavaDoc;
33
34 /**
35  * This aspect should be scoped PER_JOINPOINT
36  * It allows a field to be like a ThreadLocal
37  *
38  * @author <a HREF="mailto:bill@jboss.org">Bill Burke</a>
39  * @version $Revision: 43966 $
40  */
41 public class TransactionManagerInjector
42 {
43    public Object JavaDoc access(FieldReadInvocation invocation) throws Throwable JavaDoc
44    {
45       return TransactionManagerLocator.getInstance().locate();
46    }
47
48    public Object JavaDoc access(FieldWriteInvocation invocation) throws Throwable JavaDoc
49    {
50       throw new RuntimeException JavaDoc("It is illegal to set an injected TransactionManager field.");
51    }
52
53    public Object JavaDoc allocation(ConstructorInvocation invocation) throws Throwable JavaDoc
54    {
55       Object JavaDoc obj = invocation.invokeNext();
56
57       try
58       {
59          Object JavaDoc[] arg = {TransactionManagerLocator.getInstance().locate()};
60          Method JavaDoc[] methods = obj.getClass().getMethods();
61          for (int i = 0; i < methods.length; i++)
62          {
63             if (methods[i].getParameterTypes().length == 1)
64             {
65                if (methods[i].getParameterTypes()[0].equals(TransactionManager JavaDoc.class))
66                {
67                   if (AnnotationElement.isAnyAnnotationPresent(methods[i], Injected.class))
68                   {
69                      methods[i].invoke(obj, arg);
70                   }
71                }
72             }
73          }
74       }
75       catch (Exception JavaDoc e)
76       {
77          throw new RuntimeException JavaDoc(e); //To change body of catch statement use Options | File Templates.
78 }
79       return obj;
80    }
81 }
82
Popular Tags