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

Compare commits

base: Mirrors:48922ae405f662ec2180985d6cbc15c6c2a0c607
Branches Tags
Mirrors:main
...
compare: Mirrors:3f9474207fcaccf443259b7325c78d856b35329d
Branches Tags
Mirrors:main

4 Commits
48922ae405 ... 3f9474207f

Author SHA1 Message Date
Mohanned Anwar
3f9474207f
Merge 9eb5089eae into fb54d8b549 2023-05-22 17:37:52 -05:00
Mohanned Anwar
9eb5089eae
Update NativeUtils.java 
move function
 2023-04-05 10:16:48 +03:00
Mohanned Anwar
c51fe59e2f
Update NativeUtils.java 
Catch the rest of the exceptions
 2023-04-05 10:13:47 +03:00
Mohanned Anwar
0df138171c
Update NativeUtils.java 
Refactor function loadLibraryFromJar
 2023-04-02 05:41:34 +03:00
1 changed files with 21 additions and 16 deletions
Show Stats Expand all files Collapse all files

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

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