Merge pull request #119 from ClaytonWWilson/chain-get-ids

Add chain id list retrieval
This commit is contained in:
Nathan Merz 2020-11-15 17:29:52 -05:00 committed by GitHub
commit 254f771e6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,6 +2,7 @@ import java.sql.Connection;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
@ -9,6 +10,7 @@ public class ChainGetter implements CallHandler {
private final Connection connection; private final Connection connection;
private final String GET_CHAIN = "SELECT * FROM Chain WHERE chainID = ?;"; 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) { public ChainGetter(Connection connection, String cognitoID) {
this.connection = connection; this.connection = connection;
@ -16,8 +18,21 @@ public class ChainGetter implements CallHandler {
@Override @Override
public Object conductAction(Map<String, Object> bodyMap, HashMap<String, String> queryMap, String cognitoID) throws SQLException { 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); PreparedStatement statement = connection.prepareStatement(GET_CHAIN);
statement.setInt(1, Integer.parseInt(queryMap.get("id"))); statement.setInt(1, id);
System.out.println(statement); System.out.println(statement);
ResultSet queryResults = statement.executeQuery(); ResultSet queryResults = statement.executeQuery();
queryResults.first(); queryResults.first();