Implement custom deserializer for NetworkMessage

src/main/java/io/github/chronosx88/JGUN/examples/MainServer.java

@@ -1,11 +1,39 @@
 package io.github.chronosx88.JGUN.examples;
 
+import io.github.chronosx88.JGUN.api.Gun;
+import io.github.chronosx88.JGUN.api.graph.ArrayBuilder;
+import io.github.chronosx88.JGUN.api.graph.NodeBuilder;
+import io.github.chronosx88.JGUN.models.Result;
 import io.github.chronosx88.JGUN.network.GatewayNetworkNode;
+import io.github.chronosx88.JGUN.network.NetworkNode;
 import io.github.chronosx88.JGUN.storage.MemoryStorage;
+import io.github.chronosx88.JGUN.storage.Storage;
+
+import java.net.Inet4Address;
+import java.util.concurrent.ExecutionException;
 
 public class MainServer {
-    public static void main(String[] args) {
-        GatewayNetworkNode gunSuperNode = new GatewayNetworkNode(5054, new MemoryStorage());
-        gunSuperNode.start();
+    public static void main(String[] args) throws ExecutionException, InterruptedException {
+        GatewayNetworkNode peer = new GatewayNetworkNode(5054, new MemoryStorage());
+        Storage storage = new MemoryStorage();
+        Gun gun = new Gun(storage, peer);
+        Result result = gun.get("person").put(new NodeBuilder()
+                .add("firstName", "John")
+                .add("lastName", "Smith")
+                .add("age", 25)
+                .add("address", new NodeBuilder()
+                        .add("streetAddress", "21 2nd Street")
+                        .add("city", "New York")
+                        .add("state", "NY")
+                        .add("postalCode", "10021"))
+                .add("phoneNumber", new ArrayBuilder()
+                        .add(new NodeBuilder()
+                                .add("type", "home")
+                                .add("number", "212 555-1234"))
+                        .add(new NodeBuilder()
+                                .add("type", "fax")
+                                .add("number", "646 555-4567")))
+                .build()).get();
+        System.out.println(result.isOk());
     }
 }

src/main/java/io/github/chronosx88/JGUN/models/NetworkMessage.java

@@ -12,15 +12,9 @@ import lombok.Data;
 import lombok.experimental.SuperBuilder;
 
 @Data
-@JsonTypeInfo(use = JsonTypeInfo.Id.DEDUCTION)
-@JsonSubTypes({
-        @JsonSubTypes.Type(GetRequest.class),
-        @JsonSubTypes.Type(PutRequest.class),
-        @JsonSubTypes.Type(Ack.class),
-        @JsonSubTypes.Type(GetAck.class)
-})
+@JsonDeserialize(using = NetworkMessageDeserializer.class)
 @SuperBuilder
-public abstract class BaseMessage {
+public abstract class NetworkMessage {
     @JsonProperty("#")
     private String id;
 }

src/main/java/io/github/chronosx88/JGUN/models/NetworkMessageDeserializer.java

@@ -0,0 +1,37 @@
+package io.github.chronosx88.JGUN.models;
+
+import com.fasterxml.jackson.core.JacksonException;
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.databind.DeserializationContext;
+import com.fasterxml.jackson.databind.JsonDeserializer;
+import com.fasterxml.jackson.databind.JsonNode;
+import io.github.chronosx88.JGUN.models.acks.Ack;
+import io.github.chronosx88.JGUN.models.acks.GetAck;
+import io.github.chronosx88.JGUN.models.requests.GetRequest;
+import io.github.chronosx88.JGUN.models.requests.PutRequest;
+
+import java.io.IOException;
+
+public class NetworkMessageDeserializer extends JsonDeserializer<NetworkMessage> {
+    @Override
+    public NetworkMessage deserialize(JsonParser p, DeserializationContext ctxt) throws IOException, JacksonException {
+        JsonNode node = p.readValueAsTree();
+
+        if (node.has("#")) {
+            // parsing ack
+            if (node.has("@")) {
+                if (node.has("put")) {
+                    return p.getCodec().treeToValue(node, GetAck.class);
+                }
+                return p.getCodec().treeToValue(node, Ack.class);
+            }
+
+            if (node.has("get")) {
+                return p.getCodec().treeToValue(node, GetRequest.class);
+            } else if (node.has("put")) {
+                return p.getCodec().treeToValue(node, PutRequest.class);
+            }
+        }
+        throw new IllegalArgumentException("invalid message received");
+    }
+}

src/main/java/io/github/chronosx88/JGUN/models/acks/Ack.java

@@ -1,7 +1,7 @@
 package io.github.chronosx88.JGUN.models.acks;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import io.github.chronosx88.JGUN.models.BaseMessage;
+import io.github.chronosx88.JGUN.models.NetworkMessage;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 import lombok.experimental.SuperBuilder;
@@ -11,7 +11,7 @@ import lombok.extern.jackson.Jacksonized;
 @SuperBuilder
 @EqualsAndHashCode(callSuper = true)
 @Jacksonized
-public class Ack extends BaseMessage {
+public class Ack extends NetworkMessage {
     @JsonProperty("@")
     private String replyTo;
     private boolean ok;

src/main/java/io/github/chronosx88/JGUN/models/acks/GetAck.java

@@ -1,7 +1,7 @@
 package io.github.chronosx88.JGUN.models.acks;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import io.github.chronosx88.JGUN.models.BaseMessage;
+import io.github.chronosx88.JGUN.models.NetworkMessage;
 import io.github.chronosx88.JGUN.models.graph.MemoryGraph;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -12,7 +12,7 @@ import lombok.extern.jackson.Jacksonized;
 @SuperBuilder
 @EqualsAndHashCode(callSuper = true)
 @Jacksonized
-public class GetAck extends BaseMessage {
+public class GetAck extends NetworkMessage {
     @JsonProperty("put")
     private MemoryGraph data;
     @JsonProperty("@")

src/main/java/io/github/chronosx88/JGUN/models/requests/GetRequest.java

@@ -1,8 +1,7 @@
 package io.github.chronosx88.JGUN.models.requests;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import io.github.chronosx88.JGUN.models.BaseMessage;
-import lombok.Builder;
+import io.github.chronosx88.JGUN.models.NetworkMessage;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 import lombok.experimental.SuperBuilder;
@@ -12,7 +11,7 @@ import lombok.extern.jackson.Jacksonized;
 @SuperBuilder
 @EqualsAndHashCode(callSuper = true)
 @Jacksonized
-public class GetRequest extends BaseMessage implements Request {
+public class GetRequest extends NetworkMessage implements Request {
     @JsonProperty("get")
     private GetRequestParams params;
 }

src/main/java/io/github/chronosx88/JGUN/models/requests/PutRequest.java

@@ -1,7 +1,7 @@
 package io.github.chronosx88.JGUN.models.requests;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import io.github.chronosx88.JGUN.models.BaseMessage;
+import io.github.chronosx88.JGUN.models.NetworkMessage;
 import io.github.chronosx88.JGUN.models.graph.MemoryGraph;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -12,7 +12,7 @@ import lombok.extern.jackson.Jacksonized;
 @Jacksonized
 @SuperBuilder
 @EqualsAndHashCode(callSuper = true)
-public class PutRequest extends BaseMessage implements Request {
+public class PutRequest extends NetworkMessage implements Request {
     @JsonProperty("put")
     private MemoryGraph graph;
 }

src/main/java/io/github/chronosx88/JGUN/network/GatewayNetworkNode.java

@@ -69,4 +69,9 @@ public class GatewayNetworkNode extends WebSocketServer implements Peer {
     public int getTimeout() {
         return 60;
     }
+
+    @Override
+    public int connectedPeerCount() {
+        return this.getConnections().size();
+    }
 }

src/main/java/io/github/chronosx88/JGUN/network/NetworkHandler.java

@@ -7,7 +7,7 @@ import io.github.chronosx88.JGUN.api.FutureGet;
 import io.github.chronosx88.JGUN.api.FuturePut;
 import io.github.chronosx88.JGUN.models.GetResult;
 import io.github.chronosx88.JGUN.models.Result;
-import io.github.chronosx88.JGUN.models.BaseMessage;
+import io.github.chronosx88.JGUN.models.NetworkMessage;
 import io.github.chronosx88.JGUN.models.graph.MemoryGraph;
 import io.github.chronosx88.JGUN.models.graph.Node;
 import io.github.chronosx88.JGUN.models.graph.NodeMetadata;
@@ -52,9 +52,9 @@ public class NetworkHandler {
     }
 
     public void handleIncomingMessage(String message) {
-        BaseMessage parsedMessage;
+        NetworkMessage parsedMessage;
         try {
-            parsedMessage = objectMapper.readValue(message, BaseMessage.class);
+            parsedMessage = objectMapper.readValue(message, NetworkMessage.class);
         } catch (JsonProcessingException e) {
             throw new RuntimeException(e);
         }
@@ -64,9 +64,9 @@ public class NetworkHandler {
             return;
         }
 
-        final BaseMessage msg = parsedMessage;
+        final NetworkMessage msg = parsedMessage;
         executorService.execute(() -> {
-            BaseMessage response = null;
+            NetworkMessage response = null;
             if (msg instanceof GetRequest) {
                 response = handleGet((GetRequest) msg);
             } else if (msg instanceof PutRequest) {

src/main/java/io/github/chronosx88/JGUN/network/NetworkManager.java

@@ -5,6 +5,8 @@ import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.datatype.jdk8.Jdk8Module;
 import io.github.chronosx88.JGUN.api.FutureGet;
 import io.github.chronosx88.JGUN.api.FuturePut;
+import io.github.chronosx88.JGUN.models.GetResult;
+import io.github.chronosx88.JGUN.models.Result;
 import io.github.chronosx88.JGUN.models.graph.MemoryGraph;
 import io.github.chronosx88.JGUN.models.requests.GetRequest;
 import io.github.chronosx88.JGUN.models.requests.GetRequestParams;
@@ -51,23 +53,30 @@ public class NetworkManager {
 
     public FuturePut sendPutRequest(MemoryGraph putData) {
         String id = Dup.random();
+        var requestFuture = new FuturePut(id);
+        if (this.peer.connectedPeerCount() > 0) {
             executorService.execute(() -> this.sendRequest(PutRequest.builder()
                     .id(id)
                     .graph(putData)
                     .build()));
-        var requestFuture = new FuturePut(id);
             peer.addPendingPutRequest(requestFuture);
+        }
+        requestFuture.complete(Result.builder().ok(true).build());
         return requestFuture;
     }
 
     public FutureGet sendGetRequest(GetRequestParams params) {
         String id = Dup.random();
+        var requestFuture = new FutureGet(id, params);
+        if (this.peer.connectedPeerCount() > 0) {
             executorService.execute(() -> this.sendRequest(GetRequest.builder()
                     .id(id)
                     .params(params)
                     .build()));
-        var requestFuture = new FutureGet(id, params);
             peer.addPendingGetRequest(requestFuture);
+        } else {
+            requestFuture.complete(GetResult.builder().data(null).build());
+        }
         return requestFuture;
     }
 }

src/main/java/io/github/chronosx88/JGUN/network/NetworkNode.java

@@ -64,4 +64,10 @@ public class NetworkNode extends WebSocketClient implements Peer {
     public int getTimeout() {
         return 60;
     }
+
+    @Override
+    public int connectedPeerCount() {
+        if (this.isOpen()) return 1;
+        return 0;
+    }
 }

src/main/java/io/github/chronosx88/JGUN/network/Peer.java

@@ -9,4 +9,5 @@ public interface Peer {
     void addPendingGetRequest(FutureGet futureGet);
     void start();
     int getTimeout();
+    int connectedPeerCount();
 }