Hello rust

Signed-off-by: Harsha Vamsi Kalluri <[email protected]>

rusty-search/.gitignore

@@ -0,0 +1 @@
+target/

rusty-search/Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "rust_library"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+jni = "0.21.1"
+
+[lib]
+crate_type = ["cdylib"]

rusty-search/build.gradle

@@ -0,0 +1,9 @@
+plugins{
+    id 'io.github.arc-blroth.cargo-wrapper' version '1.0.0' apply true
+}
+
+
+cargo {
+    outputs = ['': System.mapLibraryName('rust_library')]
+    cargoCommand="/opt/homebrew/bin/cargo"
+}

rusty-search/src/lib.rs

@@ -0,0 +1,6 @@
+use jni::JNIEnv;
+
+#[no_mangle]
+pub extern "system" fn Java_org_opensearch_search_query_QueryPhase_doStuff(_env: JNIEnv) {
+    println!("Hello Rust!");
+}

server/build.gradle

@@ -58,6 +58,15 @@ sourceSets {
   }
 }
 
+configurations {
+    // Declare a custom configuration to
+    // resolve the library from :native
+    backend {
+        canBeConsumed false
+        canBeResolved true
+    }
+}
+
 dependencies {
 
   api project(':libs:opensearch-common')
@@ -131,6 +140,13 @@ dependencies {
   internalClusterTestImplementation(project(":test:framework")) {
     exclude group: 'org.opensearch', module: 'server'
   }
+
+    backend(project(':rusty-search'))
+}
+
+processResources {
+    // Copy the native library into the final jar
+    from(configurations.backend)
 }
 
 tasks.withType(JavaCompile).configureEach {

server/src/main/java/org/opensearch/search/query/QueryPhase.java

@@ -71,7 +71,12 @@
 import org.opensearch.search.suggest.SuggestProcessor;
 import org.opensearch.threadpool.ThreadPool;
 
+import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
+import java.nio.channels.Channels;
+import java.nio.channels.FileChannel;
+import java.nio.channels.ReadableByteChannel;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
@@ -101,6 +106,26 @@ public class QueryPhase {
     private final SuggestProcessor suggestProcessor;
     private final RescoreProcessor rescoreProcessor;
 
+    private native static void doStuff();
+
+    static {
+        try {
+            // Extract and load our native library
+            String nativeLibraryName = System.mapLibraryName("rust_library");
+            File tempFile = File.createTempFile("extracted_", nativeLibraryName);
+            try (
+                ReadableByteChannel src = Channels.newChannel(QueryPhase.class.getClassLoader().getResourceAsStream(nativeLibraryName));
+                FileChannel dst = new FileOutputStream(tempFile).getChannel()
+            ) {
+                dst.transferFrom(src, 0, Long.MAX_VALUE);
+            }
+            System.load(tempFile.getAbsolutePath());
+        } catch (Exception e) {
+            e.printStackTrace();
+            System.exit(-1);
+        }
+    }
+
     public QueryPhase() {
         this(DEFAULT_QUERY_PHASE_SEARCHER);
     }
@@ -353,6 +378,7 @@ private static boolean searchWithCollector(
         }
         QuerySearchResult queryResult = searchContext.queryResult();
         try {
+            doStuff();
             searcher.search(query, queryCollector);
         } catch (EarlyTerminatingCollector.EarlyTerminationException e) {
             queryResult.terminatedEarly(true);

settings.gradle

@@ -91,7 +91,8 @@ List projects = [
   'test:fixtures:old-elasticsearch',
   'test:fixtures:s3-fixture',
   'test:logger-usage',
-  'test:telemetry'
+  'test:telemetry',
+  'rusty-search'
 ]
 
 /**