update definition of native pixels cache reader

Author: bianhq

cpp/Makefile

@@ -1,7 +1,7 @@
-all: JNITest
+all: NativePixelsCacheReader
 
-JNITest:
-	$(CXX) -shared -fPIC JNITest.cc -o JNITest.so -I/usr/lib/jvm/java-1.8.0-openjdk-amd64/include/ -I/usr/lib/jvm/java-8-openjdk-amd64/include/linux/
+NativePixelsCacheReader:
+	$(CXX) -shared -fPIC $@.cc -o lib_pixels.so -I/usr/lib/jvm/java-1.8.0-openjdk-amd64/include/ -I/usr/lib/jvm/java-8-openjdk-amd64/include/linux/
 
 clean:
-	rm -rf ./JNITest.so
+	rm -rf ./lib_pixels.so

cpp/NativePixelsCacheReader.cc

@@ -0,0 +1,18 @@
+#include <iostream>
+#include "NativePixelsCacheReader.h"
+using namespace std;
+
+JNIEXPORT jbyteArray JNICALL Java_cn_edu_ruc_iir_pixels_cache_NativePixelsCacheReader_get
+  (JNIEnv *env, jclass cls, jlong blockId, jshort rowGroupId, jshort columnId)
+{
+    jbyteArray res;
+    return res;
+}
+
+
+JNIEXPORT jbyteArray JNICALL Java_cn_edu_ruc_iir_pixels_cache_NativePixelsCacheReader_sch
+  (JNIEnv *env, jclass cls, jlong blockId, jshort rowGroupId, jshort columnId)
+{
+    jbyteArray res;
+    return res;
+}

cpp/NativePixelsCacheReader.h

@@ -1,5 +1,8 @@
 package cn.edu.ruc.iir.pixels.cache;
 
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
+
 /**
  * Created at: 19-5-11
  * Author: hank
@@ -10,6 +13,7 @@ public class NativePixelsCacheReader
     private long indexFileSize;
     private String cacheFileLocation;
     private long cacheFileSize;
+    private ByteBuffer keyBuffer = ByteBuffer.allocate(PixelsCacheKey.SIZE).order(ByteOrder.BIG_ENDIAN);
 
     public NativePixelsCacheReader(String indexFileLocation, long indexFileSize,
                                    String cacheFileLocation, long cacheFileSize)
@@ -25,5 +29,49 @@ public NativePixelsCacheReader(String indexFileLocation, long indexFileSize,
         System.loadLibrary("lib_pixels.so");
     }
 
-    private native byte[] getFromCache();
+    public static class Builder
+    {
+        private String builderIndexFileLocation;
+        private long builderIndexFileSize;
+        private String builderCacheFileLocation;
+        private long builderCacheFileSize;
+
+        private Builder()
+        {
+        }
+
+        public NativePixelsCacheReader.Builder setCacheFile(String location, long size)
+        {
+            this.builderCacheFileLocation = location;
+            this.builderCacheFileSize = size;
+            return this;
+        }
+
+        public NativePixelsCacheReader.Builder setIndexFile(String location, long size)
+        {
+            this.builderIndexFileLocation = location;
+            this.builderIndexFileSize = size;
+            return this;
+        }
+
+        public NativePixelsCacheReader build()
+        {
+            return new NativePixelsCacheReader(builderIndexFileLocation, builderIndexFileSize,
+                    builderCacheFileLocation, builderCacheFileSize);
+        }
+    }
+
+    public static NativePixelsCacheReader.Builder newBuilder()
+    {
+        return new NativePixelsCacheReader.Builder();
+    }
+
+    public static native byte[] get(long blockId, short rowGroupId, short columnId);
+
+    public PixelsCacheIdx search(long blockId, short rowGroupId, short columnId)
+    {
+        return new PixelsCacheIdx(sch(blockId, rowGroupId, columnId));
+    }
+
+    private static native byte[] sch(long blockId, short rowGroupId, short columnId);
 }