TechEmpower.FrameworkBenchmarks/frameworks/Java/undertow/src/main/java/hello/QueriesSqlHandler.java

package hello;

import static hello.Helper.getQueries;
import static hello.Helper.randomWorldNumber;
import static hello.Helper.sendJson;

import io.undertow.server.HttpHandler;
import io.undertow.server.HttpServerExchange;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.Objects;
import javax.sql.DataSource;

/**
 * Handles the multi-query database test using a SQL database.
 */
final class QueriesSqlHandler implements HttpHandler {
  private final DataSource db;

  QueriesSqlHandler(DataSource db) {
    this.db = Objects.requireNonNull(db);
  }

  @Override
  public void handleRequest(HttpServerExchange exchange) throws Exception {
    int queries = getQueries(exchange);
    World[] worlds = new World[queries];
    try (Connection connection = db.getConnection();
         PreparedStatement statement =
             connection.prepareStatement("SELECT * FROM World WHERE id = ?")) {
      for (int i = 0; i < worlds.length; i++) {
        statement.setInt(1, randomWorldNumber());
        try (ResultSet resultSet = statement.executeQuery()) {
          resultSet.next();
          int id = resultSet.getInt("id");
          int randomNumber = resultSet.getInt("randomNumber");
          worlds[i] = new World(id, randomNumber);
        }
      }
    }
    sendJson(exchange, worlds);
  }
}