All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.phloc.commons.lang.ServiceLoaderBackport Maven / Gradle / Ivy

There is a newer version: 5.0.0
Show newest version
/**
 * Copyright (C) 2006-2014 phloc systems
 * http://www.phloc.com
 * office[at]phloc[dot]com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.phloc.commons.lang;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.phloc.commons.annotations.DevelopersNote;
import com.phloc.commons.annotations.UnsupportedOperation;
import com.phloc.commons.charset.CCharset;
import com.phloc.commons.io.streams.NonBlockingBufferedReader;
import com.phloc.commons.io.streams.StreamUtils;
import com.phloc.commons.string.ToStringGenerator;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;

/**
 * A simple service-provider loading facility.
 * 

* A service is a well-known set of interfaces and (usually abstract) * classes. A service provider is a specific implementation of a service. * The classes in a provider typically implement the interfaces and subclass the * classes defined in the service itself. Service providers can be installed in * an implementation of the Java platform in the form of extensions, that is, * jar files placed into any of the usual extension directories. Providers can * also be made available by adding them to the application's class path or by * some other platform-specific means. *

* For the purpose of loading, a service is represented by a single type, that * is, a single interface or abstract class. (A concrete class can be used, but * this is not recommended.) A provider of a given service contains one or more * concrete classes that extend this service type with data and code * specific to the provider. The provider class is typically not the * entire provider itself but rather a proxy which contains enough information * to decide whether the provider is able to satisfy a particular request * together with code that can create the actual provider on demand. The details * of provider classes tend to be highly service-specific; no single class or * interface could possibly unify them, so no such type is defined here. The * only requirement enforced by this facility is that provider classes must have * a zero-argument constructor so that they can be instantiated during loading. *

* A service provider is identified by placing a * provider-configuration file in the resource directory * META-INF/services. The file's name is the fully-qualified binary name of the service's type. * The file contains a list of fully-qualified binary names of concrete provider * classes, one per line. Space and tab characters surrounding each name, as * well as blank lines, are ignored. The comment character is '#' ( * '\u0023' , NUMBER SIGN); on each line all * characters following the first comment character are ignored. The file must * be encoded in UTF-8. *

* If a particular concrete provider class is named in more than one * configuration file, or is named in the same configuration file more than * once, then the duplicates are ignored. The configuration file naming a * particular provider need not be in the same jar file or other distribution * unit as the provider itself. The provider must be accessible from the same * class loader that was initially queried to locate the configuration file; * note that this is not necessarily the class loader from which the file was * actually loaded. *

* Providers are located and instantiated lazily, that is, on demand. A service * loader maintains a cache of the providers that have been loaded so far. Each * invocation of the {@link #iterator iterator} method returns an iterator that * first yields all of the elements of the cache, in instantiation order, and * then lazily locates and instantiates any remaining providers, adding each one * to the cache in turn. The cache can be cleared via the {@link #reload reload} * method. *

* Service loaders always execute in the security context of the caller. Trusted * system code should typically invoke the methods in this class, and the * methods of the iterators which they return, from within a privileged security * context. *

* Instances of this class are not safe for use by multiple concurrent threads. *

* Unless otherwise specified, passing a null argument to any method in * this class will cause a {@link NullPointerException} to be thrown. *

* Example Suppose we * have a service type com.example.CodecSet which is intended to * represent sets of encoder/decoder pairs for some protocol. In this case it is * an abstract class with two abstract methods:

* *
 * public abstract Encoder getEncoder (String encodingName);
 * 
 * public abstract Decoder getDecoder (String encodingName);
 * 
* *
Each method returns an appropriate object or null if * the provider does not support the given encoding. Typical providers support * more than one encoding. *

* If com.example.impl.StandardCodecs is an implementation of the * CodecSet service then its jar file also contains a file named *

* *
 * META - INF / services / com.example.CodecSet
 * 
* *
*

* This file contains the single line:

* *
 * com.example.impl.StandardCodecs    # Standard codecs
 * 
* *
*

* The CodecSet class creates and saves a single service instance at * initialization:

* *
 * private static ServiceLoader <CodecSet> codecSetLoader = ServiceLoader.load (CodecSet.class);
 * 
* *
*

* To locate an encoder for a given encoding name it defines a static factory * method which iterates through the known and available providers, returning * only when it has located a suitable encoder or has run out of providers. *

* *
 * public static Encoder getEncoder (String encodingName)
 * {
 *   for (CodecSet cp : codecSetLoader)
 *   {
 *     Encoder enc = cp.getEncoder (encodingName);
 *     if (enc != null)
 *       return enc;
 *   }
 *   return null;
 * }
 * 
* *
*

* A getDecoder method is defined similarly. *

* Usage Note If the * class path of a class loader that is used for provider loading includes * remote network URLs then those URLs will be dereferenced in the process of * searching for provider-configuration files. *

* This activity is normal, although it may cause puzzling entries to be created * in web-server logs. If a web server is not configured correctly, however, * then this activity may cause the provider-loading algorithm to fail * spuriously. *

* A web server should return an HTTP 404 (Not Found) response when a requested * resource does not exist. Sometimes, however, web servers are erroneously * configured to return an HTTP 200 (OK) response along with a helpful HTML * error page in such cases. This will cause a {@link ServiceLoaderException} to * be thrown when this class attempts to parse the HTML page as a * provider-configuration file. The best solution to this problem is to fix the * misconfigured web server to return the correct response code (HTTP 404) along * with the HTML error page. * * @param * The type of the service to be loaded by this loader * @author Mark Reinhold * @version 1.10, 06/04/10 * @since 1.6 */ // IFJDK5 // ELSE @Deprecated // ENDIF @DevelopersNote ("Required only in JDK5. Use the regular ServiceLoader from JDK 1.6 otherwise.") public final class ServiceLoaderBackport implements Iterable { private static final Logger s_aLogger = LoggerFactory.getLogger (ServiceLoaderBackport.class); static final String PREFIX = "META-INF/services/"; // The class or interface representing the service being loaded private final Class m_aService; // The class loader used to locate, load, and instantiate providers private final ClassLoader m_aClassLoader; // Cached providers, in instantiation order private final Map m_aProviders = new LinkedHashMap (); // The current lazy-lookup iterator private LazyIterator m_aLookupIterator; /** * Clear this loader's provider cache so that all providers will be reloaded. *

* After invoking this method, subsequent invocations of the * {@link #iterator() iterator} method will lazily look up and instantiate * providers from scratch, just as is done by a newly-created loader. *

* This method is intended for use in situations in which new providers can be * installed into a running Java virtual machine. */ public void reload () { m_aProviders.clear (); m_aLookupIterator = new LazyIterator (m_aService, m_aClassLoader); } private ServiceLoaderBackport (final Class svc, final ClassLoader cl) { if (s_aLogger.isDebugEnabled ()) s_aLogger.debug ("Loading all SPI implementations of " + svc.toString ()); m_aService = svc; m_aClassLoader = cl; reload (); } private static void _fail (final Class service, final String msg, final Throwable cause) { throw new ServiceLoaderException (service.getName () + ": " + msg, cause); } private static void _fail (final Class service, final String msg) { throw new ServiceLoaderException (service.getName () + ": " + msg); } private static void _fail (final Class service, final URL u, final int line, final String msg) { throw new ServiceLoaderException (service.getName () + ": " + u + ":" + line + ": " + msg); } // Parse a single line from the given configuration file, adding the name // on the line to the names list. // private int _parseLine (final Class aService, final URL u, final NonBlockingBufferedReader r, final int lc, final List names) throws IOException { String ln = r.readLine (); if (ln == null) return -1; final int ci = ln.indexOf ('#'); if (ci >= 0) ln = ln.substring (0, ci); ln = ln.trim (); final int n = ln.length (); if (n != 0) { if ((ln.indexOf (' ') >= 0) || (ln.indexOf ('\t') >= 0)) _fail (aService, u, lc, "Illegal configuration-file syntax"); int cp = ln.codePointAt (0); if (!Character.isJavaIdentifierStart (cp)) _fail (aService, u, lc, "Illegal provider-class name: " + ln); for (int i = Character.charCount (cp); i < n; i += Character.charCount (cp)) { cp = ln.codePointAt (i); if (!Character.isJavaIdentifierPart (cp) && (cp != '.')) _fail (aService, u, lc, "Illegal provider-class name: " + ln); } if (!m_aProviders.containsKey (ln) && !names.contains (ln)) names.add (ln); } return lc + 1; } // Parse the content of the given URL as a provider-configuration file. // // @param service // The service type for which providers are being sought; // used to construct error detail strings // // @param u // The URL naming the configuration file to be parsed // // @return A (possibly empty) iterator that will yield the provider-class // names in the given configuration file that are not yet members // of the returned set // // @ // If an I/O error occurs while reading from the given URL, or // if a configuration-file format error is detected // private Iterator _parse (final Class aService, final URL u) { InputStream in = null; NonBlockingBufferedReader r = null; final ArrayList names = new ArrayList (); try { in = u.openStream (); r = new NonBlockingBufferedReader (StreamUtils.createReader (in, CCharset.CHARSET_SERVICE_LOADER_OBJ)); int lc = 1; while ((lc = _parseLine (aService, u, r, lc, names)) >= 0) {// NOPMD // do nothing } } catch (final IOException x) { _fail (aService, "Error reading configuration file", x); } finally { try { if (r != null) r.close (); if (in != null) in.close (); } catch (final IOException y) { _fail (aService, "Error closing configuration file", y); } } return names.iterator (); } // Private inner class implementing fully-lazy provider lookup @SuppressFBWarnings ("UWF_NULL_FIELD") private final class LazyIterator implements Iterator { private final Class m_aLIService; private final ClassLoader m_aLILoader; private Enumeration m_aLIConfigs = null; private Iterator m_aLIPending = null; private String m_sLINextName = null; public LazyIterator (final Class aService, final ClassLoader aLoader) { m_aLIService = aService; m_aLILoader = aLoader; } public boolean hasNext () { if (m_sLINextName != null) return true; if (m_aLIConfigs == null) { try { final String fullName = PREFIX + m_aLIService.getName (); if (m_aLILoader == null) m_aLIConfigs = ClassLoader.getSystemResources (fullName); else m_aLIConfigs = m_aLILoader.getResources (fullName); } catch (final IOException ex) { _fail (m_aLIService, "Error locating configuration files", ex); } } while ((m_aLIPending == null) || !m_aLIPending.hasNext ()) { if (!m_aLIConfigs.hasMoreElements ()) { return false; } m_aLIPending = _parse (m_aLIService, m_aLIConfigs.nextElement ()); } m_sLINextName = m_aLIPending.next (); return true; } public SPITYPE next () { if (!hasNext ()) throw new NoSuchElementException (); final String sClassName = m_sLINextName; m_sLINextName = null; try { final SPITYPE p = m_aLIService.cast (Class.forName (sClassName, true, m_aLILoader).newInstance ()); m_aProviders.put (sClassName, p); return p; } catch (final ClassNotFoundException x) { _fail (m_aLIService, "Provider " + sClassName + " not found"); } catch (final Throwable x) { _fail (m_aLIService, "Provider " + sClassName + " could not be instantiated: " + x, x); } throw new Error (); // This cannot happen } @UnsupportedOperation public void remove () { throw new UnsupportedOperationException (); } } /** * Lazily loads the available providers of this loader's service. *

* The iterator returned by this method first yields all of the elements of * the provider cache, in instantiation order. It then lazily loads and * instantiates any remaining providers, adding each one to the cache in turn. *

* To achieve laziness the actual work of parsing the available * provider-configuration files and instantiating providers must be done by * the iterator itself. Its {@link java.util.Iterator#hasNext hasNext} and * {@link java.util.Iterator#next next} methods can therefore throw a * {@link ServiceLoaderException} if a provider-configuration file violates * the specified format, or if it names a provider class that cannot be found * and instantiated, or if the result of instantiating the class is not * assignable to the service type, or if any other kind of exception or error * is thrown as the next provider is located and instantiated. To write robust * code it is only necessary to catch {@link ServiceLoaderException} when * using a service iterator. *

* If such an error is thrown then subsequent invocations of the iterator will * make a best effort to locate and instantiate the next available provider, * but in general such recovery cannot be guaranteed.

Design Note Throwing * an error in these cases may seem extreme. The rationale for this behavior * is that a malformed provider-configuration file, like a malformed class * file, indicates a serious problem with the way the Java virtual machine is * configured or is being used. As such it is preferable to throw an error * rather than try to recover or, even worse, fail silently.
*

* The iterator returned by this method does not support removal. Invoking its * {@link java.util.Iterator#remove() remove} method will cause an * {@link UnsupportedOperationException} to be thrown. * * @return An iterator that lazily loads providers for this loader's service */ public Iterator iterator () { return new Iterator () { private final Iterator > m_aKnownProviders = m_aProviders.entrySet ().iterator (); public boolean hasNext () { return m_aKnownProviders.hasNext () || m_aLookupIterator.hasNext (); } public SPITYPE next () { if (m_aKnownProviders.hasNext ()) return m_aKnownProviders.next ().getValue (); return m_aLookupIterator.next (); } @UnsupportedOperation public void remove () { throw new UnsupportedOperationException (); } }; } /** * Returns a string describing this service. * * @return A descriptive string */ @Override public String toString () { return new ToStringGenerator (this).append ("serviceName", m_aService.getName ()).toString (); } /** * Creates a new service loader for the given service type and class loader. * * @param service * The interface or abstract class representing the service * @param loader * The class loader to be used to load provider-configuration files and * provider classes, or null if the system class loader (or, * failing that, the bootstrap class loader) is to be used * @return A new service loader */ public static ServiceLoaderBackport load (final Class service, final ClassLoader loader) { return new ServiceLoaderBackport (service, loader); } /** * Creates a new service loader for the given service type, using the current * thread's {@linkplain java.lang.Thread#getContextClassLoader context class * loader}. *

* An invocation of this convenience method of the form

* *
   * ServiceLoader.load(<i>service</i>)
   * 
* *
is equivalent to
* *
   * ServiceLoader.load(<i>service</i>,
   *                    Thread.currentThread().getContextClassLoader())
   * 
* *
* * @param service * The interface or abstract class representing the service * @return A new service loader */ public static ServiceLoaderBackport load (final Class service) { return load (service, ClassHelper.getDefaultClassLoader ()); } /** * Creates a new service loader for the given service type, using the * extension class loader. *

* This convenience method simply locates the extension class loader, call it * extClassLoader, and then returns

* *
   * ServiceLoader.load(<i>service</i>, <i>extClassLoader</i>)
   * 
* *
*

* If the extension class loader cannot be found then the system class loader * is used; if there is no system class loader then the bootstrap class loader * is used. *

* This method is intended for use when only installed providers are desired. * The resulting service will only find and load providers that have been * installed into the current Java virtual machine; providers on the * application's class path will be ignored. * * @param service * The interface or abstract class representing the service * @return A new service loader */ public static ServiceLoaderBackport loadInstalled (final Class service) { ClassLoader cl = ClassLoader.getSystemClassLoader (); ClassLoader prev = null; while (cl != null) { prev = cl; cl = cl.getParent (); } return ServiceLoaderBackport.load (service, prev); } }