-

src/transceptor/technology/Client.java

@@ -3,15 +3,15 @@
 import java.nio.channels.CompletionHandler;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Map;
 import java.util.Random;
 import java.util.stream.Collectors;
-import java.util.stream.Stream;
 
 /**
  *
  * @author tristan
  */
-public class Client {
+public class Client implements ConnectionInterface {
 
     private final List<Connection> connectionList;
     private final List<Connection> connectionPriorityList;
@@ -43,6 +43,7 @@ Connection randomConnection() {
                 .get(rand.nextInt(l.size()));
     }
 
+    @Override
     public void connect(CompletionHandler handler) {
         for (String[] strings : hostlist) {
             Connection connection = new Connection(username, password, dbname, strings[0], Integer.parseInt(strings[1]));
@@ -57,15 +58,23 @@ public void connect(CompletionHandler handler) {
             }
         }
     }
+
+    @Override
+    public void insert(Map map, CompletionHandler handler) {
+        randomConnection().insert(map, handler);
+    }
 
+    @Override
     public void query(String query, CompletionHandler handler) {
         randomConnection().query(query, handler);
     }
 
+    @Override
     public void query(String query, int timePrecision, CompletionHandler handler) {
         randomConnection().query(query, timePrecision, handler);
     }
 
+    @Override
     public void close() {
         connectionList.forEach((connection) -> {
             connection.close();

src/transceptor/technology/Connection.java

@@ -5,7 +5,6 @@
 import java.nio.ByteBuffer;
 import java.nio.channels.AsynchronousSocketChannel;
 import java.nio.channels.CompletionHandler;
-import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.concurrent.ExecutionException;
@@ -18,7 +17,7 @@
  *
  * @author Tristan Nottelman
  */
-public class Connection {
+public class Connection implements ConnectionInterface {
 
     private AsynchronousSocketChannel channel;
     private String username;
@@ -177,6 +176,7 @@ public void failed(Throwable exc, Object attachment) {
      *
      * @param handler
      */
+    @Override
     public void connect(CompletionHandler handler) {
         try {
             // connect to SiriDB
@@ -202,6 +202,7 @@ public void connect(CompletionHandler handler) {
     /**
      * Closes the connection
      */
+    @Override
     public void close() {
         try {
             channel.close();
@@ -216,6 +217,7 @@ public void close() {
      * @param map
      * @param handler
      */
+    @Override
     public void insert(Map map, CompletionHandler handler) {
         channelWriter((byte) CPROTO_RES_INSERT, qpack.pack(map), handler);
     }
@@ -226,6 +228,7 @@ public void insert(Map map, CompletionHandler handler) {
      * @param query
      * @param handler
      */
+    @Override
     public void query(String query, CompletionHandler handler) {
         channelWriter((byte) CPROTO_REQ_QUERY, qpack.pack(new String[]{query, null}), handler);
     }
@@ -237,6 +240,7 @@ public void query(String query, CompletionHandler handler) {
      * @param timePrecision
      * @param handler
      */
+    @Override
     public void query(String query, int timePrecision, CompletionHandler handler) {
         channelWriter((byte) CPROTO_REQ_QUERY, qpack.pack(new String[]{query, timePrecision + ""}), handler);
     }

src/transceptor/technology/ConnectionInterface.java

@@ -0,0 +1,16 @@
+package transceptor.technology;
+
+import java.nio.channels.CompletionHandler;
+import java.util.Map;
+
+/**
+ *
+ * @author tristan
+ */
+public interface ConnectionInterface {
+    public void close();
+    public void connect(CompletionHandler handler);
+    public void insert(Map map, CompletionHandler handler);
+    public void query(String query, CompletionHandler handler);
+    public void query(String query, int timePrecision, CompletionHandler handler);
+}