ohair@286: /*
alanb@368:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
ohair@286:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286:  *
ohair@286:  * This code is free software; you can redistribute it and/or modify it
ohair@286:  * under the terms of the GNU General Public License version 2 only, as
ohair@286:  * published by the Free Software Foundation.  Oracle designates this
ohair@286:  * particular file as subject to the "Classpath" exception as provided
ohair@286:  * by Oracle in the LICENSE file that accompanied this code.
ohair@286:  *
ohair@286:  * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
ohair@286:  * version 2 for more details (a copy is included in the LICENSE file that
ohair@286:  * accompanied this code).
ohair@286:  *
ohair@286:  * You should have received a copy of the GNU General Public License version
ohair@286:  * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286:  *
ohair@286:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286:  * or visit www.oracle.com if you need additional information or have any
ohair@286:  * questions.
ohair@286:  */
ohair@286: 
ohair@286: package com.sun.xml.internal.ws.api.client;
ohair@286: 
ohair@286: import com.sun.istack.internal.NotNull;
ohair@286: import com.sun.istack.internal.Nullable;
ohair@286: import com.sun.xml.internal.ws.api.WSService;
ohair@286: import com.sun.xml.internal.ws.util.ServiceFinder;
ohair@286: 
ohair@286: import javax.xml.ws.Service;
ohair@286: import java.util.ArrayList;
ohair@286: import java.util.HashSet;
ohair@286: import java.util.List;
ohair@286: import java.util.Set;
ohair@286: 
ohair@286: /**
ohair@286:  * Creates {@link ServiceInterceptor}.
ohair@286:  *
ohair@286:  * <p>
ohair@286:  * Code that wishes to inject {@link ServiceInterceptor} into {@link WSService}
ohair@286:  * must implement this class. There are two ways to have the JAX-WS RI
ohair@286:  * recognize your {@link ServiceInterceptor}s.
ohair@286:  *
ohair@286:  * <h3>Use {@link ServiceFinder}</h3>
ohair@286:  * <p>
ohair@286:  * {@link ServiceInterceptorFactory}s discovered via {@link ServiceFinder}
ohair@286:  * will be incorporated to all {@link WSService} instances.
ohair@286:  *
ohair@286:  * <h3>Register per-thread</h3>
ohair@286:  *
ohair@286:  *
ohair@286:  * @author Kohsuke Kawaguchi
ohair@286:  * @see ServiceInterceptor
ohair@286:  * @see 2.1 EA3
ohair@286:  */
ohair@286: public abstract class ServiceInterceptorFactory {
ohair@286:     public abstract ServiceInterceptor create(@NotNull WSService service);
ohair@286: 
ohair@286:     /**
ohair@286:      * Loads all {@link ServiceInterceptor}s and return aggregated one.
ohair@286:      */
ohair@286:     public static @NotNull ServiceInterceptor load(@NotNull WSService service, @Nullable ClassLoader cl) {
ohair@286:         List<ServiceInterceptor> l = new ArrayList<ServiceInterceptor>();
ohair@286: 
ohair@286:         // first service look-up
ohair@286:         for( ServiceInterceptorFactory f : ServiceFinder.find(ServiceInterceptorFactory.class))
ohair@286:             l.add(f.create(service));
ohair@286: 
ohair@286:         // then thread-local
ohair@286:         for( ServiceInterceptorFactory f : threadLocalFactories.get())
ohair@286:             l.add(f.create(service));
ohair@286: 
ohair@286:         return ServiceInterceptor.aggregate(l.toArray(new ServiceInterceptor[l.size()]));
ohair@286:     }
ohair@286: 
ohair@286:     private static ThreadLocal<Set<ServiceInterceptorFactory>> threadLocalFactories = new ThreadLocal<Set<ServiceInterceptorFactory>>() {
ohair@286:         protected Set<ServiceInterceptorFactory> initialValue() {
ohair@286:             return new HashSet<ServiceInterceptorFactory>();
ohair@286:         }
ohair@286:     };
ohair@286: 
ohair@286:     /**
ohair@286:      * Registers {@link ServiceInterceptorFactory} for this thread.
ohair@286:      *
ohair@286:      * <p>
ohair@286:      * Once registered, {@link ServiceInterceptorFactory}s are consulted for every
ohair@286:      * {@link Service} created in this thread, until it gets unregistered.
ohair@286:      */
ohair@286:     public static boolean registerForThread(ServiceInterceptorFactory factory) {
ohair@286:         return threadLocalFactories.get().add(factory);
ohair@286:     }
ohair@286: 
ohair@286:     /**
ohair@286:      * Removes previously registered {@link ServiceInterceptorFactory} for this thread.
ohair@286:      */
ohair@286:     public static boolean unregisterForThread(ServiceInterceptorFactory factory) {
ohair@286:         return threadLocalFactories.get().remove(factory);
ohair@286:     }
ohair@286: }