Update packaged JARs to use standard JarLauncher
Change CLI generated JARs to use the standard `JarLauncher` instead of a custom `JarRunner`. The `PackagedSpringApplicationLauncher` is used as the `Start-Class` which in turn calls `SpringApplication.run()`.
This commit is contained in:
Phillip Webb
@@ -28,13 +28,10 @@ import java.net.URISyntaxException;
|
||||
import java.net.URL;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Enumeration;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
import java.util.jar.JarEntry;
|
||||
import java.util.jar.JarFile;
|
||||
import java.util.jar.Manifest;
|
||||
|
||||
import joptsimple.OptionSet;
|
||||
@@ -58,8 +55,10 @@ import org.springframework.boot.cli.compiler.GroovyCompilerConfiguration;
|
||||
import org.springframework.boot.cli.compiler.GroovyCompilerConfigurationAdapter;
|
||||
import org.springframework.boot.cli.compiler.RepositoryConfigurationFactory;
|
||||
import org.springframework.boot.cli.compiler.grape.RepositoryConfiguration;
|
||||
import org.springframework.boot.loader.ArchiveResolver;
|
||||
import org.springframework.boot.cli.jar.PackagedSpringApplicationLauncher;
|
||||
import org.springframework.boot.loader.tools.JarWriter;
|
||||
import org.springframework.boot.loader.tools.Layout;
|
||||
import org.springframework.boot.loader.tools.Layouts;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
/**
|
||||
@@ -69,6 +68,8 @@ import org.springframework.util.StringUtils;
|
||||
*/
|
||||
public class JarCommand extends OptionParsingCommand {
|
||||
|
||||
private static final Layout LAYOUT = new Layouts.Jar();
|
||||
|
||||
public JarCommand() {
|
||||
super(
|
||||
"jar",
|
||||
@@ -146,8 +147,11 @@ public class JarCommand extends OptionParsingCommand {
|
||||
addDependencies(jarWriter, dependencyUrls);
|
||||
addClasspathEntries(jarWriter, classpathEntries);
|
||||
addApplicationClasses(jarWriter, compiledClasses);
|
||||
String runnerClassName = getClassFile(PackagedSpringApplicationLauncher.class
|
||||
.getName());
|
||||
jarWriter.writeEntry(runnerClassName,
|
||||
getClass().getResourceAsStream("/" + runnerClassName));
|
||||
jarWriter.writeLoaderClasses();
|
||||
addJarRunner(jarWriter);
|
||||
}
|
||||
finally {
|
||||
jarWriter.close();
|
||||
@@ -206,9 +210,10 @@ public class JarCommand extends OptionParsingCommand {
|
||||
"Application-Classes",
|
||||
StringUtils.collectionToCommaDelimitedString(compiledClasses
|
||||
.keySet()));
|
||||
|
||||
manifest.getMainAttributes()
|
||||
.putValue("Main-Class", JarRunner.class.getName());
|
||||
manifest.getMainAttributes().putValue("Main-Class",
|
||||
LAYOUT.getLauncherClassName());
|
||||
manifest.getMainAttributes().putValue("Start-Class",
|
||||
PackagedSpringApplicationLauncher.class.getName());
|
||||
return manifest;
|
||||
}
|
||||
|
||||
@@ -243,27 +248,14 @@ public class JarCommand extends OptionParsingCommand {
|
||||
final Map<String, byte[]> compiledClasses) throws IOException {
|
||||
|
||||
for (Entry<String, byte[]> entry : compiledClasses.entrySet()) {
|
||||
String className = entry.getKey().replace(".", "/") + ".class";
|
||||
jarWriter.writeEntry(className,
|
||||
jarWriter.writeEntry(getClassFile(entry.getKey()),
|
||||
new ByteArrayInputStream(entry.getValue()));
|
||||
}
|
||||
}
|
||||
|
||||
private void addJarRunner(JarWriter jar) throws IOException, URISyntaxException {
|
||||
JarFile jarFile = getJarFile();
|
||||
String namePrefix = JarRunner.class.getName().replace(".", "/");
|
||||
Enumeration<JarEntry> entries = jarFile.entries();
|
||||
while (entries.hasMoreElements()) {
|
||||
JarEntry entry = entries.nextElement();
|
||||
if (entry.getName().startsWith(namePrefix)) {
|
||||
jar.writeEntry(entry.getName(), jarFile.getInputStream(entry));
|
||||
}
|
||||
}
|
||||
private String getClassFile(String className) {
|
||||
return className.replace(".", "/") + ".class";
|
||||
}
|
||||
|
||||
private JarFile getJarFile() throws URISyntaxException, IOException {
|
||||
return new JarFile(
|
||||
new ArchiveResolver().resolveArchiveLocation(JarCommand.class));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,98 +0,0 @@
|
||||
/*
|
||||
* Copyright 2012-2014 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 org.springframework.boot.cli.command.jar;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.InvocationTargetException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.net.MalformedURLException;
|
||||
import java.net.URISyntaxException;
|
||||
import java.net.URL;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.boot.loader.ArchiveResolver;
|
||||
import org.springframework.boot.loader.AsciiBytes;
|
||||
import org.springframework.boot.loader.LaunchedURLClassLoader;
|
||||
import org.springframework.boot.loader.archive.Archive;
|
||||
import org.springframework.boot.loader.archive.Archive.Entry;
|
||||
import org.springframework.boot.loader.archive.Archive.EntryFilter;
|
||||
|
||||
/**
|
||||
* A runner for a CLI application that has been compiled and packaged as a jar file
|
||||
*
|
||||
* @author Andy Wilkinson
|
||||
*/
|
||||
public class JarRunner {
|
||||
|
||||
private static final AsciiBytes LIB = new AsciiBytes("lib/");
|
||||
|
||||
public static void main(String[] args) throws URISyntaxException, IOException,
|
||||
ClassNotFoundException, SecurityException, NoSuchMethodException,
|
||||
IllegalArgumentException, IllegalAccessException, InvocationTargetException {
|
||||
|
||||
Archive archive = new ArchiveResolver().resolveArchive(JarRunner.class);
|
||||
|
||||
ClassLoader classLoader = createClassLoader(archive);
|
||||
Class<?>[] classes = loadApplicationClasses(archive, classLoader);
|
||||
|
||||
Thread.currentThread().setContextClassLoader(classLoader);
|
||||
|
||||
// Use reflection to load and call Spring
|
||||
Class<?> application = classLoader
|
||||
.loadClass("org.springframework.boot.SpringApplication");
|
||||
Method method = application.getMethod("run", Object[].class, String[].class);
|
||||
method.invoke(null, classes, args);
|
||||
|
||||
}
|
||||
|
||||
private static ClassLoader createClassLoader(Archive archive) throws IOException,
|
||||
MalformedURLException {
|
||||
List<Archive> nestedArchives = archive.getNestedArchives(new EntryFilter() {
|
||||
|
||||
@Override
|
||||
public boolean matches(Entry entry) {
|
||||
return entry.getName().startsWith(LIB);
|
||||
}
|
||||
|
||||
});
|
||||
|
||||
List<URL> urls = new ArrayList<URL>();
|
||||
urls.add(archive.getUrl());
|
||||
for (Archive nestedArchive : nestedArchives) {
|
||||
urls.add(nestedArchive.getUrl());
|
||||
}
|
||||
|
||||
ClassLoader classLoader = new LaunchedURLClassLoader(urls.toArray(new URL[urls
|
||||
.size()]), JarRunner.class.getClassLoader());
|
||||
return classLoader;
|
||||
}
|
||||
|
||||
private static Class<?>[] loadApplicationClasses(Archive archive,
|
||||
ClassLoader classLoader) throws ClassNotFoundException, IOException {
|
||||
String[] classNames = archive.getManifest().getMainAttributes()
|
||||
.getValue("Application-Classes").split(",");
|
||||
|
||||
Class<?>[] classes = new Class<?>[classNames.length];
|
||||
|
||||
for (int i = 0; i < classNames.length; i++) {
|
||||
Class<?> applicationClass = classLoader.loadClass(classNames[i]);
|
||||
classes[i] = applicationClass;
|
||||
}
|
||||
return classes;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,62 @@
|
||||
/*
|
||||
* Copyright 2012-2014 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 org.springframework.boot.cli.jar;
|
||||
|
||||
import java.lang.reflect.Method;
|
||||
import java.net.URL;
|
||||
import java.net.URLClassLoader;
|
||||
import java.util.jar.Manifest;
|
||||
|
||||
/**
|
||||
* A launcher for a CLI application that has been compiled and packaged as a jar file.
|
||||
*
|
||||
* @author Andy Wilkinson
|
||||
* @author Phillip Webb
|
||||
*/
|
||||
public class PackagedSpringApplicationLauncher {
|
||||
|
||||
private static final String SPRING_APPLICATION_CLASS = "org.springframework.boot.SpringApplication";
|
||||
|
||||
private void run(String[] args) throws Exception {
|
||||
URLClassLoader classLoader = (URLClassLoader) Thread.currentThread()
|
||||
.getContextClassLoader();
|
||||
Class<?> application = classLoader.loadClass(SPRING_APPLICATION_CLASS);
|
||||
Method method = application.getMethod("run", Object[].class, String[].class);
|
||||
method.invoke(null, getSources(classLoader), args);
|
||||
}
|
||||
|
||||
private Object[] getSources(URLClassLoader classLoader) throws Exception {
|
||||
URL url = classLoader.findResource("META-INF/MANIFEST.MF");
|
||||
Manifest manifest = new Manifest(url.openStream());
|
||||
String attribute = manifest.getMainAttributes().getValue("Application-Classes");
|
||||
return loadClasses(classLoader, attribute.split(","));
|
||||
}
|
||||
|
||||
private Class<?>[] loadClasses(ClassLoader classLoader, String[] names)
|
||||
throws ClassNotFoundException {
|
||||
Class<?>[] classes = new Class<?>[names.length];
|
||||
for (int i = 0; i < names.length; i++) {
|
||||
classes[i] = classLoader.loadClass(names[i]);
|
||||
}
|
||||
return classes;
|
||||
}
|
||||
|
||||
public static void main(String[] args) throws Exception {
|
||||
new PackagedSpringApplicationLauncher().run(args);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,22 @@
|
||||
/*
|
||||
* Copyright 2012-2014 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.
|
||||
*/
|
||||
|
||||
/**
|
||||
* Class that are packaged as part of CLI generated JARs.
|
||||
* @see org.springframework.boot.cli.command.jar.JarCommand
|
||||
*/
|
||||
package org.springframework.boot.cli.jar;
|
||||
|
||||
Reference in New Issue
Block a user