KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > test > idgen > ejb > IdGeneratorBean


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.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  * @see <related>
34  * @author $Author: starksm $
35  * @version $Revision: 37406 $
36  */
37 public class IdGeneratorBean
38    extends SessionSupport
39 {
40    IdCounterHome counterHome;
41    
42    static final String JavaDoc SIZE = "java:comp/env/size";
43
44    public long getNewId(String JavaDoc beanName)
45       throws RemoteException
46    {
47       IdCounter counter;
48       
49       // Acquire counter
50 try {
51          counter = counterHome.findByPrimaryKey(beanName);
52       }
53       catch (FinderException e) {
54          try {
55             counter = counterHome.create(beanName);
56          } catch (CreateException ex) {
57             throw new EJBException("Could not find or create counter for "+beanName);
58          }
59       }
60       
61       // Get id
62 return counter.getNextValue();
63    }
64
65    public void setSessionContext(SessionContext context)
66    {
67       super.setSessionContext(context);
68       
69       try {
70          counterHome = (IdCounterHome)new InitialContext().lookup("java:comp/env/ejb/IdCounter");
71       }
72       catch (Exception JavaDoc e) {
73          throw new EJBException(e);
74       }
75    }
76 }
77
Popular Tags