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

org.apache.camel.util.ResourceHelper Maven / Gradle / Ivy

There is a newer version: 4.6.0
Show newest version
/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.camel.util;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.URLConnection;
import java.net.URLDecoder;
import java.util.Map;

import org.apache.camel.CamelContext;
import org.apache.camel.RuntimeCamelException;
import org.apache.camel.spi.ClassResolver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Helper class for loading resources on the classpath or file system.
 */
public final class ResourceHelper {

    private static final Logger LOG = LoggerFactory.getLogger(ResourceHelper.class);

    private ResourceHelper() {
        // utility class
    }

    /**
     * Resolves the expression/predicate whether it refers to an external script on the file/classpath etc.
     * This requires to use the prefix resource: such as resource:classpath:com/foo/myscript.groovy,
     * resource:file:/var/myscript.groovy.
     * 

* If not then the returned value is returned as-is. */ public static String resolveOptionalExternalScript(CamelContext camelContext, String expression) { if (expression == null) { return null; } String external = expression; // must be one line only int newLines = StringHelper.countChar(expression, '\n'); if (newLines > 1) { // okay then just use as-is return expression; } // must start with resource: to denote an external resource if (external.startsWith("resource:")) { external = external.substring(9); if (hasScheme(external)) { InputStream is = null; try { is = resolveMandatoryResourceAsInputStream(camelContext, external); expression = camelContext.getTypeConverter().convertTo(String.class, is); } catch (IOException e) { throw new RuntimeCamelException("Cannot load resource " + external, e); } finally { IOHelper.close(is); } } } return expression; } /** * Determines whether the URI has a scheme (e.g. file:, classpath: or http:) * * @param uri the URI * @return true if the URI starts with a scheme */ public static boolean hasScheme(String uri) { if (uri == null) { return false; } return uri.startsWith("file:") || uri.startsWith("classpath:") || uri.startsWith("http:"); } /** * Gets the scheme from the URI (e.g. file:, classpath: or http:) * * @param uri the uri * @return the scheme, or null if no scheme */ public static String getScheme(String uri) { if (hasScheme(uri)) { return uri.substring(0, uri.indexOf(":") + 1); } else { return null; } } /** * Resolves the mandatory resource. *

* The resource uri can refer to the following systems to be loaded from *

    * file:nameOfFile - to refer to the file system * classpath:nameOfFile - to refer to the classpath (default) * http:uri - to load the resoufce using HTTP * ref:nameOfBean - to lookup the resource in the {@link org.apache.camel.spi.Registry} *
* If no prefix has been given, then the resource is loaded from the classpath *

* If possible recommended to use {@link #resolveMandatoryResourceAsUrl(org.apache.camel.spi.ClassResolver, String)} * * @param camelContext the Camel Context * @param uri URI of the resource * @return the resource as an {@link InputStream}. Remember to close this stream after usage. * @throws java.io.IOException is thrown if the resource file could not be found or loaded as {@link InputStream} */ public static InputStream resolveMandatoryResourceAsInputStream(CamelContext camelContext, String uri) throws IOException { if (uri.startsWith("ref:")) { String ref = uri.substring(4); String value = CamelContextHelper.mandatoryLookup(camelContext, ref, String.class); return new ByteArrayInputStream(value.getBytes()); } InputStream is = resolveResourceAsInputStream(camelContext.getClassResolver(), uri); if (is == null) { String resolvedName = resolveUriPath(uri); throw new FileNotFoundException("Cannot find resource: " + resolvedName + " in classpath for URI: " + uri); } else { return is; } } /** * Resolves the mandatory resource. *

* If possible recommended to use {@link #resolveMandatoryResourceAsUrl(org.apache.camel.spi.ClassResolver, String)} * * @param classResolver the class resolver to load the resource from the classpath * @param uri URI of the resource * @return the resource as an {@link InputStream}. Remember to close this stream after usage. * @throws java.io.IOException is thrown if the resource file could not be found or loaded as {@link InputStream} * @deprecated use {@link #resolveMandatoryResourceAsInputStream(CamelContext, String)} */ @Deprecated public static InputStream resolveMandatoryResourceAsInputStream(ClassResolver classResolver, String uri) throws IOException { InputStream is = resolveResourceAsInputStream(classResolver, uri); if (is == null) { String resolvedName = resolveUriPath(uri); throw new FileNotFoundException("Cannot find resource: " + resolvedName + " in classpath for URI: " + uri); } else { return is; } } /** * Resolves the resource. *

* If possible recommended to use {@link #resolveMandatoryResourceAsUrl(org.apache.camel.spi.ClassResolver, String)} * * @param classResolver the class resolver to load the resource from the classpath * @param uri URI of the resource * @return the resource as an {@link InputStream}. Remember to close this stream after usage. Or null if not found. * @throws java.io.IOException is thrown if error loading the resource */ public static InputStream resolveResourceAsInputStream(ClassResolver classResolver, String uri) throws IOException { if (uri.startsWith("file:")) { uri = ObjectHelper.after(uri, "file:"); uri = tryDecodeUri(uri); LOG.trace("Loading resource: {} from file system", uri); return new FileInputStream(uri); } else if (uri.startsWith("http:")) { URL url = new URL(uri); LOG.trace("Loading resource: {} from HTTP", uri); URLConnection con = url.openConnection(); con.setUseCaches(false); try { return con.getInputStream(); } catch (IOException e) { // close the http connection to avoid // leaking gaps in case of an exception if (con instanceof HttpURLConnection) { ((HttpURLConnection) con).disconnect(); } throw e; } } else if (uri.startsWith("classpath:")) { uri = ObjectHelper.after(uri, "classpath:"); uri = tryDecodeUri(uri); } // load from classpath by default String resolvedName = resolveUriPath(uri); LOG.trace("Loading resource: {} from classpath", resolvedName); return classResolver.loadResourceAsStream(resolvedName); } /** * Resolves the mandatory resource. * * @param classResolver the class resolver to load the resource from the classpath * @param uri uri of the resource * @return the resource as an {@link java.net.URL}. * @throws java.io.FileNotFoundException is thrown if the resource file could not be found * @throws java.net.MalformedURLException if the URI is malformed */ public static URL resolveMandatoryResourceAsUrl(ClassResolver classResolver, String uri) throws FileNotFoundException, MalformedURLException { URL url = resolveResourceAsUrl(classResolver, uri); if (url == null) { String resolvedName = resolveUriPath(uri); throw new FileNotFoundException("Cannot find resource: " + resolvedName + " in classpath for URI: " + uri); } else { return url; } } /** * Resolves the resource. * * @param classResolver the class resolver to load the resource from the classpath * @param uri uri of the resource * @return the resource as an {@link java.net.URL}. Or null if not found. * @throws java.net.MalformedURLException if the URI is malformed */ public static URL resolveResourceAsUrl(ClassResolver classResolver, String uri) throws MalformedURLException { if (uri.startsWith("file:")) { // check if file exists first String name = ObjectHelper.after(uri, "file:"); uri = tryDecodeUri(uri); LOG.trace("Loading resource: {} from file system", uri); File file = new File(name); if (!file.exists()) { return null; } return new URL(uri); } else if (uri.startsWith("http:")) { LOG.trace("Loading resource: {} from HTTP", uri); return new URL(uri); } else if (uri.startsWith("classpath:")) { uri = ObjectHelper.after(uri, "classpath:"); uri = tryDecodeUri(uri); } // load from classpath by default String resolvedName = resolveUriPath(uri); LOG.trace("Loading resource: {} from classpath", resolvedName); return classResolver.loadResourceAsURL(resolvedName); } /** * Is the given uri a http uri? * * @param uri the uri * @return true if the uri starts with http: or https: */ public static boolean isHttpUri(String uri) { if (uri == null) { return false; } return uri.startsWith("http:") || uri.startsWith("https:"); } /** * Appends the parameters to the given uri * * @param uri the uri * @param parameters the additional parameters (will clear the map) * @return a new uri with the additional parameters appended * @throws URISyntaxException is thrown if the uri is invalid */ public static String appendParameters(String uri, Map parameters) throws URISyntaxException { // add additional parameters to the resource uri if (!parameters.isEmpty()) { String query = URISupport.createQueryString(parameters); URI u = new URI(uri); u = URISupport.createURIWithQuery(u, query); parameters.clear(); return u.toString(); } else { return uri; } } /** * Helper operation used to remove relative path notation from * resources. Most critical for resources on the Classpath * as resource loaders will not resolve the relative paths correctly. * * @param name the name of the resource to load * @return the modified or unmodified string if there were no changes */ private static String resolveUriPath(String name) { // compact the path and use / as separator as that's used for loading resources on the classpath return FileUtil.compactPath(name, '/'); } /** * Tries decoding the uri. * * @param uri the uri * @return the decoded uri, or the original uri */ private static String tryDecodeUri(String uri) { try { // try to decode as the uri may contain %20 for spaces etc uri = URLDecoder.decode(uri, "UTF-8"); } catch (Exception e) { LOG.trace("Error URL decoding uri using UTF-8 encoding: {}. This exception is ignored.", uri); // ignore } return uri; } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy