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.ejb.plugins.cmp.ejbql;
23:
24: import java.util.Collections;
25: import java.util.HashMap;
26: import java.util.Map;
27: import java.util.Set;
28:
29: import org.jboss.ejb.plugins.cmp.bridge.EntityBridge;
30:
31: /**
32: * This class maintains a map of all entitie bridges in an application by name.
33: *
34: * @author <a href="mailto:dain@daingroup.com">Dain Sundstrom</a>
35: * @version $Revision: 57209 $
36: */
37: public final class Catalog {
38: private final Map entityByAbstractSchemaName = new HashMap();
39: private final Map entityByEJBName = new HashMap();
40: private final Map entityByInterface = new HashMap();
41:
42: public void addEntity(EntityBridge entityBridge) {
43: entityByAbstractSchemaName.put(entityBridge
44: .getAbstractSchemaName(), entityBridge);
45: entityByEJBName.put(entityBridge.getEntityName(), entityBridge);
46:
47: Class remote = entityBridge.getRemoteInterface();
48: if (remote != null) {
49: entityByInterface.put(remote, entityBridge);
50: }
51: Class local = entityBridge.getLocalInterface();
52: if (local != null) {
53: entityByInterface.put(local, entityBridge);
54: }
55:
56: }
57:
58: public EntityBridge getEntityByAbstractSchemaName(
59: String abstractSchemaName) {
60: return (EntityBridge) entityByAbstractSchemaName
61: .get(abstractSchemaName);
62: }
63:
64: public EntityBridge getEntityByInterface(Class intf) {
65: return (EntityBridge) entityByInterface.get(intf);
66: }
67:
68: public EntityBridge getEntityByEJBName(String ejbName) {
69: return (EntityBridge) entityByEJBName.get(ejbName);
70: }
71:
72: public int getEntityCount() {
73: return entityByEJBName.size();
74: }
75:
76: public Set getEJBNames() {
77: return Collections.unmodifiableSet(entityByEJBName.keySet());
78: }
79: }
|