Merge pull request #2 from nodesmith/mongo

Mongo read only concept

modDbImpl/module-info.java

@@ -5,6 +5,7 @@
     requires leveldbjni.all;
     requires rocksdbjni;
     requires h2.mvstore;
+    requires java.security.sasl;
 
 	exports org.aion.db.impl;
 	exports org.aion.db.impl.leveldb;

modDbImpl/src/org/aion/db/impl/mongodb/MongoDB.java

@@ -166,10 +166,13 @@ public boolean matchedExpectation(WriteBatch batch) {
     private ClientSession clientSession;
     private MongoCollection<BsonDocument> collection = null;
     private WriteBatch batch = null;
+    private boolean isReader = false;
 
     public MongoDB(String dbName, String mongoClientUri) {
         super(dbName);
         this.mongoClientUri = mongoClientUri;
+
+        this.isReader = mongoClientUri.contains("reader");
     }
 
     /**
@@ -272,6 +275,10 @@ public boolean commitCache(Map<ByteArrayWrapper, byte[]> cache) {
         check();
         check(cache.keySet().stream().map(k -> k.getData()).collect(Collectors.toList()));
 
+        if (this.isReader) {
+            return true;
+        }
+
         WriteBatch edits = new WriteBatch().addEditsWrapper(cache);
         WriteBatchResult result = doBulkWrite(edits);
 
@@ -293,6 +300,10 @@ public void put(byte[] key, byte[] value) {
         check();
         check(key);
 
+        if (this.isReader) {
+            return;
+        }
+
         // Write this single edit in as a batch
         WriteBatch edits = new WriteBatch().addEdit(key, value);
         doBulkWrite(edits);
@@ -303,6 +314,10 @@ public void delete(byte[] key) {
         check();
         check(key);
 
+        if (this.isReader) {
+            return;
+        }
+
         // Write this single edit in as a batch
         WriteBatch edits = new WriteBatch().addEdit(key, null);
         doBulkWrite(edits);
@@ -313,6 +328,10 @@ public void putBatch(Map<byte[], byte[]> inputMap) {
         check();
         check(inputMap.keySet());
 
+        if (this.isReader) {
+            return;
+        }
+
         WriteBatch edits = new WriteBatch().addEdits(inputMap);
         doBulkWrite(edits);
     }
@@ -322,6 +341,10 @@ public void putToBatch(byte[] key, byte[] value) {
         check();
         check(key);
 
+        if (this.isReader) {
+            return;
+        }
+
         if (this.batch == null) {
             this.batch = new WriteBatch();
         }
@@ -333,6 +356,10 @@ public void putToBatch(byte[] key, byte[] value) {
     public void commitBatch() {
         check();
 
+        if (this.isReader) {
+            return;
+        }
+
         if (this.batch != null) {
             LOG.info("Committing batch of writes");
             doBulkWrite(this.batch);
@@ -348,6 +375,11 @@ public void deleteBatch(Collection<byte[]> keys) {
         check();
         check(keys);
 
+
+        if (this.isReader) {
+            return;
+        }
+
         if (!keys.isEmpty()) {
             Map<byte[], byte[]> batch = new HashMap();
             keys.forEach(key -> batch.put(key, null));
@@ -374,6 +406,10 @@ public void close() {
     public void drop() {
         check();
 
+        if (this.isReader) {
+            return;
+        }
+
         LOG.info("Dropping collection {}", this.name);
         this.collection.drop(this.clientSession);
     }

script/prepack.sh

@@ -21,7 +21,7 @@ fi
 
 # generate aion runtime
 if [ ! -d "$JDK_RT" ]; then
-  $JDK_PATH/bin/jlink --module-path $JDK_PATH/jmods --add-modules java.base,java.xml,java.logging,java.management,jdk.unsupported,javafx.graphics,javafx.controls,javafx.base,jdk.sctp,javafx.fxml,javafx.swing --output $JDK_RT
+  $JDK_PATH/bin/jlink --module-path $JDK_PATH/jmods --add-modules java.base,java.xml,java.logging,java.management,jdk.unsupported,javafx.graphics,javafx.controls,javafx.base,jdk.sctp,javafx.fxml,javafx.swing,java.security.sasl --output $JDK_RT
   cp $JDK_PATH/bin/jstack $JDK_RT/bin
 fi