package hello;

import java.io.*;
import java.sql.*;
import java.util.*;
import java.util.concurrent.*;

import javax.annotation.*;
import javax.servlet.*;
import javax.servlet.http.*;
import javax.sql.*;

/**
 * Database connectivity (with a Servlet-container managed pool) test.
 */
@SuppressWarnings("serial")
public class DbPoolServlet extends HttpServlet
{
  // Database details.
  private static final String DB_QUERY = "SELECT * FROM World WHERE id = ?";
  private static final int    DB_ROWS  = 10000;

  // Database connection pool.
  @Resource(name="jdbc/hello_world")
  private DataSource mysqlDataSource;
    
  @Override
  protected void doGet(HttpServletRequest req, HttpServletResponse res)
      throws ServletException, IOException
  {
    // Set content type to JSON
    res.setHeader(Common.HEADER_CONTENT_TYPE, Common.CONTENT_TYPE_JSON);

    // Reference the data source.
    final DataSource source = mysqlDataSource;
    
    // Get the count of queries to run.
    int count = 1;
    try
    {
      count = Integer.parseInt(req.getParameter("queries"));
      
      // Bounds check.
      if (count > 500)
      {
        count = 500;
      }
      if (count < 1)
      {
        count = 1;
      }
    }
    catch (NumberFormatException nfexc)
    {
      // Do nothing.
    }

    // Fetch some rows from the database.
    final World[] worlds = new World[count];
    final Random random = ThreadLocalRandom.current();
    
    try (Connection conn = source.getConnection())
    {
      try (PreparedStatement statement = conn.prepareStatement(DB_QUERY, 
          ResultSet.TYPE_FORWARD_ONLY, ResultSet.CONCUR_READ_ONLY))
      {
        // Run the query the number of times requested.
        for (int i = 0; i < count; i++)
        {
          final int id = random.nextInt(DB_ROWS) + 1;
          statement.setInt(1, id);
          
          try (ResultSet results = statement.executeQuery())
          {
            if (results.next())
            {
              worlds[i] = new World(id, results.getInt("randomNumber"));
            }
          }
        }
      }
    }
    catch (SQLException sqlex)
    {
      System.err.println("SQL Exception: " + sqlex);
    }
    
    // Write JSON encoded message to the response.
    try
    {
      Common.MAPPER.writeValue(res.getOutputStream(), worlds);
    }
    catch (IOException ioe) 
    {
      // do nothing
    }
  }
}

@@ -2262,24 +2262,152 @@
   try%0A    %7B%0A
+      if (count == 1)%0A      %7B%0A        Common.MAPPER.writeValue(res.getOutputStream(), worlds%5B0%5D); %0A      %7D%0A      else%0A      %7B%0A  
       Common
@@ -2457,16 +2457,24 @@
 orlds);%0A
+      %7D%0A
     %7D%0A  

verify Fixed servlet-raw's db test