Create download-batch target

Summary: Create a separate `DownloadBatch.hs` module for internal and oss build (as internal one will download batch from cas)

Reviewed By: donsbot

Differential Revision: D69668663

fbshipit-source-id: e5c2af667ea7f17091f0a88178a422eb1988ad87

glean.cabal.in

@@ -194,6 +194,7 @@ library stubs
         Glean.BuildInfo
         Glean.Datasource.Scribe.Write
         Glean.Init
+        Glean.Server.DownloadBatch
         Glean.Server.PublishShards
         Glean.Index
         Glean.Username

glean/db/Glean/Database/Writes.hs

@@ -80,6 +80,7 @@ import Glean.Database.Types
 import qualified Glean.RTS.Foreign.Subst as Subst
 import Glean.RTS.Foreign.Ownership (DefineOwnership)
 import qualified Glean.ServerConfig.Types as ServerConfig
+import Glean.Server.DownloadBatch
 import Glean.Types as Thrift hiding (Database)
 import Glean.Write.JSON
 import Glean.Util.Metric
@@ -377,11 +378,11 @@ enqueueBatchDescriptor env repo enqueueBatch waitPolicy = do
   traceMsg (envTracer env)
     (GleanTraceEnqueue repo EnqueueBatchDescriptor 0) $ do
   handle <- UUID.toText <$> UUID.nextRandom
-  _ <- case enqueueBatch of
+  descriptor <- case enqueueBatch of
     Thrift.EnqueueBatch_descriptor descriptor -> return descriptor
     Thrift.EnqueueBatch_EMPTY -> throwIO $ Thrift.Exception "empty batch"
   write <- enqueueWrite env repo 0 Nothing False $
-    throwIO $ Exception "not implemented"
+    writeContentFromBatch <$> downloadBatch descriptor
   when (waitPolicy == Thrift.EnqueueBatchWaitPolicy_Remember)
     $ rememberWrite env handle write
   return $ def { enqueueBatchResponse_handle = handle }

glean/github/Glean/Server/DownloadBatch.hs

@@ -0,0 +1,18 @@
+{-
+  Copyright (c) Meta Platforms, Inc. and affiliates.
+  All rights reserved.
+
+  This source code is licensed under the BSD-style license found in the
+  LICENSE file in the root directory of this source tree.
+-}
+
+module Glean.Server.DownloadBatch
+  ( downloadBatch
+  ) where
+
+import Glean.Types as Thrift
+import Control.Exception
+
+downloadBatch :: Thrift.BatchDescriptor -> IO Thrift.Batch
+downloadBatch _ = throwIO $
+  Thrift.Exception "couldn't download batch from location"