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 bodyMap, HashMap 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 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();