001/**
002 *   GRANITE DATA SERVICES
003 *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004 *
005 *   This file is part of the Granite Data Services Platform.
006 *
007 *   Granite Data Services is free software; you can redistribute it and/or
008 *   modify it under the terms of the GNU Lesser General Public
009 *   License as published by the Free Software Foundation; either
010 *   version 2.1 of the License, or (at your option) any later version.
011 *
012 *   Granite Data Services 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 Lesser
015 *   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 library; if not, write to the Free Software
019 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020 *   USA, or see <http://www.gnu.org/licenses/>.
021 */
022package org.granite.tide.ejb;
023
024import java.lang.reflect.Method;
025import java.util.HashMap;
026import java.util.HashSet;
027import java.util.Map;
028import java.util.Set;
029
030import javax.ejb.Local;
031import javax.ejb.Remote;
032import javax.ejb.Stateful;
033import javax.ejb.Stateless;
034
035import org.granite.logging.Logger;
036import org.granite.scan.ScannedItem;
037import org.granite.scan.ScannedItemHandler;
038import org.granite.tide.util.Observer;
039
040/**
041 * @author Franck WOLFF
042 */
043public class EjbScannedItemHandler implements ScannedItemHandler {
044
045        private static final Logger log = Logger.getLogger(EjbScannedItemHandler.class);
046        private static final EjbScannedItemHandler instance = new EjbScannedItemHandler();
047        
048        private final Map<Class<?>, Class<?>> scannedClasses = new HashMap<Class<?>, Class<?>>();
049        private final Map<String, Set<Method>> observers = new HashMap<String, Set<Method>>();
050        
051        public static EjbScannedItemHandler instance() {
052                return instance;
053        }
054        
055        static EjbScannedItemHandler instance(boolean reset) {
056                instance.scannedClasses.clear();
057                instance.observers.clear();
058                return instance;
059        }
060        
061        private EjbScannedItemHandler() {
062        }
063        
064        public boolean handleMarkerItem(ScannedItem item) {
065                return false;
066        }
067
068        public void handleScannedItem(ScannedItem item) {
069                if ("class".equals(item.getExtension()) && item.getName().indexOf('$') == -1) {
070                        try {
071                                Class<?> clazz = item.loadAsClass();
072                                if (clazz.isAnnotationPresent(Stateless.class) || clazz.isAnnotationPresent(Stateful.class)) {
073                                        scannedClasses.put(clazz, clazz);       // Interface-less EJB 3.1
074                                        if (clazz.isAnnotationPresent(Local.class)) {
075                                                for (Class<?> i : clazz.getAnnotation(Local.class).value())
076                                                        scannedClasses.put(i, clazz);
077                                        }
078                                        if (clazz.isAnnotationPresent(Remote.class)) {
079                                                for (Class<?> i : clazz.getAnnotation(Remote.class).value())
080                                                        scannedClasses.put(i, clazz);
081                                        }
082                            for (Class<?> i : clazz.getInterfaces()) {
083                                if (i.isAnnotationPresent(Local.class))
084                                        scannedClasses.put(i, clazz);
085                                if (i.isAnnotationPresent(Remote.class))
086                                        scannedClasses.put(i, clazz);
087                            }
088                                        
089                                        for (Method method : clazz.getMethods()) {
090                                                if (method.isAnnotationPresent(Observer.class)) {
091                                                        Observer o = method.getAnnotation(Observer.class);
092                                                        Set<Method> methods = observers.get(o.value());
093                                                        if (methods == null) {
094                                                                methods = new HashSet<Method>();
095                                                                observers.put(o.value(), methods);
096                                                        }
097                                                        methods.add(method);
098                                                }
099                                        }
100                                }
101                        }
102                        catch (Exception e) {
103                                log.debug(e, "Could not introspect scanned item: %s", item);
104                        }
105                }
106        }
107        
108        public Map<Class<?>, Class<?>> getScannedClasses() {
109                return scannedClasses;
110        }
111        
112        public Map<String, Set<Method>> getObservers() {
113                return observers;
114        }
115}