Adding JNI abstractions for logging in rust through RustLoggerBridge (#19978)

* Adding JNI abstractions for logging in rust through RustLoggerBridge

Signed-off-by: Raghuvansh Raj <[email protected]>

* Adding logging for java in parquet-data-format module and cleaning up stale files

Signed-off-by: Raghuvansh Raj <[email protected]>

---------

Signed-off-by: Raghuvansh Raj <[email protected]>

Author: raghuvanshraj

modules/parquet-data-format/src/main/java/com/parquet/parquetdataformat/bridge/RustBridge.java

@@ -21,8 +21,13 @@ public class RustBridge {
 
     static {
         NativeLibraryLoader.load("parquet_dataformat_jni");
+
+        initLogger();
     }
 
+    // Logger initialization method
+    public static native void initLogger();
+
     // Enhanced native methods that handle validation and provide better error reporting
     public static native void createWriter(String file, long schemaAddress) throws IOException;
     public static native void write(String file, long arrayAddress, long schemaAddress) throws IOException;

modules/parquet-data-format/src/main/java/com/parquet/parquetdataformat/bridge/RustLoggerBridge.java

@@ -0,0 +1,34 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package com.parquet.parquetdataformat.bridge;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class RustLoggerBridge {
+
+    private static final Logger logger = LoggerFactory.getLogger(RustLoggerBridge.class);
+
+    // Instance methods for direct Java usage
+    public static void logInfo(String message) {
+        logger.info(message);
+    }
+
+    public static void logWarn(String message) {
+        logger.warn(message);
+    }
+
+    public static void logError(String message) {
+        logger.error(message);
+    }
+
+    public static void logDebug(String message) {
+        logger.debug(message);
+    }
+}

modules/parquet-data-format/src/main/java/com/parquet/parquetdataformat/memory/MemoryPressureMonitor.java

@@ -10,6 +10,8 @@
 import org.apache.arrow.c.ArrowArray;
 import org.apache.arrow.c.ArrowSchema;
 import org.apache.arrow.c.Data;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.concurrent.locks.ReadWriteLock;
@@ -23,6 +25,8 @@
  */
 public class ManagedVSR implements AutoCloseable {
 
+    private static final Logger logger = LogManager.getLogger(ManagedVSR.class);
+
     private final String id;
     private final VectorSchemaRoot vsr;
     private final BufferAllocator allocator;
@@ -109,9 +113,7 @@ public FieldVector getVector(String fieldName) {
     public void setState(VSRState newState) {
         VSRState oldState = state.getAndSet(newState);
 
-        System.out.println(String.format(
-            "[VSR] State transition: %s -> %s for VSR %s",
-            oldState, newState, id));
+        logger.debug("State transition: {} -> {} for VSR {}", oldState, newState, id);
     }
 
     /**
@@ -226,34 +228,4 @@ public String toString() {
         return String.format("ManagedVSR{id='%s', state=%s, rows=%d, immutable=%s}",
             id, state.get(), getRowCount(), isImmutable());
     }
-
-    public static void main(String[] args) {
-        RootAllocator allocator = new RootAllocator();
-        BigIntVector vector = new BigIntVector("vector", allocator);
-        vector.allocateNew(10);
-        vector.set(0, 100);  // Set position 0
-//        vector.setNull(1);
-        vector.set(2, 300);  // Set position 2
-// Position 1 is not set!
-        vector.setValueCount(3);  // Claims vector has 3 elements
-
-// Position 1 now contains undefined data
-//        long value = vector.get(1);  // Could be any value!
-        System.out.println(readBit(vector.getValidityBuffer(), 0));
-        System.out.println(readBit(vector.getValidityBuffer(), 1));
-        System.out.println(readBit(vector.getValidityBuffer(), 2));
-        System.out.println(readBit(vector.getValidityBuffer(), 3));
-    }
-
-    public static byte readBit(ArrowBuf validityBuffer, long index) {
-        // it can be observed that some logic is duplicate of the logic in setValidityBit.
-        // this is because JIT cannot always remove the if branch in setValidityBit,
-        // so we give a dedicated implementation for setting bits.
-        final long byteIndex = byteIndex(index);
-
-        // the byte is promoted to an int, because according to Java specification,
-        // bytes will be promoted to ints automatically, upon expression evaluation.
-        // by promoting it manually, we avoid the unnecessary conversions.
-        return validityBuffer.getByte(byteIndex);
-    }
 }