feat: add serverUUID as part of states mgmt (persistence), fix(UserMgmtTest), version bump to 0.9.0.3

Author: dxps

README.md

@@ -55,12 +55,12 @@ Just include immudb4j as a dependency in your project:
   <dependency>
       <groupId>io.codenotary</groupId>
       <artifactId>immudb4j</artifactId>
-      <version>0.9.0.2</version>
+      <version>0.9.0.3</version>
   </dependency> 
   ```
 - if using Gradle:
   ```groovy
-  compile 'io.codenotary:immudb4j:0.9.0.2'
+  compile 'io.codenotary:immudb4j:0.9.0.3'
   ```
 
 `immudb4j` is currently hosted on both [Maven Central] and [Github Packages].
@@ -77,9 +77,9 @@ and _Configuring Gradle for use with GitHub Packages_ at [Github Packages].
 
 ## Supported Versions
 
-immudb4j supports the [latest immudb release], that is 0.9.1 at the time of this writing.
+immudb4j supports the [latest immudb server] release, that is 0.9.1 at the time of updating this document.
 
-[latest immudb release]: https://github.com/codenotary/immudb/releases/tag/v0.9.1
+[latest immudb server]: https://github.com/codenotary/immudb/releases/tag/v0.9.1
 
 ## Quickstart
 

build.gradle

@@ -38,7 +38,7 @@ apply plugin: 'signing'
 
 group = 'io.codenotary'
 archivesBaseName = 'immudb4j'
-version = '0.9.0.2'
+version = '0.9.0.3'
 
 sourceCompatibility = 1.8
 targetCompatibility = 1.8

immudb/clean.sh

@@ -13,5 +13,5 @@ fi
 rm -rf data
 rm -rf states
 
-echo "immudb4j data and states folders were removed."
+echo "immudb's data and immudb4j's states folders were removed."
 

src/main/java/io/codenotary/immudb4j/FileImmuStateHolder.java

@@ -45,10 +45,10 @@ private FileImmuStateHolder(Builder builder) throws IOException, IllegalStateExc
 
         stateHolder = new SerializableImmuStateHolder();
 
-        String lastRootFilename = new String(Files.readAllBytes(currentStateFile));
+        String lastStateFilename = new String(Files.readAllBytes(currentStateFile));
 
-        if (!lastRootFilename.isEmpty()) {
-            stateHolderFile = statesFolder.resolve(lastRootFilename);
+        if (!lastStateFilename.isEmpty()) {
+            stateHolderFile = statesFolder.resolve(lastStateFilename);
 
             if (Files.notExists(stateHolderFile)) {
                 throw new IllegalStateException("Inconsistent current state file");
@@ -59,40 +59,40 @@ private FileImmuStateHolder(Builder builder) throws IOException, IllegalStateExc
     }
 
     @Override
-    public synchronized ImmuState getState(String database) {
-        return stateHolder.getState(database);
+    public synchronized ImmuState getState(String serverUuid, String database) {
+        return stateHolder.getState(serverUuid, database);
     }
 
     @Override
-    public synchronized void setState(ImmuState state) throws IllegalStateException {
-        ImmuState currentState = stateHolder.getState(state.database);
+    public synchronized void setState(String serverUuid, ImmuState state) throws IllegalStateException {
 
+        ImmuState currentState = stateHolder.getState(serverUuid, state.database);
         if (currentState != null && currentState.txId >= state.txId) {
             return;
         }
 
-        stateHolder.setState(state);
+        stateHolder.setState(serverUuid, state);
 
-        Path newStateHolderFile = statesFolder.resolve("state_" + System.nanoTime());
+        Path newStateFile = statesFolder.resolve("state_" + serverUuid + "_" + state.database + "_" + System.nanoTime());
 
-        if (Files.exists(newStateHolderFile)) {
-            throw new RuntimeException("Failed attempting to create fresh state file. Please retry.");
+        if (Files.exists(newStateFile)) {
+            throw new RuntimeException("Failed attempting to create a new state file. Please retry.");
         }
 
         try {
-            Files.createFile(newStateHolderFile);
-            stateHolder.writeTo(Files.newOutputStream(newStateHolderFile));
+            Files.createFile(newStateFile);
+            stateHolder.writeTo(Files.newOutputStream(newStateFile));
 
             BufferedWriter bufferedWriter = Files.newBufferedWriter(currentStateFile, StandardOpenOption.TRUNCATE_EXISTING);
-            bufferedWriter.write(newStateHolderFile.getFileName().toString());
+            bufferedWriter.write(newStateFile.getFileName().toString());
             bufferedWriter.flush();
             bufferedWriter.close();
 
             if (stateHolderFile != null) {
                 Files.delete(stateHolderFile);
             }
 
-            stateHolderFile = newStateHolderFile;
+            stateHolderFile = newStateFile;
         } catch (IOException e) {
             e.printStackTrace();
             throw new IllegalStateException("Unexpected error " + e);

src/main/java/io/codenotary/immudb4j/ImmuClient.java

@@ -57,6 +57,7 @@ public class ImmuClient {
     private final ImmuStateHolder stateHolder;
     private ManagedChannel channel;
     private String authToken;
+    private String currentServerUuid;
     private String currentDb = "defaultdb";
 
     public ImmuClient(Builder builder) {
@@ -72,10 +73,24 @@ public static Builder newBuilder() {
     private ImmuServiceGrpc.ImmuServiceBlockingStub createStubFrom(Builder builder) {
         channel = ManagedChannelBuilder.forAddress(builder.getServerUrl(), builder.getServerPort())
                 .usePlaintext()
+                .intercept(new ImmuServerUUIDInterceptor(this))
                 .build();
         return ImmuServiceGrpc.newBlockingStub(channel);
     }
 
+    // ---------------------------------------------------------------------
+    // These two currentServerUuid related methods are not publicly exposed,
+    // since these should be called by the ImmuServerUUIDInterceptor only.
+
+    void setCurrentServerUuid(String serverUuid) {
+        currentServerUuid = serverUuid;
+    }
+
+    String getCurrentServerUuid() {
+        return currentServerUuid;
+    }
+    // ---------------------------------------------------------------------
+
     public synchronized void shutdown() {
         if (channel == null) {
             return;
@@ -99,7 +114,6 @@ private ImmuServiceGrpc.ImmuServiceBlockingStub getStub() {
         if (!withAuthToken || authToken == null) {
             return stub;
         }
-
         Metadata metadata = new Metadata();
         metadata.put(Metadata.Key.of(AUTH_HEADER, Metadata.ASCII_STRING_MARSHALLER), "Bearer " + authToken);
 
@@ -128,10 +142,10 @@ public synchronized void logout() {
      * If nothing exists already, it is fetched from the server and save it locally.
      */
     public ImmuState state() {
-        ImmuState state = stateHolder.getState(currentDb);
+        ImmuState state = stateHolder.getState(currentServerUuid, currentDb);
         if (state == null) {
             state = currentState();
-            stateHolder.setState(state);
+            stateHolder.setState(currentServerUuid, state);
         }
         return state;
     }
@@ -336,7 +350,7 @@ private Entry verifiedGet(ImmudbProto.KeyRequest keyReq, ImmuState state) throws
         // TODO: to-be-implemented (see pkg/client/client.go:620, newState.CheckSignature(c.serverSigningPubKey))
         // if (serverSigningPubKey != null) { }
 
-        stateHolder.setState(newState);
+        stateHolder.setState(currentServerUuid, newState);
 
         return Entry.valueOf(vEntry.getEntry());
     }
@@ -527,7 +541,7 @@ public TxMetadata verifiedSet(byte[] key, byte[] value) throws VerificationExcep
         // TODO: to-be-implemented (see pkg/client/client.go:803 newState.CheckSignature ...)
         // if (serverSigningPubKey != null) { ... }
 
-        stateHolder.setState(newState);
+        stateHolder.setState(currentServerUuid, newState);
 
         return TxMetadata.valueOf(vtx.getTx().getMetadata());
     }
@@ -579,7 +593,7 @@ public TxMetadata verifiedSetReferenceAt(byte[] key, byte[] referencedKey, long
         // TODO: to-be-implemented (see pkg/client/client.go:1122 newState.CheckSignature ...)
         // if (serverSigningPubKey != null) { ... }
 
-        stateHolder.setState(newState);
+        stateHolder.setState(currentServerUuid, newState);
 
         return TxMetadata.valueOf(vtx.getTx().getMetadata());
     }
@@ -686,7 +700,7 @@ public TxMetadata verifiedZAddAt(byte[] set, double score, byte[] key, long atTx
         // TODO: to-be-implemented (see pkg/client/client.go:803 newState.CheckSignature ...)
         // if (serverSigningPubKey != null) { ... }
 
-        stateHolder.setState(newState);
+        stateHolder.setState(currentServerUuid, newState);
 
         return TxMetadata.valueOf(vtx.getTx().getMetadata());
     }
@@ -769,7 +783,7 @@ public Tx verifiedTxById(long txId) throws VerificationException {
         // TODO: to-be-implemented (see pkg/client/client.go:803 newState.CheckSignature ...)
         // if (serverSigningPubKey != null) { ... }
 
-        stateHolder.setState(newState);
+        stateHolder.setState(currentServerUuid, newState);
 
         Tx tx = null;
         try {

src/main/java/io/codenotary/immudb4j/ImmuServerUUIDInterceptor.java

@@ -0,0 +1,46 @@
+package io.codenotary.immudb4j;
+
+import io.grpc.*;
+import io.grpc.ForwardingClientCallListener.SimpleForwardingClientCallListener;
+
+
+public class ImmuServerUUIDInterceptor implements ClientInterceptor {
+
+    private static final String SERVER_UUID = "immudb-uuid";
+
+    private final ImmuClient client;
+
+    public ImmuServerUUIDInterceptor(ImmuClient client) {
+        this.client = client;
+    }
+
+    @Override
+    public <ReqT, RespT> ClientCall<ReqT, RespT> interceptCall(
+            MethodDescriptor<ReqT, RespT> method,
+            CallOptions callOptions, Channel next) {
+
+        return new ForwardingClientCall.SimpleForwardingClientCall<ReqT, RespT>(next.newCall(method, callOptions)) {
+
+            @Override
+            public void start(Listener<RespT> responseListener, Metadata headers) {
+
+                SimpleForwardingClientCallListener<RespT> listener = new SimpleForwardingClientCallListener<RespT>(responseListener) {
+
+                    @Override
+                    public void onHeaders(Metadata headers) {
+                        String serverUuid = headers.get(Metadata.Key.of(SERVER_UUID, Metadata.ASCII_STRING_MARSHALLER));
+                        if (serverUuid != null && !serverUuid.equals(client.getCurrentServerUuid())) {
+                            client.setCurrentServerUuid(serverUuid);
+                            // System.out.printf(">>> ImmuServerUUIDInterceptor > Updated currentServerUuid to '%s'.\n", serverUuid);
+                        }
+                        super.onHeaders(headers);
+                    }
+                };
+                super.start(listener, headers);
+            }
+
+        };
+
+    }
+
+}

src/main/java/io/codenotary/immudb4j/ImmuState.java

@@ -15,7 +15,6 @@
 */
 package io.codenotary.immudb4j;
 
-import java.security.PublicKey;
 import java.util.Base64;
 
 /**
@@ -47,4 +46,5 @@ public String toString() {
                 ", signature(base64)=" + enc.encodeToString(signature) +
                 " }";
     }
+
 }

src/main/java/io/codenotary/immudb4j/ImmuStateHolder.java

@@ -18,8 +18,8 @@
 
 public interface ImmuStateHolder {
 
-    ImmuState getState(String database);
+    ImmuState getState(String serverUuid, String database);
 
-    void setState(ImmuState state);
+    void setState(String serverUuid, ImmuState state);
 
 }

src/main/java/io/codenotary/immudb4j/SerializableImmuStateHolder.java

@@ -26,6 +26,9 @@
 
 public class SerializableImmuStateHolder implements ImmuStateHolder {
 
+    /**
+     * Mapping "{serverUuid}_{databaseName}" to the appropriate state.
+     */
     private Map<String, ImmuState> statesMap = new HashMap<>();
 
     public void readFrom(InputStream is) {
@@ -43,13 +46,13 @@ public void writeTo(OutputStream os) throws IOException {
     }
 
     @Override
-    public ImmuState getState(String database) {
-        return this.statesMap.get(database);
+    public ImmuState getState(String serverUuid, String database) {
+        return this.statesMap.get(serverUuid + "_" + database);
     }
 
     @Override
-    public void setState(ImmuState state) {
-        this.statesMap.put(state.database, state);
+    public void setState(String serverUuid, ImmuState state) {
+        this.statesMap.put(serverUuid + "_" + state.database, state);
     }
 
 }

src/test/java/io/codenotary/immudb4j/FileImmuStateHolderTest.java

@@ -18,7 +18,7 @@ public void t1() {
         File statesDir = null;
         try {
             String tempDir = System.getProperty("java.io.tmpdir");
-            statesDir = new File(tempDir, "fileimmustateholdertest_states/");
+            statesDir = new File(tempDir, "FileImmuStateHolderTest_states/");
             File currStateFile = new File(statesDir, "current_state");
 
             // Write some fake "state_..." into "current_state" file.
@@ -31,6 +31,9 @@ public void t1() {
         } catch (IOException e) {
             // If that would be the case, it's not the test's fault.
             System.out.println(">>> Got IO ex (expected): " + e.getMessage());
+            if (e.getCause() != null) {
+                System.out.println(">>> Got IO ex (expected) cause: " + e.getCause().getMessage());
+            }
         } catch (IllegalStateException ignored) {
             // This should actually happen with that "fake" state entry.
         }