generated from creek-service/single-module-template
-
-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Include minimum Java version supported by implementations
fixes: #57
- Loading branch information
1 parent
ede81c0
commit 35d2e3e
Showing
8 changed files
with
407 additions
and
226 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
190 changes: 190 additions & 0 deletions
190
src/main/java/org/creekservice/kafka/test/perf/util/ImplJarFile.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,190 @@ | ||
| /* | ||
| * Copyright 2023 Creek Contributors (https://github.com/creek-service) | ||
| * | ||
| * 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.creekservice.kafka.test.perf.util; | ||
|
|
||
| import static java.util.Objects.requireNonNull; | ||
|
|
||
| import com.google.common.annotations.VisibleForTesting; | ||
| import edu.umd.cs.findbugs.annotations.SuppressFBWarnings; | ||
| import java.io.IOException; | ||
| import java.io.InputStream; | ||
| import java.net.URL; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.nio.file.Paths; | ||
| import java.security.CodeSource; | ||
| import java.util.Enumeration; | ||
| import java.util.function.Function; | ||
| import java.util.jar.JarEntry; | ||
| import java.util.jar.JarFile; | ||
| import org.objectweb.asm.ClassReader; | ||
|
|
||
| /** Utility for determining the size of a jar file a class is loaded from. */ | ||
| public final class ImplJarFile { | ||
|
|
||
| /** | ||
| * Determine the size of the jar the supplied type is loaded from. | ||
| * | ||
| * @param type a class loaded from the jar. | ||
| * @return the size of the jar, in bytes. | ||
| * @throws IllegalArgumentException if the supplied {@code type} isn't loaded from an accessible | ||
| * jar. | ||
| */ | ||
| public static long jarSizeForClass(final Class<?> type) { | ||
| return new JarSize(JarTaskBase::sourceCode).size(type); | ||
| } | ||
|
|
||
| /** | ||
| * Determine the minimum supported Java version for the jar the supplied type is loaded from. | ||
| * | ||
| * @param type a class loaded from the jar. | ||
| * @return the minimum supported Java version | ||
| * @throws IllegalArgumentException if the supplied {@code type} isn't loaded from an accessible | ||
| * jar. | ||
| */ | ||
| public static String jarMinJavaVersion(final Class<?> type) { | ||
| return new JarMinJavaVersion(JarTaskBase::sourceCode).minJavaVersion(type); | ||
| } | ||
|
|
||
| private ImplJarFile() {} | ||
|
|
||
| @VisibleForTesting | ||
| static class JarTaskBase { | ||
|
|
||
| final Function<Class<?>, CodeSource> sourceAccessor; | ||
|
|
||
| JarTaskBase(final Function<Class<?>, CodeSource> locator) { | ||
| this.sourceAccessor = requireNonNull(locator, "locator"); | ||
| } | ||
|
|
||
| @SuppressFBWarnings(value = "PATH_TRAVERSAL_IN", justification = "Location supplied by JVM") | ||
| Path pathToJar(final Class<?> type) { | ||
| final URL location = location(type); | ||
|
|
||
| try { | ||
| return Paths.get(location.toURI()); | ||
| } catch (final Exception e) { | ||
| throw new IllegalArgumentException( | ||
| "Type not loaded from accessible jar file. location: " + location, e); | ||
| } | ||
| } | ||
|
|
||
| private URL location(final Class<?> type) { | ||
| final CodeSource codeSource = sourceAccessor.apply(type); | ||
| if (codeSource == null) { | ||
| throw new IllegalArgumentException("Type not loaded from a jar file: " + type); | ||
| } | ||
|
|
||
| return codeSource.getLocation(); | ||
| } | ||
|
|
||
| private static CodeSource sourceCode(final Class<?> type) { | ||
| return type.getProtectionDomain().getCodeSource(); | ||
| } | ||
| } | ||
|
|
||
| @VisibleForTesting | ||
| static final class JarSize extends JarTaskBase { | ||
|
|
||
| JarSize(final Function<Class<?>, CodeSource> locator) { | ||
| super(locator); | ||
| } | ||
|
|
||
| long size(final Class<?> type) { | ||
| final Path path = pathToJar(type); | ||
|
|
||
| try { | ||
| return Files.size(path); | ||
| } catch (final Exception e) { | ||
| throw new IllegalArgumentException( | ||
| "Failed to read file size. location: " + path, e); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| @VisibleForTesting | ||
| static final class JarMinJavaVersion extends JarTaskBase { | ||
|
|
||
| JarMinJavaVersion(final Function<Class<?>, CodeSource> locator) { | ||
| super(locator); | ||
| } | ||
|
|
||
| String minJavaVersion(final Class<?> type) { | ||
| final Path path = pathToJar(type); | ||
|
|
||
| try (JarFile jarFile = new JarFile(path.toFile())) { | ||
| final Enumeration<JarEntry> entries = jarFile.entries(); | ||
|
|
||
| while (entries.hasMoreElements()) { | ||
| final JarEntry entry = entries.nextElement(); | ||
|
|
||
| if (entry.getName().endsWith(".class")) { | ||
| try (InputStream classInputStream = jarFile.getInputStream(entry)) { | ||
| return minJavaVersion(classInputStream); | ||
| } | ||
| } | ||
| } | ||
| } catch (IOException e) { | ||
| throw new IllegalArgumentException( | ||
| "Failed to extract the min Java version the jar supports. Path: " + path, | ||
| e); | ||
| } | ||
|
|
||
| return "Unknown"; | ||
| } | ||
|
|
||
| public static String minJavaVersion(final InputStream classFile) throws IOException { | ||
| final ClassReader classReader = new ClassReader(classFile); | ||
| final VersionClassVisitor versionClassVisitor = new VersionClassVisitor(); | ||
| classReader.accept(versionClassVisitor, 0); | ||
|
|
||
| final int majorVersion = versionClassVisitor.majorVersion; | ||
| if (majorVersion < 0) { | ||
| throw new IllegalArgumentException("Failed to determine Java version"); | ||
| } | ||
|
|
||
| final String javaVersion = | ||
| majorVersion < 49 | ||
| ? "1." + (majorVersion - 44) | ||
| : String.valueOf(majorVersion - 44); | ||
|
|
||
| return "Java " + javaVersion; | ||
| } | ||
|
|
||
| private static class VersionClassVisitor extends org.objectweb.asm.ClassVisitor { | ||
|
|
||
| private int majorVersion = -1; | ||
|
|
||
| VersionClassVisitor() { | ||
| super(org.objectweb.asm.Opcodes.ASM9); | ||
| } | ||
|
|
||
| @Override | ||
| public void visit( | ||
| final int version, | ||
| final int access, | ||
| final String name, | ||
| final String signature, | ||
| final String superName, | ||
| final String[] interfaces) { | ||
| if (version > majorVersion) { | ||
| majorVersion = version; | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } |
91 changes: 0 additions & 91 deletions
91
src/main/java/org/creekservice/kafka/test/perf/util/JarFile.java
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.