catch authentication error with new InvalidAuthenticationException

cursusdb · Dec 24, 2023 · 0735d40 · 0735d40
1 parent f23fec8
commit 0735d40
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/cursusdb.java b/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.");
 

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

diff --git a/test.java b/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);
         }