From 7c8295ee82ceaf79b34ab6eb52014fed14f3c561 Mon Sep 17 00:00:00 2001
From: gojomo <gojogit@gmail.com>
Date: Tue, 12 Mar 2013 11:42:29 -0700
Subject: [PATCH] checkpoint support

---
 .../modules/forms/FormLoginProcessor.java     | 25 ++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/modules/src/main/java/org/archive/modules/forms/FormLoginProcessor.java b/modules/src/main/java/org/archive/modules/forms/FormLoginProcessor.java
index aa642efb..e9f1be04 100644
--- a/modules/src/main/java/org/archive/modules/forms/FormLoginProcessor.java
+++ b/modules/src/main/java/org/archive/modules/forms/FormLoginProcessor.java
@@ -28,6 +28,7 @@ import java.util.logging.Logger;
 
 import org.apache.commons.httpclient.URIException;
 import org.apache.commons.lang.StringUtils;
+import org.archive.checkpointing.Checkpointable;
 import org.archive.modules.CoreAttributeConstants;
 import org.archive.modules.CrawlURI;
 import org.archive.modules.CrawlURI.FetchType;
@@ -36,6 +37,9 @@ import org.archive.modules.SchedulingConstants;
 import org.archive.modules.extractor.Hop;
 import org.archive.modules.extractor.LinkContext;
 import org.archive.modules.extractor.UriErrorLoggerModule;
+import org.archive.util.JSONUtils;
+import org.json.JSONException;
+import org.json.JSONObject;
 import org.springframework.beans.factory.annotation.Autowired;
 
 import com.google.common.base.Function;
@@ -108,7 +112,7 @@ import com.google.common.collect.MapMaker;
  * @author gojomo
  * @version $Date$, $Revision$
  */
-public class FormLoginProcessor extends Processor {
+public class FormLoginProcessor extends Processor implements Checkpointable {
     @SuppressWarnings("unused")
     private static final long serialVersionUID = -1L;
     private static final Logger logger =
@@ -276,4 +280,23 @@ public class FormLoginProcessor extends Processor {
                 +","+eligibleFormsSeenCount.get(formProvince).get()
                 +","+formProvince;
     }
+    
+    @Override
+    protected JSONObject toCheckpointJson() throws JSONException {
+        JSONObject json = super.toCheckpointJson();
+        json.put("eligibleFormsAttemptsCount", eligibleFormsAttemptsCount);
+        json.put("eligibleFormsSeenCount", eligibleFormsSeenCount);
+        return json;
+    }
+    
+    @Override
+    protected void fromCheckpointJson(JSONObject json) throws JSONException {
+        super.fromCheckpointJson(json);
+        JSONUtils.putAllAtomicLongs(
+                eligibleFormsAttemptsCount,
+                json.getJSONObject("eligibleFormsAttemptsCount"));
+        JSONUtils.putAllAtomicLongs(
+                eligibleFormsSeenCount,
+                json.getJSONObject("eligibleFormsSeenCount"));
+    }
 }