WIP: [Dataset] Initialize compute module

Author: lidavidm

dataset/src/main/cpp/jni_wrapper.cc

@@ -23,6 +23,7 @@
 #include "arrow/array/concatenate.h"
 #include "arrow/c/bridge.h"
 #include "arrow/c/helpers.h"
+#include "arrow/compute/initialize.h"
 #include "arrow/dataset/api.h"
 #include "arrow/dataset/file_base.h"
 #ifdef ARROW_CSV
@@ -807,6 +808,13 @@ JNIEXPORT void JNICALL Java_org_apache_arrow_dataset_jni_JniWrapper_ensureS3Fina
   JNI_METHOD_END()
 }
 
+JNIEXPORT void JNICALL Java_org_apache_arrow_dataset_jni_JniWrapper_initialize(
+    JNIEnv* env, jobject) {
+  JNI_METHOD_START
+  JniAssertOkOrThrow(arrow::compute::Initialize());
+  JNI_METHOD_END()
+}
+
 /*
  * Class:     org_apache_arrow_dataset_file_JniWrapper
  * Method:    makeFileSystemDatasetFactory

dataset/src/main/java/org/apache/arrow/dataset/jni/JniLoader.java

@@ -56,6 +56,7 @@ public void ensureLoaded() {
     }
     loadRemaining();
     ensureS3FinalizedOnShutdown();
+    JniWrapper.get().initialize();
   }
 
   private synchronized void loadRemaining() {

dataset/src/main/java/org/apache/arrow/dataset/jni/JniWrapper.java

@@ -124,4 +124,9 @@ public native long createScanner(
    * uninitialized, then this is a noop.
    */
   public native void ensureS3Finalized();
+
+  /**
+   * Initialize Arrow Compute.
+   */
+  public native void initialize();
 }