Adding updateIndividualProfile API to MonitoredCCBAPI

1 files changed, 19 insertions, 6 deletions

diff --git a/src/com/p4square/grow/ccb/MonitoredCCBAPI.java b/src/com/p4square/grow/ccb/MonitoredCCBAPI.java
index 6903460..1463c77 100644
--- a/src/com/p4square/grow/ccb/MonitoredCCBAPI.java
+++ b/src/com/p4square/grow/ccb/MonitoredCCBAPI.java
@@ -3,9 +3,7 @@ package com.p4square.grow.ccb;
 import com.codahale.metrics.MetricRegistry;
 import com.codahale.metrics.Timer;
 import com.p4square.ccbapi.CCBAPI;
-import com.p4square.ccbapi.model.GetCustomFieldLabelsResponse;
-import com.p4square.ccbapi.model.GetIndividualProfilesRequest;
-import com.p4square.ccbapi.model.GetIndividualProfilesResponse;
+import com.p4square.ccbapi.model.*;
 
 import java.io.IOException;
 
@@ -51,13 +49,28 @@ public class MonitoredCCBAPI implements CCBAPI {
         boolean success = false;
         try {
             final GetIndividualProfilesResponse resp = mAPI.getIndividualProfiles(request);
-            mMetricRegistry.counter("CCBAPI.getCustomFieldLabels.count").inc(resp.getIndividuals().size());
+            mMetricRegistry.counter("CCBAPI.getIndividualProfiles.count").inc(resp.getIndividuals().size());
             success = true;
             return resp;
         } finally {
             timer.stop();
-            mMetricRegistry.counter("CCBAPI.getCustomFieldLabels.success").inc(success ? 1 : 0);
-            mMetricRegistry.counter("CCBAPI.getCustomFieldLabels.failure").inc(!success ? 1 : 0);
+            mMetricRegistry.counter("CCBAPI.getIndividualProfiles.success").inc(success ? 1 : 0);
+            mMetricRegistry.counter("CCBAPI.getIndividualProfiles.failure").inc(!success ? 1 : 0);
+        }
+    }
+
+    @Override
+    public UpdateIndividualProfileResponse updateIndividualProfile(UpdateIndividualProfileRequest request) throws IOException {
+        final Timer.Context timer = mMetricRegistry.timer("CCBAPI.updateIndividualProfile").time();
+        boolean success = false;
+        try {
+            final UpdateIndividualProfileResponse resp = mAPI.updateIndividualProfile(request);
+            success = true;
+            return resp;
+        } finally {
+            timer.stop();
+            mMetricRegistry.counter("CCBAPI.updateIndividualProfile.success").inc(success ? 1 : 0);
+            mMetricRegistry.counter("CCBAPI.updateIndividualProfile.failure").inc(!success ? 1 : 0);
         }
     }