dolysis/rocksdb
3
0
Fork 0
mirror of https://github.com/facebook/rocksdb.git synced 2024-11-27 11:43:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #471 from fyrz/RocksJava-Fix-NativeLibraryLoader

Browse source 
[RocksJava] Fix native library loader
This commit is contained in:
Yueh-Hsuan Chiang 2015-01-26 01:35:00 -08:00
parent 26b50783d3 e61f38e5a0
commit b08b2fe732
1 changed files with 26 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
java/org/rocksdb/NativeLibraryLoader.java
View file

@ -13,6 +13,7 @@ import org.rocksdb.util.Environment;
public class NativeLibraryLoader { public class NativeLibraryLoader {
//singleton //singleton
private static final NativeLibraryLoader instance = new NativeLibraryLoader(); private static final NativeLibraryLoader instance = new NativeLibraryLoader();
private static boolean initialized = false;
private static final String sharedLibraryName = Environment.getJniLibraryName("rocksdb"); private static final String sharedLibraryName = Environment.getJniLibraryName("rocksdb");
private static final String tempFilePrefix = "librocksdbjni"; private static final String tempFilePrefix = "librocksdbjni";
@ -41,31 +42,35 @@ public class NativeLibraryLoader {
* *
* @throws java.io.IOException if a filesystem operation fails. * @throws java.io.IOException if a filesystem operation fails.
*/ */
public void loadLibraryFromJar(final String tmpDir) public synchronized void loadLibraryFromJar(final String tmpDir)
throws IOException { throws IOException {
final File temp; if (!initialized) {
if(tmpDir == null || tmpDir.equals("")) { final File temp;
temp = File.createTempFile(tempFilePrefix, tempFileSuffix); if (tmpDir == null || tmpDir.equals("")) {
} else { temp = File.createTempFile(tempFilePrefix, tempFileSuffix);
temp = new File(tmpDir, sharedLibraryName);
}
if (!temp.exists()) {
throw new RuntimeException("File " + temp.getAbsolutePath() + " does not exist.");
} else {
temp.deleteOnExit();
}
// attempt to copy the library from the Jar file to the temp destination
try(final InputStream is = getClass().getClassLoader().getResourceAsStream(sharedLibraryName)) {
if (is == null) {
throw new RuntimeException(sharedLibraryName + " was not found inside JAR.");
} else { } else {
Files.copy(is, temp.toPath(), StandardCopyOption.REPLACE_EXISTING); temp = new File(tmpDir, sharedLibraryName);
} }
}
System.load(temp.getAbsolutePath()); if (!temp.exists()) {
throw new RuntimeException("File " + temp.getAbsolutePath() + " does not exist.");
} else {
temp.deleteOnExit();
}
// attempt to copy the library from the Jar file to the temp destination
try (final InputStream is = getClass().getClassLoader().
getResourceAsStream(sharedLibraryName)) {
if (is == null) {
throw new RuntimeException(sharedLibraryName + " was not found inside JAR.");
} else {
Files.copy(is, temp.toPath(), StandardCopyOption.REPLACE_EXISTING);
}
}
System.load(temp.getAbsolutePath());
initialized = true;
}
} }
/** /**
* Private constructor to disallow instantiation * Private constructor to disallow instantiation