File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
sdk/src/main/java/com/google/cloud/dataflow/sdk/runners/worker Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -32,9 +32,13 @@ class ShuffleLibrary {
32
32
*/
33
33
static void load () {
34
34
try {
35
+ final String shuffleClientLibraryPropertyKey = "batch.shuffle_client_library" ;
36
+ String shuffleClientLibrary = "libshuffle_client_jni.so.stripped" ;
37
+ if (System .getProperties ().containsKey (shuffleClientLibraryPropertyKey )) {
38
+ shuffleClientLibrary = System .getProperty (shuffleClientLibraryPropertyKey );
39
+ }
35
40
File tempfile = File .createTempFile ("libshuffle_client_jni" , ".so" );
36
- InputStream input = ClassLoader .getSystemResourceAsStream (
37
- "libshuffle_client_jni.so.stripped" );
41
+ InputStream input = ClassLoader .getSystemResourceAsStream (shuffleClientLibrary );
38
42
Files .copy (input , tempfile .toPath (), StandardCopyOption .REPLACE_EXISTING );
39
43
System .load (tempfile .getAbsolutePath ());
40
44
} catch (IOException e ) {
You can’t perform that action at this time.
0 commit comments