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

11
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,10 +42,11 @@ 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 {
if (!initialized) {
final File temp; final File temp;
if(tmpDir == null || tmpDir.equals("")) { if (tmpDir == null || tmpDir.equals("")) {
temp = File.createTempFile(tempFilePrefix, tempFileSuffix); temp = File.createTempFile(tempFilePrefix, tempFileSuffix);
} else { } else {
temp = new File(tmpDir, sharedLibraryName); temp = new File(tmpDir, sharedLibraryName);
@ -57,7 +59,8 @@ public class NativeLibraryLoader {
} }
// attempt to copy the library from the Jar file to the temp destination // attempt to copy the library from the Jar file to the temp destination
try(final InputStream is = getClass().getClassLoader().getResourceAsStream(sharedLibraryName)) { try (final InputStream is = getClass().getClassLoader().
getResourceAsStream(sharedLibraryName)) {
if (is == null) { if (is == null) {
throw new RuntimeException(sharedLibraryName + " was not found inside JAR."); throw new RuntimeException(sharedLibraryName + " was not found inside JAR.");
} else { } else {
@ -66,6 +69,8 @@ public class NativeLibraryLoader {
} }
System.load(temp.getAbsolutePath()); System.load(temp.getAbsolutePath());
initialized = true;
}
} }
/** /**
* Private constructor to disallow instantiation * Private constructor to disallow instantiation