From f8db33bde490e5b3f65e09666a2d493c7b0812ee Mon Sep 17 00:00:00 2001
From: NMerz <nmerz@icloud.com>
Date: Sun, 15 Nov 2020 17:29:37 -0500
Subject: [PATCH] Add chain id list retrieval

Update ChainGetter to work with getListOfIds
---
 Lambdas/Lists/Chain/src/ChainGetter.java | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/Lambdas/Lists/Chain/src/ChainGetter.java b/Lambdas/Lists/Chain/src/ChainGetter.java
index f1775a0..e6cee3b 100644
--- a/Lambdas/Lists/Chain/src/ChainGetter.java
+++ b/Lambdas/Lists/Chain/src/ChainGetter.java
@@ -2,6 +2,7 @@ import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -9,6 +10,7 @@ public class ChainGetter implements CallHandler {
     private final Connection connection;
 
     private final String GET_CHAIN = "SELECT * FROM Chain WHERE chainID = ?;";
+    private final String GET_CHAINS = "SELECT chainID FROM Chain;";
 
     public ChainGetter(Connection connection, String cognitoID) {
         this.connection = connection;
@@ -16,8 +18,21 @@ public class ChainGetter implements CallHandler {
 
     @Override
     public Object conductAction(Map<String, Object> bodyMap, HashMap<String, String> queryMap, String cognitoID) throws SQLException {
+        Integer id = Integer.parseInt(queryMap.get("id"));
+        if (id == -1) {
+            PreparedStatement getChains = connection.prepareStatement(GET_CHAINS);
+            System.out.println(getChains);
+            ResultSet getChainsResults = getChains.executeQuery();
+            System.out.println(getChainsResults);
+            ArrayList<Integer> chainIDs = new ArrayList<>();
+            while (getChainsResults.next()) {
+                chainIDs.add(getChainsResults.getInt("chainID"));
+            }
+            return chainIDs;
+        }
+
         PreparedStatement statement = connection.prepareStatement(GET_CHAIN);
-        statement.setInt(1, Integer.parseInt(queryMap.get("id")));
+        statement.setInt(1, id);
         System.out.println(statement);
         ResultSet queryResults = statement.executeQuery();
         queryResults.first();