001: /*
002: * JBoss, Home of Professional Open Source.
003: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
004: * as indicated by the @author tags. See the copyright.txt file in the
005: * distribution for a full listing of individual contributors.
006: *
007: * This is free software; you can redistribute it and/or modify it
008: * under the terms of the GNU Lesser General Public License as
009: * published by the Free Software Foundation; either version 2.1 of
010: * the License, or (at your option) any later version.
011: *
012: * This software is distributed in the hope that it will be useful,
013: * but WITHOUT ANY WARRANTY; without even the implied warranty of
014: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015: * Lesser General Public License for more details.
016: *
017: * You should have received a copy of the GNU Lesser General Public
018: * License along with this software; if not, write to the Free
019: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
020: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
021: */
022: package org.jboss.test.jca.ejb;
023:
024: import java.rmi.RemoteException;
025: import java.sql.Connection;
026:
027: import javax.ejb.CreateException;
028: import javax.ejb.EJBException;
029: import javax.ejb.EntityBean;
030: import javax.ejb.EntityContext;
031: import javax.naming.InitialContext;
032: import javax.sql.DataSource;
033:
034: import org.jboss.test.jca.interfaces.Reentrant;
035:
036: /**
037: * ReentrantBean.java tests if CachedConnectionManager works with reentrant ejbs.
038: *
039: * @author <a href="mailto:d_jencks@users.sourceforge.net">David Jencks</a>
040: * @version <tt>$Revision: 57211 $</tt>
041: *
042: * @ejb.bean
043: * jndi-name="ejb/jca/Reentrant"
044: * name="Reentrant"
045: * type="BMP"
046: * view-type="remote"
047: * reentrant="true"
048: * @ejb.pk class="java.lang.Integer"
049: * @ejb.transaction
050: * type="Required"
051: *
052: */
053:
054: public class ReentrantBean implements EntityBean {
055: /** The serialVersionUID */
056: private static final long serialVersionUID = 1L;
057:
058: private Integer id;
059:
060: private EntityContext ctx;
061:
062: public ReentrantBean() {
063:
064: }
065:
066: /**
067: * Creates a new <code>ejbCreate</code> instance.
068: *
069: * @param id an <code>Integer</code> value
070: * @param other a <code>Reentrant</code> value
071: * @exception CreateException if an error occurs
072: * @exception RemoteException if an error occurs
073: *
074: * @ejb.create-method
075: */
076: public Integer ejbCreate(Integer id, Reentrant other)
077: throws CreateException, RemoteException {
078: this .id = id;
079: return id;
080: }
081:
082: /**
083: * Creates a new <code>ejbPostCreate</code> instance.
084: *
085: * @param id an <code>Integer</code> value
086: * @param other a <code>Reentrant</code> value
087: * @exception CreateException if an error occurs
088: * @exception RemoteException if an error occurs
089: */
090: public void ejbPostCreate(Integer id, Reentrant other)
091: throws CreateException, RemoteException {
092: this .id = id;
093: Reentrant me = (Reentrant) ctx.getEJBObject();
094: Connection c = null;
095: try {
096: try {
097: DataSource ds = (DataSource) new InitialContext()
098: .lookup("java:/DefaultDS");
099: c = ds.getConnection();
100: if (other != null) {
101: other.doSomething(me);
102: }
103: } finally {
104: c.close();
105: }
106: } catch (Exception e) {
107: throw new CreateException(
108: "could not get DataSource or Connection"
109: + e.getMessage());
110: }
111: }
112:
113: /**
114: * Describe <code>doSomething</code> method here.
115: *
116: * @param first a <code>Reentrant</code> value
117: * @exception RemoteException if an error occurs
118: *
119: * @ejb.interface-method
120: */
121: public void doSomething(Reentrant first) throws RemoteException {
122: if (first != null) {
123: first.doSomething(null);
124: }
125: }
126:
127: public Integer ejbFindByPrimaryKey(Integer id) {
128: return id;
129: }
130:
131: // implementation of javax.ejb.EntityBean interface
132:
133: public void ejbActivate() {
134: }
135:
136: public void ejbLoad() {
137: this .id = (Integer) ctx.getPrimaryKey();
138: }
139:
140: public void ejbPassivate() {
141: }
142:
143: public void ejbRemove() throws EJBException {
144: }
145:
146: public void ejbStore() throws EJBException {
147: }
148:
149: public void setEntityContext(EntityContext ctx) {
150: this .ctx = ctx;
151: }
152:
153: public void unsetEntityContext() {
154: ctx = null;
155: }
156:
157: public String toString() {
158: if (id == null)
159: return null;
160: else
161: return id.toString();
162: }
163: }
|