catch authentication error with new InvalidAuthenticationException

Author: Alex Gaetano Padula

cursusdb.java

@@ -30,7 +30,6 @@
 
 // CursusDB class
 class CursusDB {
-
     // CursusDB client connection class
     static class Client {
         private String host; // Cluster host
@@ -43,6 +42,14 @@ static class Client {
         private Socket socket; // Socket
         private SSLSocket secureSocket; // Secured socket
 
+
+        public class InvalidAuthenticationException
+                extends RuntimeException {
+            public InvalidAuthenticationException(String errorMessage) {
+                super(errorMessage);
+            }
+        }
+
         // Constructor for CursusDB Client
         Client(String hostIn, int portIn, String usernameIn, String passwordIn, boolean tlsIn) {
             host = hostIn;
@@ -87,6 +94,12 @@ void Connect() throws IOException {
 
                 String clusterResponse = reader.readLine();
 
+                if (clusterResponse.startsWith("0")) {
+
+                } else {
+                    throw new InvalidAuthenticationException("Could not authenticate to cluster");
+                }
+
 
                 System.out.println("Connected to cluster.");
 

readme.md

@@ -15,7 +15,7 @@ package cursusdbjava;
             // Connect
             client.Connect();
 
-        } catch (IOException e) {
+        } catch (IOException | CursusDB.Client.InvalidAuthenticationException e) {
             throw new RuntimeException(e);
         }
 

test.java

@@ -12,7 +12,7 @@ public static void main(String[] args) {
 
             client.Connect();
 
-        } catch (IOException e) {
+        } catch (IOException | CursusDB.Client.InvalidAuthenticationException e) {
             throw new RuntimeException(e);
         }