01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * 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.test.cts.ejb;
23:
24: import javax.ejb.EJBException;
25: import javax.ejb.SessionContext;
26: import javax.ejb.SessionBean;
27: import javax.naming.InitialContext;
28: import javax.naming.NamingException;
29: import org.jboss.logging.Logger;
30:
31: /** A session bean that asserts the count of instances active in the
32: * methodA business method do not exceed the maxActiveCount.
33: *
34: * @author Scott.Stark@jboss.org
35: * @version $Revision: 57211 $
36: */
37: public class StrictlyPooledSessionBean implements SessionBean {
38: private static Logger log = Logger
39: .getLogger(StatelessSessionBean.class);
40: /** The class wide max count of instances allows */
41: private static int maxActiveCount = 5;
42: /** The class wide count of instances active in business code */
43: private static int activeCount;
44:
45: private SessionContext ctx;
46:
47: private static synchronized int incActiveCount() {
48: return activeCount++;
49: }
50:
51: private static synchronized int decActiveCount() {
52: return activeCount--;
53: }
54:
55: public void ejbCreate() {
56: }
57:
58: public void ejbActivate() {
59: }
60:
61: public void ejbPassivate() {
62: }
63:
64: public void ejbRemove() {
65: }
66:
67: public void setSessionContext(SessionContext ctx) {
68: this .ctx = ctx;
69: try {
70: InitialContext iniCtx = new InitialContext();
71: Integer i = (Integer) iniCtx
72: .lookup("java:comp/env/maxActiveCount");
73: maxActiveCount = i.intValue();
74: } catch (NamingException e) {
75: // Use default count of 5
76: }
77: }
78:
79: public void methodA() {
80: int count = incActiveCount();
81: log.debug("Begin methodA, activeCount=" + count + ", ctx="
82: + ctx);
83: try {
84: if (count > maxActiveCount) {
85: String msg = "IllegalState, activeCount > maxActiveCount, "
86: + count + " > " + maxActiveCount;
87: throw new EJBException(msg);
88: }
89: // Sleep to let the client thread pile up
90: Thread.currentThread().sleep(1000);
91: } catch (InterruptedException e) {
92: } finally {
93: count = decActiveCount();
94: log.debug("End methodA, activeCount=" + count + ", ctx="
95: + ctx);
96: }
97: }
98: }
|