modern sync: add upload_trees method

Summary: Unused for now, this is prep for sending hg data

Reviewed By: andreacampi

Differential Revision: D67006514

fbshipit-source-id: a554ca3523aff387f8e88e6fe7d32a4970870b3e

eden/mononoke/modern_sync/src/sender.rs

@@ -7,6 +7,7 @@
 
 use anyhow::Result;
 use async_trait::async_trait;
+use edenapi_types::UploadTreeEntry;
 use mononoke_types::ContentId;
 use mononoke_types::FileContents;
 pub mod dummy;
@@ -15,4 +16,7 @@ pub mod edenapi;
 #[async_trait]
 pub trait ModernSyncSender {
     async fn upload_content(&self, content_id: ContentId, _blob: FileContents) -> Result<()>;
+
+    #[allow(unused)]
+    async fn upload_tree(&self, trees: Vec<UploadTreeEntry>) -> Result<()>;
 }

eden/mononoke/modern_sync/src/sender/dummy.rs

@@ -7,6 +7,7 @@
 
 use anyhow::Result;
 use async_trait::async_trait;
+use edenapi_types::UploadTreeEntry;
 use mononoke_types::ContentId;
 use mononoke_types::FileContents;
 use slog::info;
@@ -31,4 +32,11 @@ impl ModernSyncSender for DummySender {
         info!(&self.logger, "Uploading content with id: {:?}", content_id);
         Ok(())
     }
+
+    async fn upload_tree(&self, trees: Vec<UploadTreeEntry>) -> Result<()> {
+        for tree in trees {
+            info!(&self.logger, "Uploading tree with id {:?}", tree.node_id);
+        }
+        Ok(())
+    }
 }

eden/mononoke/modern_sync/src/sender/edenapi.rs

@@ -16,6 +16,7 @@ use edenapi::HttpClientBuilder;
 use edenapi::HttpClientConfig;
 use edenapi::SaplingRemoteApi;
 use edenapi_types::AnyFileContentId;
+use edenapi_types::UploadTreeEntry;
 use futures::TryStreamExt;
 use mononoke_app::args::TLSArgs;
 use mononoke_types::FileContents;
@@ -99,4 +100,14 @@ impl ModernSyncSender for EdenapiSender {
 
         Ok(())
     }
+
+    async fn upload_tree(&self, trees: Vec<UploadTreeEntry>) -> Result<()> {
+        let res = self.client.upload_trees_batch(trees).await?;
+        info!(
+            &self.logger,
+            "Upload tree response: {:?}",
+            res.entries.try_collect::<Vec<_>>().await?
+        );
+        Ok(())
+    }
 }