Merge pull request #5 from cs0320-2021/handler-moving

added part of handler for profit

1 files changed, 19 insertions, 0 deletions

diff --git a/src/main/java/edu/brown/cs/student/term/Main.java b/src/main/java/edu/brown/cs/student/term/Main.java
index 08cadd7..55efbce 100644
--- a/src/main/java/edu/brown/cs/student/term/Main.java
+++ b/src/main/java/edu/brown/cs/student/term/Main.java
@@ -160,6 +160,7 @@ public final class Main {
 
     Spark.before((request, response) -> response.header("Access-Control-Allow-Origin", "*"));
     Spark.post("/data", new DataHandler());
+    Spark.post("/profit", new ProfitQueryHandler());
   }
 
 
@@ -187,6 +188,24 @@ public final class Main {
     }
   }
 
+  private static class ProfitQueryHandler implements Route {
+    @Override
+    public Object handle(Request request, Response response) throws Exception {
+      JSONObject req = new JSONObject(request.body());
+      String person = req.getString("person");
+      Date startPeriod = new Date(req.getLong("startDate"));
+      Date endPeriod = new Date(req.getLong("endDate"));
+
+      try {
+        DatabaseQuerier db = new DatabaseQuerier("data/trades.sqlite3");
+        new ProfitCalculation(DatabaseQuerier.getConn(), "person", startPeriod, endPeriod);
+      } catch (Exception e) {
+        System.out.println("DBQuerier Test, couldn't connect to db???");
+        return "Error";
+      }
+    }
+  }
+
   /**
    * Display an error page when an exception occurs in the server.
    */