wbtiger
/
foundationdb
mirror of https://github.com/apple/foundationdb.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4930 from apple/anoyes/external-clients-in-fdb-java 

Add exportLibrary for exporting external clients from jar
This commit is contained in:
A.J. Beamon 2021-06-15 15:36:08 -07:00 committed by GitHub
parent d3422b64c4 b92b930b93
commit a370776e9b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
bindings/java/src/main/com/apple/foundationdb/JNIUtil.java
View File

@ -30,7 +30,7 @@ import java.io.OutputStream;
* Utility for loading a dynamic library from the classpath.
*
*/
class JNIUtil {
public class JNIUtil {
private static final String SEPARATOR = "/";
private static final String LOADABLE_PREFIX = "FDB_LIBRARY_PATH_";
private static final String TEMPFILE_PREFIX = "fdbjni";
@ -92,7 +92,7 @@ class JNIUtil {
File exported;
try {
exported = exportResource(path);
exported = exportResource(path, libName);
}
catch (IOException e) {
throw new UnsatisfiedLinkError(e.getMessage());
@ -109,6 +109,19 @@ class JNIUtil {
}
}
/**
* Export a library from classpath resources to a temporary file.
*
* @param libName the name of the library to attempt to export. This name should be
* undecorated with file extensions and, in the case of *nix, "lib" prefixes.
* @return the exported temporary file
*/
public static File exportLibrary(String libName) throws IOException {
OS os = getRunningOS();
String path = getPath(os, libName);
return exportResource(path, libName);
}
/**
* Gets a relative path for a library. The path will be of the form:
* {@code {os}/{arch}/{name}}.
@ -127,20 +140,21 @@ class JNIUtil {
* Export a resource from the classpath to a temporary file.
*
* @param path the relative path of the file to load from the classpath
* @param name an optional descriptive name to include in the temporary file's path
*
* @return the absolute path to the exported file
* @throws IOException
*/
private static File exportResource(String path) throws IOException {
private static File exportResource(String path, String name) throws IOException {
InputStream resource = JNIUtil.class.getResourceAsStream(path);
if(resource == null)
throw new IllegalStateException("Embedded library jar:" + path + " not found");
File f = saveStreamAsTempFile(resource);
File f = saveStreamAsTempFile(resource, name);
return f;
}
private static File saveStreamAsTempFile(InputStream resource) throws IOException {
File f = File.createTempFile(TEMPFILE_PREFIX, TEMPFILE_SUFFIX);
private static File saveStreamAsTempFile(InputStream resource, String name) throws IOException {
File f = File.createTempFile(name.length() > 0 ? name : TEMPFILE_PREFIX, TEMPFILE_SUFFIX);
FileOutputStream outputStream = new FileOutputStream(f);
copyStream(resource, outputStream);
outputStream.flush();