Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-12-23 02:31:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge 9eb5089eae into fb54d8b549

Browse Source
This commit is contained in:
Mohanned Anwar 2023-05-22 17:37:52 -05:00 committed by GitHub
commit 3f9474207f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
ann/src/main/java/com/twitter/ann/faiss/NativeUtils.java
View File

@ -45,6 +45,9 @@ public final class NativeUtils {
} catch (NullPointerException e) {
temp.delete();
throw new FileNotFoundException("File " + path + " was not found inside JAR.");
} catch (Exception e) {
file.delete();
throw e;
}
return temp;
@ -91,22 +94,24 @@ public final class NativeUtils {
* JAR.
*/
public static void loadLibraryFromJar(String path) throws IOException {
File temp = unpackLibraryFromJarInternal(path);
try (InputStream is = NativeUtils.class.getResourceAsStream(path)) {
Files.copy(is, temp.toPath(), StandardCopyOption.REPLACE_EXISTING);
} catch (IOException e) {
temp.delete();
throw e;
} catch (NullPointerException e) {
temp.delete();
throw new FileNotFoundException("File " + path + " was not found inside JAR.");
File tempFile = unpackLibraryFromJarInternal(path);
copyResourceToFile(path, tempFile);
try {
System.load(tempFile.getAbsolutePath());
} finally {
tempFile.deleteOnExit();
}
}
try {
System.load(temp.getAbsolutePath());
} finally {
temp.deleteOnExit();
private static void copyResourceToFile(String resourcePath, File file) throws IOException {
try ( InputStream inputStream = NativeUtils.class.getResourceAsStream(resourcePath)) {
Files.copy(inputStream, file.toPath(), StandardCopyOption.REPLACE_EXISTING);
} catch (IOException e) {
file.delete();
throw e;
} catch (NullPointerException e) {
file.delete();
throw new FileNotFoundException("File " + resourcePath + " was not found inside JAR.");
}
}