[tor-commits] [metrics-web/master] Let users download votes from a given valid-after time.

[karsten at torproject.org]

commit b7f0ba813e9468c1635457a513a77496d62aed23
Author: Karsten Loesing <karsten.loesing at gmx.net>
Date:   Wed Jul 13 13:12:41 2011 +0200

    Let users download votes from a given valid-after time.
---
 ChangeLog                                      |    1 +
 etc/web.xml                                    |   11 +++
 src/org/torproject/ernie/web/VotesServlet.java |  110 ++++++++++++++++++++++++
 3 files changed, 122 insertions(+), 0 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 94884cf..387e813 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,6 +1,7 @@
 Metrics database and website change log:
 
 Changes in version 0.0.2 - 2011-??-??
+  - Let users download votes from a given valid-after time.
 
 Changes in version 0.0.1 - 2011-06-08
   - Initial release
diff --git a/etc/web.xml b/etc/web.xml
index 688e6f3..dbaa0a2 100644
--- a/etc/web.xml
+++ b/etc/web.xml
@@ -299,6 +299,17 @@
   </servlet-mapping>
 
   <servlet>
+    <servlet-name>Votes</servlet-name>
+    <servlet-class>
+      org.torproject.ernie.web.VotesServlet
+    </servlet-class>
+  </servlet>
+  <servlet-mapping>
+    <servlet-name>Votes</servlet-name>
+    <url-pattern>/votes</url-pattern>
+  </servlet-mapping>
+
+  <servlet>
     <servlet-name>ErnieGeneratedFile</servlet-name>
     <servlet-class>
       org.torproject.ernie.web.ErnieGeneratedFileServlet
diff --git a/src/org/torproject/ernie/web/VotesServlet.java b/src/org/torproject/ernie/web/VotesServlet.java
new file mode 100644
index 0000000..3af9705
--- /dev/null
+++ b/src/org/torproject/ernie/web/VotesServlet.java
@@ -0,0 +1,110 @@
+package org.torproject.ernie.web;
+
+import java.io.*;
+import java.sql.*;
+import java.text.*;
+import java.util.*;
+import java.util.logging.*;
+
+import javax.naming.*;
+import javax.servlet.*;
+import javax.servlet.http.*;
+import javax.sql.*;
+
+public class VotesServlet extends HttpServlet {
+
+  private DataSource ds;
+
+  private Logger logger;
+
+  public void init() {
+
+    /* Initialize logger. */
+    this.logger = Logger.getLogger(VotesServlet.class.toString());
+
+    /* Look up data source. */
+    try {
+      Context cxt = new InitialContext();
+      this.ds = (DataSource) cxt.lookup("java:comp/env/jdbc/tordir");
+      this.logger.info("Successfully looked up data source.");
+    } catch (NamingException e) {
+      this.logger.log(Level.WARNING, "Could not look up data source", e);
+    }
+  }
+
+  public void doGet(HttpServletRequest request,
+      HttpServletResponse response) throws IOException,
+      ServletException {
+
+    /* Check valid-after parameter. */
+    String validAfterParameter = request.getParameter("valid-after");
+    if (validAfterParameter == null ||
+        validAfterParameter.length() != "yyyy-MM-dd-HH-mm-ss".length()) {
+      response.sendError(HttpServletResponse.SC_BAD_REQUEST);
+      return;
+    }
+    SimpleDateFormat parameterFormat = new SimpleDateFormat(
+        "yyyy-MM-dd-HH-mm-ss");
+    parameterFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
+    long parsedTimestamp = -1L;
+    try {
+      parsedTimestamp = parameterFormat.parse(validAfterParameter).
+          getTime();
+    } catch (ParseException e) {
+      response.sendError(HttpServletResponse.SC_BAD_REQUEST);
+      return;
+    }
+    if (parsedTimestamp < 0L) {
+      response.sendError(HttpServletResponse.SC_BAD_REQUEST);
+      return;
+    }
+
+    /* Look up consensus in the database. */
+    SimpleDateFormat databaseFormat = new SimpleDateFormat(
+        "yyyy-MM-dd HH:mm:ss");
+    databaseFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
+    String databaseParameter = databaseFormat.format(parsedTimestamp);
+    List<byte[]> rawDescriptors = new ArrayList<byte[]>();
+    try {
+      Connection conn = this.ds.getConnection();
+      Statement statement = conn.createStatement();
+      String query = "SELECT rawdesc FROM vote "
+          + "WHERE validafter = '" + databaseParameter + "'";
+      ResultSet rs = statement.executeQuery(query);
+      while (rs.next()) {
+        rawDescriptors.add(rs.getBytes(1));
+      }
+      rs.close();
+      statement.close();
+      conn.close();
+    } catch (SQLException e) {
+      response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+      return;
+    }
+
+    /* Write response. */
+    if (rawDescriptors.size() == 0) {
+      response.sendError(HttpServletResponse.SC_NOT_FOUND);
+      return;
+    }
+    try {
+      response.setContentType("text/plain");
+      int responseLength = 0;
+      for (byte[] rawDescriptor : rawDescriptors) {
+        responseLength += rawDescriptor.length;
+      }
+      response.setHeader("Content-Length", String.valueOf(
+          responseLength));
+      response.setHeader("Content-Disposition", "inline; filename=\""
+          + validAfterParameter + "-votes\"");
+      BufferedOutputStream output = new BufferedOutputStream(
+          response.getOutputStream());
+      for (byte[] rawDescriptor : rawDescriptors) {
+        output.write(rawDescriptor);
+      }
+    } finally {
+      /* Nothing to do here. */
+    }
+  }
+}
+