T138861426: [Bootcamp][PCF][BE][1/2] Migrate the DotproductGame to use the new API that accepts metric collector. (#1967)

Summary:
Pull Request resolved: #1967

X-link: facebookresearch/fbpcf#453

As titled.

Reviewed By: xyguo

Differential Revision: D41640581

fbshipit-source-id: d55b320cea233003d36fe52171d2c4a6cab58014

Author: slaxman1978

fbpcs/emp_games/dotproduct/DotproductApp.h

@@ -51,7 +51,9 @@ class DotproductApp {
                          ->create();
 
     DotproductGame<schedulerId> game(
-        std::move(scheduler), std::move(communicationAgentFactory_));
+        std::move(scheduler),
+        std::move(communicationAgentFactory_),
+        metricCollector_);
 
     XLOG(INFO) << "Start Reading input file ";
     auto inputTuple = readCSVInput(inputFilePath_, labelWidth_, numFeatures_);

fbpcs/emp_games/dotproduct/DotproductGame.h

@@ -9,6 +9,7 @@
 
 #include "folly/logging/xlog.h"
 
+#include <fbpcf/util/MetricCollector.h>
 #include "fbpcf/frontend/mpcGame.h"
 #include "fbpcs/emp_games/common/Debug.h"
 #include "fbpcs/emp_games/common/Util.h"
@@ -23,9 +24,11 @@ class DotproductGame : public fbpcf::frontend::MpcGame<schedulerId> {
       std::unique_ptr<fbpcf::scheduler::IScheduler> scheduler,
       std::shared_ptr<
           fbpcf::engine::communication::IPartyCommunicationAgentFactory>
-          communicationAgentFactory)
+          communicationAgentFactory,
+      std::shared_ptr<fbpcf::util::MetricCollector> metricCollector)
       : fbpcf::frontend::MpcGame<schedulerId>(std::move(scheduler)),
-        communicationAgentFactory_(communicationAgentFactory) {}
+        communicationAgentFactory_(communicationAgentFactory),
+        metricCollector_{metricCollector} {}
 
   std::vector<double> computeDotProduct(
       const int myRole,
@@ -44,6 +47,8 @@ class DotproductGame : public fbpcf::frontend::MpcGame<schedulerId> {
   std::shared_ptr<fbpcf::engine::communication::IPartyCommunicationAgentFactory>
       communicationAgentFactory_;
 
+  std::shared_ptr<fbpcf::util::MetricCollector> metricCollector_;
+
   std::vector<fbpcf::frontend::Bit<true, schedulerId, true>>
   createSecretLabelShare(const std::vector<std::vector<bool>>& labelValues);
 

fbpcs/emp_games/dotproduct/DotproductGame_impl.h

@@ -95,7 +95,8 @@ std::vector<double> DotproductGame<schedulerId>::computeDotProduct(
         divisor,
         *communicationAgentFactory_,
         std::move(prgFactory),
-        std::move(cotWRMFactory));
+        std::move(cotWRMFactory),
+        metricCollector_);
 
     XLOG(INFO, "Created Matrix Multiplication Factory");
 
@@ -117,7 +118,8 @@ std::vector<double> DotproductGame<schedulerId>::computeDotProduct(
         divisor,
         *communicationAgentFactory_,
         std::move(prgFactory),
-        std::move(cotWRMFactory));
+        std::move(cotWRMFactory),
+        metricCollector_);
     XLOG(INFO, "Created Matrix Multiplication Factory");
 
     matMulFactoryPartner->create()->matrixVectorMultiplication(

fbpcs/emp_games/dotproduct/test/DotproductGameTest.cpp

@@ -36,10 +36,12 @@ class MockDotProductGame : public DotproductGame<schedulerId> {
       std::unique_ptr<fbpcf::scheduler::IScheduler> scheduler,
       std::shared_ptr<
           fbpcf::engine::communication::IPartyCommunicationAgentFactory>
-          communicationAgentFactory)
+          communicationAgentFactory,
+      std::shared_ptr<fbpcf::util::MetricCollector> metricCollector)
       : DotproductGame<schedulerId>(
             std::move(scheduler),
-            std::move(communicationAgentFactory)) {}
+            std::move(communicationAgentFactory),
+            metricCollector) {}
 
   MOCK_METHOD(
       std::vector<double>,
@@ -55,8 +57,11 @@ std::vector<bool> runORLabelsGame(
     fbpcf::SchedulerCreator schedulerCreator,
     std::vector<std::vector<bool>> labels) {
   auto scheduler = schedulerCreator(PARTY, *factory);
+  auto metricCollector =
+      std::make_shared<fbpcf::util::MetricCollector>("dotproduct_test");
 
-  DotproductGame<schedulerId> game(std::move(scheduler), std::move(factory));
+  DotproductGame<schedulerId> game(
+      std::move(scheduler), std::move(factory), metricCollector);
 
   // Create label secret shares
   auto labelShare = game.createSecretLabelShare(labels);
@@ -84,9 +89,11 @@ std::vector<double> runGame(
     std::vector<double> dpNoise) {
   auto scheduler = schedulerCreator(PARTY, *factory);
 
+  auto metricCollector =
+      std::make_shared<fbpcf::util::MetricCollector>("dotproduct_test");
   // create a mock Dotproduct Game
   MockDotProductGame<schedulerId> mockGame(
-      std::move(scheduler), std::move(factory));
+      std::move(scheduler), std::move(factory), metricCollector);
 
   // mock the dpNoise generation in DotproductGame
   ON_CALL(mockGame, generateDpNoise(numFeatures, delta, eps, addDpNoise))

fbpcs/emp_games/dotproduct/test/test_correctness/outpub_dotprodtest_0.csv

@@ -0,0 +1 @@
+[0,0,0,0,0,0,0,0.001448442,0,0,0.06863192,0.14435472,0.033341374,0,0.12865546,0.020235294,0.14462507,0,0.17825273,0,0,0,0.46016255,0,0,0.128087239,0.4376182,0.000390805,0.033707865,0.13740458,0,0.000504521,0.07516815,0.03580519,1,0,0.46016255,0.050997782,0.46016258,0,0.002447232,0,0.8,0,0,0.042172838,0.505038259,0,1,0.1516517]