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.idgen.ejb;
23:
24: import java.rmi.*;
25: import javax.naming.*;
26: import javax.ejb.*;
27:
28: import org.jboss.test.util.ejb.SessionSupport;
29: import org.jboss.test.idgen.interfaces.*;
30:
31: /**
32: *
33: * @author Rickard Oberg
34: * @author $Author: dimitris@jboss.org $
35: * @version $Revision: 57211 $
36: */
37: public class IdGeneratorBean extends SessionSupport {
38: IdCounterHome counterHome;
39:
40: static final String SIZE = "java:comp/env/size";
41:
42: public long getNewId(String beanName) throws RemoteException {
43: IdCounter counter;
44:
45: // Acquire counter
46: try {
47: counter = counterHome.findByPrimaryKey(beanName);
48: } catch (FinderException e) {
49: try {
50: counter = counterHome.create(beanName);
51: } catch (CreateException ex) {
52: throw new EJBException(
53: "Could not find or create counter for "
54: + beanName);
55: }
56: }
57:
58: // Get id
59: return counter.getNextValue();
60: }
61:
62: public void setSessionContext(SessionContext context) {
63: super .setSessionContext(context);
64:
65: try {
66: counterHome = (IdCounterHome) new InitialContext()
67: .lookup("java:comp/env/ejb/IdCounter");
68: } catch (Exception e) {
69: throw new EJBException(e);
70: }
71: }
72: }
|