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

com.wl4g.component.common.resource.resolver.ClassPathResourcePatternResolver Maven / Gradle / Ivy

/*
 * Copyright 2017 ~ 2025 the original author or authors. 
 *
 * 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.wl4g.component.common.resource.resolver;

import static com.wl4g.component.common.lang.Assert2.notNull;
import static com.wl4g.component.common.log.SmartLoggerFactory.getLogger;
import static java.util.stream.Collectors.toCollection;

/**
 * Retention of upstream license agreement statement:
* Thank you very much spring framework, We fully comply with and support the open license * agreement of spring. The purpose of migration is to solve the problem * that these elegant API programs can still be easily used without running * in the spring environment. *
* Copyright 2002-2017 the original author or authors. * * 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. */ import java.io.File; import java.io.IOException; import java.lang.reflect.InvocationHandler; import java.lang.reflect.Method; import java.lang.reflect.Proxy; import java.net.JarURLConnection; import java.net.MalformedURLException; import java.net.URISyntaxException; import java.net.URL; import java.net.URLClassLoader; import java.net.URLConnection; import java.util.Arrays; import java.util.Collections; import java.util.Enumeration; import java.util.LinkedHashSet; import java.util.Set; import java.util.jar.JarEntry; import java.util.jar.JarFile; import java.util.zip.ZipException; import org.apache.commons.lang3.StringUtils; import com.wl4g.component.common.lang.Assert2; import com.wl4g.component.common.lang.ClassUtils2; import com.wl4g.component.common.lang.StringUtils2; import com.wl4g.component.common.log.SmartLogger; import com.wl4g.component.common.matching.AntPathMatcher; import com.wl4g.component.common.matching.PathMatcher; import com.wl4g.component.common.reflect.ReflectionUtils2; import com.wl4g.component.common.resource.FileStreamResource; import com.wl4g.component.common.resource.ResourceUtils2; import com.wl4g.component.common.resource.StreamResource; import com.wl4g.component.common.resource.UrlStreamResource; import com.wl4g.component.common.resource.VfsStreamResource; import com.wl4g.component.common.resource.VfsUtils2; /** * A {@link ResourcePatternResolver} implementation that is able to resolve a * specified resource location path into one or more matching Resources. The * source path may be a simple path which has a one-to-one mapping to a target * {@link org.StreamResource.core.io.Resource}, or alternatively may contain the * special "{@code classpath*:}" prefix and/or internal Ant-style regular * expressions (matched using Spring's * {@link org.springframework.util.AntPathMatcher} utility). Both of the latter * are effectively wildcards. * *

* No Wildcards: * *

* In the simple case, if the specified location path does not start with the * {@code "classpath*:}" prefix, and does not contain a PathMatcher pattern, * this resolver will simply return a single resource via a * {@code getResource()} call on the underlying {@code ResourceLoader}. Examples * are real URLs such as "{@code file:C:/context.xml}", pseudo-URLs such as * "{@code classpath:/context.xml}", and simple unprefixed paths such as * "{@code /WEB-INF/context.xml}". The latter will resolve in a fashion specific * to the underlying {@code ResourceLoader} (e.g. {@code ServletContextResource} * for a {@code WebApplicationContext}). * *

* Ant-style Patterns: * *

* When the path location contains an Ant-style pattern, e.g.: * *

 * /WEB-INF/*-context.xml
 * com/mycompany/**/applicationContext.xml
 * file:C:/some/path/*-context.xml
 * classpath:com/mycompany/**/applicationContext.xml
 * 
* * the resolver follows a more complex but defined procedure to try to resolve * the wildcard. It produces a {@code Resource} for the path up to the last * non-wildcard segment and obtains a {@code URL} from it. If this URL is not a * "{@code jar:}" URL or container-specific variant (e.g. "{@code zip:}" in * WebLogic, "{@code wsjar}" in WebSphere", etc.), then a {@code java.io.File} * is obtained from it, and used to resolve the wildcard by walking the * filesystem. In the case of a jar URL, the resolver either gets a * {@code java.net.JarURLConnection} from it, or manually parses the jar URL, * and then traverses the contents of the jar file, to resolve the wildcards. * *

* Implications on portability: * *

* If the specified path is already a file URL (either explicitly, or implicitly * because the base {@code ResourceLoader} is a filesystem one, then wildcarding * is guaranteed to work in a completely portable fashion. * *

* If the specified path is a classpath location, then the resolver must obtain * the last non-wildcard path segment URL via a * {@code Classloader.getResource()} call. Since this is just a node of the path * (not the file at the end) it is actually undefined (in the ClassLoader * Javadocs) exactly what sort of a URL is returned in this case. In practice, * it is usually a {@code java.io.File} representing the directory, where the * classpath resource resolves to a filesystem location, or a jar URL of some * sort, where the classpath resource resolves to a jar location. Still, there * is a portability concern on this operation. * *

* If a jar URL is obtained for the last non-wildcard segment, the resolver must * be able to get a {@code java.net.JarURLConnection} from it, or manually parse * the jar URL, to be able to walk the contents of the jar, and resolve the * wildcard. This will work in most environments, but will fail in others, and * it is strongly recommended that the wildcard resolution of resources coming * from jars be thoroughly tested in your specific environment before you rely * on it. * *

* {@code classpath*:} Prefix: * *

* There is special support for retrieving multiple class path resources with * the same name, via the "{@code classpath*:}" prefix. For example, * "{@code classpath*:META-INF/beans.xml}" will find all "beans.xml" files in * the class path, be it in "classes" directories or in JAR files. This is * particularly useful for autodetecting config files of the same name at the * same location within each jar file. Internally, this happens via a * {@code ClassLoader.getResources()} call, and is completely portable. * *

* The "classpath*:" prefix can also be combined with a PathMatcher pattern in * the rest of the location path, for example "classpath*:META-INF/*-beans.xml". * In this case, the resolution strategy is fairly simple: a * {@code ClassLoader.getResources()} call is used on the last non-wildcard path * segment to get all the matching resources in the class loader hierarchy, and * then off each resource the same PathMatcher resolution strategy described * above is used for the wildcard subpath. * *

* Other notes: * *

* WARNING: Note that "{@code classpath*:}" when combined with Ant-style * patterns will only work reliably with at least one root directory before the * pattern starts, unless the actual target files reside in the file system. * This means that a pattern like "{@code classpath*:*.xml}" will not * retrieve files from the root of jar files but rather only from the root of * expanded directories. This originates from a limitation in the JDK's * {@code ClassLoader.getResources()} method which only returns file system * locations for a passed-in empty String (indicating potential roots to * search). This {@code ResourcePatternResolver} implementation is trying to * mitigate the jar root lookup limitation through {@link URLClassLoader} * introspection and "java.class.path" manifest evaluation; however, without * portability guarantees. * *

* WARNING: Ant-style patterns with "classpath:" resources are not * guaranteed to find matching resources if the root package to search is * available in multiple class path locations. This is because a resource such * as * *

 * com / mycompany / package1 / service - context.xml
 * 
* * may be in only one location, but when a path such as * *
 *     classpath:com/mycompany/**/service-context.xml
 * 
* * is used to try to resolve it, the resolver will work off the (first) URL * returned by {@code getResource("com/mycompany");}. If this base package node * exists in multiple classloader locations, the actual end resource may not be * underneath. Therefore, preferably, use "{@code classpath*:}" with the same * Ant-style pattern in such a case, which will search all class path * locations that contain the root package. * * @author Juergen Hoeller * @author Colin Sampaleanu * @author Marius Bogoevici * @author Costin Leau * @author Phil Webb * @since 1.0.2 * @see #CLASSPATH_ALL_URL_PREFIX * @see {@link org.springframework.util.AntPathMatcher} * @see {@link org.springframework.core.io.ResourceLoader#getResource(String)} * @see {@link ClassLoader#getResources(String)} */ public class ClassPathResourcePatternResolver implements ResourcePatternResolver { protected final SmartLogger log = getLogger(getClass()); private final ResourceLoader resourceLoader; private PathMatcher pathMatcher = new AntPathMatcher(); /** * Create a new PathMatchingResourcePatternResolver with a * DefaultResourceLoader. *

* ClassLoader access will happen via the thread context class loader. * * @see org.springframework.core.io.DefaultResourceLoader */ public ClassPathResourcePatternResolver() { this(new DefaultResourceLoader()); } /** * Create a new PathMatchingResourcePatternResolver with a * DefaultResourceLoader. * * @param classLoader * the ClassLoader to load classpath resources with, or * {@code null} for using the thread context class loader at the * time of actual resource access * @see org.springframework.core.io.DefaultResourceLoader */ public ClassPathResourcePatternResolver(ClassLoader classLoader) { this(new DefaultResourceLoader(classLoader)); } /** * Create a new PathMatchingResourcePatternResolver. *

* ClassLoader access will happen via the thread context class loader. * * @param resourceLoader * the ResourceLoader to load root directories and actual * resources with */ public ClassPathResourcePatternResolver(ResourceLoader resourceLoader) { this.resourceLoader = notNull(resourceLoader, "ResourceLoader must not be null"); } /** * Return the ResourceLoader that this pattern resolver works with. */ public ResourceLoader getResourceLoader() { return this.resourceLoader; } @Override public ClassLoader getClassLoader() { return getResourceLoader().getClassLoader(); } /** * Set the PathMatcher implementation to use for this resource pattern * resolver. Default is AntPathMatcher. * * @see com.wl4g.component.common.matching.AntPathMatcher */ public void setPathMatcher(PathMatcher pathMatcher) { Assert2.notNull(pathMatcher, "PathMatcher must not be null"); this.pathMatcher = pathMatcher; } /** * Return the PathMatcher that this resource pattern resolver uses. */ public PathMatcher getPathMatcher() { return this.pathMatcher; } @Override public StreamResource getResource(String location) { return getResourceLoader().getResource(location); } @Override public Set getResources(String... locationPatterns) throws IOException { notNull(locationPatterns, "Path locationPatterns can't null"); return Arrays.asList(locationPatterns).stream().map(pattern -> { try { return doGetResources(pattern); } catch (IOException e) { throw new IllegalStateException(e); } }).flatMap(rss -> rss.stream()).collect(toCollection(LinkedHashSet::new)); } /** * Resolve the given location pattern into Resource objects. * * @param locationPattern * @return */ protected Set doGetResources(String locationPattern) throws IOException { Assert2.notNull(locationPattern, "Location pattern must not be null"); if (locationPattern.startsWith(CLASSPATH_ALL_URL_PREFIX)) { // a class path resource (multiple resources for same name possible) if (getPathMatcher().isPattern(locationPattern.substring(CLASSPATH_ALL_URL_PREFIX.length()))) { // a class path resource pattern return findPathMatchingResources(locationPattern); } else { // all class path resources with the given name return findAllClassPathResources(locationPattern.substring(CLASSPATH_ALL_URL_PREFIX.length())); } } else { // Generally only look for a pattern after a prefix here, // and on Tomcat only after the "*/" separator for its "war:" // protocol. int prefixEnd = (locationPattern.startsWith("war:") ? locationPattern.indexOf("*/") + 1 : locationPattern.indexOf(':') + 1); if (getPathMatcher().isPattern(locationPattern.substring(prefixEnd))) { // a file pattern return findPathMatchingResources(locationPattern); } else { // a single resource with the given name return new LinkedHashSet() { private static final long serialVersionUID = 1L; { add(getResourceLoader().getResource(locationPattern)); } }; } } } /** * Find all class location resources with the given location via the * ClassLoader. Delegates to {@link #doFindAllClassPathResources(String)}. * * @param location * the absolute path within the classpath * @return the result as Resource array * @throws IOException * in case of I/O errors * @see java.lang.ClassLoader#getResources * @see #convertClassLoaderURL */ protected Set findAllClassPathResources(String location) throws IOException { String path = location; if (path.startsWith("/")) { path = path.substring(1); } Set result = doFindAllClassPathResources(path); log.trace("Resolved classpath location: {} to resources: {}", location, result); return result; } /** * Find all class location resources with the given path via the * ClassLoader. Called by {@link #findAllClassPathResources(String)}. * * @param path * the absolute path within the classpath (never a leading slash) * @return a mutable Set of matching Resource instances * @since 4.1.1 */ protected Set doFindAllClassPathResources(String path) throws IOException { Set result = new LinkedHashSet(16); ClassLoader cl = getClassLoader(); Enumeration resourceUrls = (cl != null ? cl.getResources(path) : ClassLoader.getSystemResources(path)); while (resourceUrls.hasMoreElements()) { URL url = resourceUrls.nextElement(); result.add(convertClassLoaderURL(url)); } if ("".equals(path)) { // The above result is likely to be incomplete, i.e. only containing // file system references. // We need to have pointers to each of the jar files on the // classpath as well... addAllClassLoaderJarRoots(cl, result); } return result; } /** * Convert the given URL as returned from the ClassLoader into a * {@link StreamResource}. *

* The default implementation simply creates a {@link UrlStreamResource} * instance. * * @param url * a URL as returned from the ClassLoader * @return the corresponding Resource object * @see java.lang.ClassLoader#getResources * @see org.StreamResource.core.io.Resource */ protected StreamResource convertClassLoaderURL(URL url) { return new UrlStreamResource(url); } /** * Search all {@link URLClassLoader} URLs for jar file references and add * them to the given set of resources in the form of pointers to the root of * the jar file content. * * @param classLoader * the ClassLoader to search (including its ancestors) * @param result * the set of resources to add jar roots to * @since 4.1.1 */ protected void addAllClassLoaderJarRoots(ClassLoader classLoader, Set result) { if (classLoader instanceof URLClassLoader) { try { for (URL url : ((URLClassLoader) classLoader).getURLs()) { try { UrlStreamResource jarResource = new UrlStreamResource( ResourceUtils2.JAR_URL_PREFIX + url + ResourceUtils2.JAR_URL_SEPARATOR); if (jarResource.exists()) { result.add(jarResource); } } catch (MalformedURLException ex) { if (log.isDebugEnabled()) { log.debug("Cannot search for matching files underneath [" + url + "] because it cannot be converted to a valid 'jar:' URL: " + ex.getMessage()); } } } } catch (Exception ex) { if (log.isDebugEnabled()) { log.debug("Cannot introspect jar files since ClassLoader [" + classLoader + "] does not support 'getURLs()': " + ex); } } } if (classLoader == ClassLoader.getSystemClassLoader()) { // "java.class.path" manifest evaluation... addClassPathManifestEntries(result); } if (classLoader != null) { try { // Hierarchy traversal... addAllClassLoaderJarRoots(classLoader.getParent(), result); } catch (Exception ex) { if (log.isDebugEnabled()) { log.debug("Cannot introspect jar files in parent ClassLoader since [" + classLoader + "] does not support 'getParent()': " + ex); } } } } /** * Determine jar file references from the "java.class.path." manifest * property and add them to the given set of resources in the form of * pointers to the root of the jar file content. * * @param result * the set of resources to add jar roots to * @since 4.3 */ protected void addClassPathManifestEntries(Set result) { try { String javaClassPathProperty = System.getProperty("java.class.path"); for (String path : StringUtils2.delimitedListToStringArray(javaClassPathProperty, System.getProperty("path.separator"))) { try { String filePath = new File(path).getAbsolutePath(); int prefixIndex = filePath.indexOf(':'); if (prefixIndex == 1) { // Possibly "c:" drive prefix on Windows, to be // upper-cased for proper duplicate detection filePath = StringUtils.capitalize(filePath); } UrlStreamResource jarResource = new UrlStreamResource(ResourceUtils2.JAR_URL_PREFIX + ResourceUtils2.FILE_URL_PREFIX + filePath + ResourceUtils2.JAR_URL_SEPARATOR); // Potentially overlapping with URLClassLoader.getURLs() // result above! if (!result.contains(jarResource) && !hasDuplicate(filePath, result) && jarResource.exists()) { result.add(jarResource); } } catch (MalformedURLException ex) { if (log.isDebugEnabled()) { log.debug("Cannot search for matching files underneath [" + path + "] because it cannot be converted to a valid 'jar:' URL: " + ex.getMessage()); } } } } catch (Exception ex) { if (log.isDebugEnabled()) { log.debug("Failed to evaluate 'java.class.path' manifest entries: " + ex); } } } /** * Check whether the given file path has a duplicate but differently * structured entry in the existing result, i.e. with or without a leading * slash. * * @param filePath * the file path (with or without a leading slash) * @param result * the current result * @return {@code true} if there is a duplicate (i.e. to ignore the given * file path), {@code false} to proceed with adding a corresponding * resource to the current result */ private boolean hasDuplicate(String filePath, Set result) { if (result.isEmpty()) { return false; } String duplicatePath = (filePath.startsWith("/") ? filePath.substring(1) : "/" + filePath); try { return result.contains(new UrlStreamResource(ResourceUtils2.JAR_URL_PREFIX + ResourceUtils2.FILE_URL_PREFIX + duplicatePath + ResourceUtils2.JAR_URL_SEPARATOR)); } catch (MalformedURLException ex) { // Ignore: just for testing against duplicate. return false; } } /** * Find all resources that match the given location pattern via the * Ant-style PathMatcher. Supports resources in jar files and zip files and * in the file system. * * @param locationPattern * the location pattern to match * @return the result as Resource array * @throws IOException * in case of I/O errors * @see #doFindPathMatchingJarResources * @see #doFindPathMatchingFileResources * @see com.wl4g.component.common.matching.PathMatcher */ protected Set findPathMatchingResources(String locationPattern) throws IOException { String rootDirPath = determineRootDir(locationPattern); String subPattern = locationPattern.substring(rootDirPath.length()); Set rootDirResources = doGetResources(rootDirPath); Set result = new LinkedHashSet(16); for (StreamResource rootDirResource : rootDirResources) { rootDirResource = resolveRootDirResource(rootDirResource); URL rootDirUrl = rootDirResource.getURL(); if (equinoxResolveMethod != null) { if (rootDirUrl.getProtocol().startsWith("bundle")) { rootDirUrl = (URL) ReflectionUtils2.invokeMethod(equinoxResolveMethod, null, rootDirUrl); rootDirResource = new UrlStreamResource(rootDirUrl); } } if (rootDirUrl.getProtocol().startsWith(ResourceUtils2.URL_PROTOCOL_VFS)) { result.addAll(VfsResourceMatchingDelegate.findMatchingResources(rootDirUrl, subPattern, getPathMatcher())); } else if (ResourceUtils2.isJarURL(rootDirUrl) || isJarResource(rootDirResource)) { result.addAll(doFindPathMatchingJarResources(rootDirResource, rootDirUrl, subPattern)); } else { result.addAll(doFindPathMatchingFileResources(rootDirResource, subPattern)); } } if (log.isDebugEnabled()) { log.debug("Resolved location pattern [" + locationPattern + "] to resources " + result); } return result; } /** * Determine the root directory for the given location. *

* Used for determining the starting point for file matching, resolving the * root directory location to a {@code java.io.File} and passing it into * {@code retrieveMatchingFiles}, with the remainder of the location as * pattern. *

* Will return "/WEB-INF/" for the pattern "/WEB-INF/*.xml", for example. * * @param location * the location to check * @return the part of the location that denotes the root directory * @see #retrieveMatchingFiles */ protected String determineRootDir(String location) { int prefixEnd = location.indexOf(':') + 1; int rootDirEnd = location.length(); while (rootDirEnd > prefixEnd && getPathMatcher().isPattern(location.substring(prefixEnd, rootDirEnd))) { rootDirEnd = location.lastIndexOf('/', rootDirEnd - 2) + 1; } if (rootDirEnd == 0) { rootDirEnd = prefixEnd; } return location.substring(0, rootDirEnd); } /** * Resolve the specified resource for path matching. *

* By default, Equinox OSGi "bundleresource:" / "bundleentry:" URL will be * resolved into a standard jar file URL that be traversed using Spring's * standard jar file traversal algorithm. For any preceding custom * resolution, override this method and replace the resource handle * accordingly. * * @param original * the resource to resolve * @return the resolved resource (may be identical to the passed-in * resource) * @throws IOException * in case of resolution failure */ protected StreamResource resolveRootDirResource(StreamResource original) throws IOException { return original; } /** * Return whether the given resource handle indicates a jar resource that * the {@code doFindPathMatchingJarResources} method can handle. *

* By default, the URL protocols "jar", "zip", "vfszip and "wsjar" will be * treated as jar resources. This template method allows for detecting * further kinds of jar-like resources, e.g. through {@code instanceof} * checks on the resource handle type. * * @param resource * the resource handle to check (usually the root directory to * start path matching from) * @see #doFindPathMatchingJarResources * @see com.wl4g.component.common.resource.ResourceUtils2#isJarURL */ protected boolean isJarResource(StreamResource resource) throws IOException { return false; } /** * Find all resources in jar files that match the given location pattern via * the Ant-style PathMatcher. * * @param rootDirResource * the root directory as Resource * @param rootDirURL * the pre-resolved root directory URL * @param subPattern * the sub pattern to match (below the root directory) * @return a mutable Set of matching Resource instances * @throws IOException * in case of I/O errors * @since 4.3 * @see java.net.JarURLConnection * @see com.wl4g.component.common.matching.PathMatcher */ protected Set doFindPathMatchingJarResources(StreamResource rootDirResource, URL rootDirURL, String subPattern) throws IOException { // Check deprecated variant for potential overriding first... Set result = doFindPathMatchingJarResources(rootDirResource, subPattern); if (result != null) { return result; } URLConnection con = rootDirURL.openConnection(); JarFile jarFile; String jarFileUrl; String rootEntryPath; boolean closeJarFile; if (con instanceof JarURLConnection) { // Should usually be the case for traditional JAR files. JarURLConnection jarCon = (JarURLConnection) con; ResourceUtils2.useCachesIfNecessary(jarCon); jarFile = jarCon.getJarFile(); jarFileUrl = jarCon.getJarFileURL().toExternalForm(); JarEntry jarEntry = jarCon.getJarEntry(); rootEntryPath = (jarEntry != null ? jarEntry.getName() : ""); closeJarFile = !jarCon.getUseCaches(); } else { // No JarURLConnection -> need to resort to URL file parsing. // We'll assume URLs of the format "jar:path!/entry", with the // protocol // being arbitrary as long as following the entry format. // We'll also handle paths with and without leading "file:" prefix. String urlFile = rootDirURL.getFile(); try { int separatorIndex = urlFile.indexOf(ResourceUtils2.WAR_URL_SEPARATOR); if (separatorIndex == -1) { separatorIndex = urlFile.indexOf(ResourceUtils2.JAR_URL_SEPARATOR); } if (separatorIndex != -1) { jarFileUrl = urlFile.substring(0, separatorIndex); rootEntryPath = urlFile.substring(separatorIndex + 2); // both // separators // are // 2 // chars jarFile = getJarFile(jarFileUrl); } else { jarFile = new JarFile(urlFile); jarFileUrl = urlFile; rootEntryPath = ""; } closeJarFile = true; } catch (ZipException ex) { if (log.isDebugEnabled()) { log.debug("Skipping invalid jar classpath entry [" + urlFile + "]"); } return Collections.emptySet(); } } try { if (log.isDebugEnabled()) { log.debug("Looking for matching resources in jar file [" + jarFileUrl + "]"); } if (!"".equals(rootEntryPath) && !rootEntryPath.endsWith("/")) { // Root entry path must end with slash to allow for proper // matching. // The Sun JRE does not return a slash here, but BEA JRockit // does. rootEntryPath = rootEntryPath + "/"; } result = new LinkedHashSet(8); for (Enumeration entries = jarFile.entries(); entries.hasMoreElements();) { JarEntry entry = entries.nextElement(); String entryPath = entry.getName(); if (entryPath.startsWith(rootEntryPath)) { String relativePath = entryPath.substring(rootEntryPath.length()); if (getPathMatcher().match(subPattern, relativePath)) { result.add(rootDirResource.createRelative(relativePath)); } } } return result; } finally { if (closeJarFile) { jarFile.close(); } } } /** * Find all resources in jar files that match the given location pattern via * the Ant-style PathMatcher. * * @param rootDirResource * the root directory as Resource * @param subPattern * the sub pattern to match (below the root directory) * @return a mutable Set of matching Resource instances * @throws IOException * in case of I/O errors * @deprecated as of Spring 4.3, in favor of * {@link #doFindPathMatchingJarResources(StreamResource, URL, String)} */ @Deprecated protected Set doFindPathMatchingJarResources(StreamResource rootDirResource, String subPattern) throws IOException { return null; } /** * Resolve the given jar file URL into a JarFile object. */ protected JarFile getJarFile(String jarFileUrl) throws IOException { if (jarFileUrl.startsWith(ResourceUtils2.FILE_URL_PREFIX)) { try { return new JarFile(ResourceUtils2.toURI(jarFileUrl).getSchemeSpecificPart()); } catch (URISyntaxException ex) { // Fallback for URLs that are not valid URIs (should hardly ever // happen). return new JarFile(jarFileUrl.substring(ResourceUtils2.FILE_URL_PREFIX.length())); } } else { return new JarFile(jarFileUrl); } } /** * Find all resources in the file system that match the given location * pattern via the Ant-style PathMatcher. * * @param rootDirResource * the root directory as Resource * @param subPattern * the sub pattern to match (below the root directory) * @return a mutable Set of matching Resource instances * @throws IOException * in case of I/O errors * @see #retrieveMatchingFiles * @see com.wl4g.component.common.matching.PathMatcher */ protected Set doFindPathMatchingFileResources(StreamResource rootDirResource, String subPattern) throws IOException { File rootDir; try { rootDir = rootDirResource.getFile().getAbsoluteFile(); } catch (IOException ex) { if (log.isWarnEnabled()) { log.warn("Cannot search for matching files underneath " + rootDirResource + " because it does not correspond to a directory in the file system", ex); } return Collections.emptySet(); } return doFindMatchingFileSystemResources(rootDir, subPattern); } /** * Find all resources in the file system that match the given location * pattern via the Ant-style PathMatcher. * * @param rootDir * the root directory in the file system * @param subPattern * the sub pattern to match (below the root directory) * @return a mutable Set of matching Resource instances * @throws IOException * in case of I/O errors * @see #retrieveMatchingFiles * @see com.wl4g.component.common.matching.PathMatcher */ protected Set doFindMatchingFileSystemResources(File rootDir, String subPattern) throws IOException { if (log.isDebugEnabled()) { log.debug("Looking for matching resources in directory tree [" + rootDir.getPath() + "]"); } Set matchingFiles = retrieveMatchingFiles(rootDir, subPattern); Set result = new LinkedHashSet(matchingFiles.size()); for (File file : matchingFiles) { result.add(new FileStreamResource(file)); } return result; } /** * Retrieve files that match the given path pattern, checking the given * directory and its subdirectories. * * @param rootDir * the directory to start from * @param pattern * the pattern to match against, relative to the root directory * @return a mutable Set of matching Resource instances * @throws IOException * if directory contents could not be retrieved */ protected Set retrieveMatchingFiles(File rootDir, String pattern) throws IOException { if (!rootDir.exists()) { // Silently skip non-existing directories. if (log.isDebugEnabled()) { log.debug("Skipping [" + rootDir.getAbsolutePath() + "] because it does not exist"); } return Collections.emptySet(); } if (!rootDir.isDirectory()) { // Complain louder if it exists but is no directory. if (log.isWarnEnabled()) { log.warn("Skipping [" + rootDir.getAbsolutePath() + "] because it does not denote a directory"); } return Collections.emptySet(); } if (!rootDir.canRead()) { if (log.isWarnEnabled()) { log.warn("Cannot search for matching files underneath directory [" + rootDir.getAbsolutePath() + "] because the application is not allowed to read the directory"); } return Collections.emptySet(); } String fullPattern = StringUtils.replace(rootDir.getAbsolutePath(), File.separator, "/"); if (!pattern.startsWith("/")) { fullPattern += "/"; } fullPattern = fullPattern + StringUtils.replace(pattern, File.separator, "/"); Set result = new LinkedHashSet(8); doRetrieveMatchingFiles(fullPattern, rootDir, result); return result; } /** * Recursively retrieve files that match the given pattern, adding them to * the given result list. * * @param fullPattern * the pattern to match against, with prepended root directory * path * @param dir * the current directory * @param result * the Set of matching File instances to add to * @throws IOException * if directory contents could not be retrieved */ protected void doRetrieveMatchingFiles(String fullPattern, File dir, Set result) throws IOException { if (log.isDebugEnabled()) { log.debug("Searching directory [" + dir.getAbsolutePath() + "] for files matching pattern [" + fullPattern + "]"); } File[] dirContents = dir.listFiles(); if (dirContents == null) { if (log.isWarnEnabled()) { log.warn("Could not retrieve contents of directory [" + dir.getAbsolutePath() + "]"); } return; } Arrays.sort(dirContents); for (File content : dirContents) { String currPath = StringUtils.replace(content.getAbsolutePath(), File.separator, "/"); if (content.isDirectory() && getPathMatcher().matchStart(fullPattern, currPath + "/")) { if (!content.canRead()) { if (log.isDebugEnabled()) { log.debug("Skipping subdirectory [" + dir.getAbsolutePath() + "] because the application is not allowed to read the directory"); } } else { doRetrieveMatchingFiles(fullPattern, content, result); } } if (getPathMatcher().match(fullPattern, currPath)) { result.add(content); } } } /** * Inner delegate class, avoiding a hard JBoss VFS API dependency at * runtime. */ private static class VfsResourceMatchingDelegate { public static Set findMatchingResources(URL rootDirURL, String locationPattern, PathMatcher pathMatcher) throws IOException { Object root = VfsPatternUtils.findRoot(rootDirURL); PatternVirtualFileVisitor visitor = new PatternVirtualFileVisitor(VfsPatternUtils.getPath(root), locationPattern, pathMatcher); VfsPatternUtils.visit(root, visitor); return visitor.getResources(); } } /** * VFS visitor for path matching purposes. */ @SuppressWarnings("unused") private static class PatternVirtualFileVisitor implements InvocationHandler { private final String subPattern; private final PathMatcher pathMatcher; private final String rootPath; private final Set resources = new LinkedHashSet(); public PatternVirtualFileVisitor(String rootPath, String subPattern, PathMatcher pathMatcher) { this.subPattern = subPattern; this.pathMatcher = pathMatcher; this.rootPath = (rootPath.isEmpty() || rootPath.endsWith("/") ? rootPath : rootPath + "/"); } @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { String methodName = method.getName(); if (Object.class == method.getDeclaringClass()) { if (methodName.equals("equals")) { // Only consider equal when proxies are identical. return (proxy == args[0]); } else if (methodName.equals("hashCode")) { return System.identityHashCode(proxy); } } else if ("getAttributes".equals(methodName)) { return getAttributes(); } else if ("visit".equals(methodName)) { visit(args[0]); return null; } else if ("toString".equals(methodName)) { return toString(); } throw new IllegalStateException("Unexpected method invocation: " + method); } public void visit(Object vfsResource) { if (this.pathMatcher.match(this.subPattern, VfsPatternUtils.getPath(vfsResource).substring(this.rootPath.length()))) { this.resources.add(new VfsStreamResource(vfsResource)); } } public Object getAttributes() { return VfsPatternUtils.getVisitorAttribute(); } public Set getResources() { return this.resources; } public int size() { return this.resources.size(); } @Override public String toString() { return "sub-pattern: " + this.subPattern + ", resources: " + this.resources; } } /** * Artificial class used for accessing the {@link VfsUtils2} methods without * exposing them to the entire world. * * @author Costin Leau * @since 3.0.3 */ abstract static class VfsPatternUtils extends VfsUtils2 { static Object getVisitorAttribute() { return doGetVisitorAttribute(); } static String getPath(Object resource) { return doGetPath(resource); } static Object findRoot(URL url) throws IOException { return getRoot(url); } static void visit(Object resource, InvocationHandler visitor) throws IOException { Object visitorProxy = Proxy.newProxyInstance(VIRTUAL_FILE_VISITOR_INTERFACE.getClassLoader(), new Class[] { VIRTUAL_FILE_VISITOR_INTERFACE }, visitor); invokeVfsMethod(VIRTUAL_FILE_METHOD_VISIT, resource, visitorProxy); } } private static Method equinoxResolveMethod; static { try { // Detect Equinox OSGi (e.g. on WebSphere 6.1) Class fileLocatorClass = ClassUtils2.forName("org.eclipse.core.runtime.FileLocator", ClassPathResourcePatternResolver.class.getClassLoader()); equinoxResolveMethod = fileLocatorClass.getMethod("resolve", URL.class); getLogger(ClassPathResourcePatternResolver.class).debug("Found Equinox FileLocator for OSGi bundle URL resolution"); } catch (Throwable ex) { equinoxResolveMethod = null; } } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy