grapesy-1.0.0: interop/Interop/Server/TestService/StreamingOutputCall.hs

module Interop.Server.TestService.StreamingOutputCall (
    handle
  , handleRequest
  ) where

import Control.Concurrent
import Control.Monad

import Network.GRPC.Common
import Network.GRPC.Common.Protobuf
import Network.GRPC.Server

import Interop.Util.Messages

import Proto.API.Interop

-- | Handle @TestService.StreamingOutputCall@
--
-- <https://github.com/grpc/grpc/blob/master/doc/interop-test-descriptions.md#streamingoutputcall>
handle :: Call StreamingOutputCall -> IO ()
handle call = do
    request <- recvFinalInput call
    handleRequest call request
    sendTrailers call def

-- | Handle specific request
--
-- Abstracted out because also used in the @fullDuplexCall@ test.
handleRequest :: forall rpc.
     (Output rpc ~ Proto StreamingOutputCallResponse)
  => Call rpc
  -> Proto StreamingOutputCallRequest
  -> IO ()
handleRequest call request =
    forM_ (request ^. #responseParameters) $ \responseParameters -> do
      let size, intervalUs :: Int
          size       = fromIntegral $ responseParameters ^. #size
          intervalUs = fromIntegral $ responseParameters ^. #intervalUs

          shouldCompress :: Bool
          shouldCompress = responseParameters ^. #compressed ^. #value

      payload <- payloadOfType (Proto COMPRESSABLE) size

      let meta :: OutboundMeta
          meta = def { outboundEnableCompression = shouldCompress }

          response :: Proto StreamingOutputCallResponse
          response = defMessage & #payload .~ payload

      sendOutputWithMeta call $ StreamElem (meta, response)
      threadDelay intervalUs