{-# LANGUAGE DataKinds       #-}
{-# LANGUAGE RecordWildCards #-}
--------------------------------------------------------------------------------
-- |
-- Module : Database.EventStore.Internal.Operation.ReadStreamEvents
-- Copyright : (C) 2015 Yorick Laupa
-- License : (see the file LICENSE)
--
-- Maintainer : Yorick Laupa <yo.eight@gmail.com>
-- Stability : provisional
-- Portability : non-portable
--
--------------------------------------------------------------------------------
module Database.EventStore.Internal.Operation.ReadStreamEvents
    ( readStreamEvents ) where

--------------------------------------------------------------------------------
import Data.Int

--------------------------------------------------------------------------------
import Data.ProtocolBuffers

--------------------------------------------------------------------------------
import Database.EventStore.Internal.Command
import Database.EventStore.Internal.Control (publishWith)
import Database.EventStore.Internal.Communication (Transmit(..))
import Database.EventStore.Internal.Exec (Exec)
import Database.EventStore.Internal.Operation
import Database.EventStore.Internal.Operation.Read.Common
import Database.EventStore.Internal.Operation.ReadStreamEvents.Message
import Database.EventStore.Internal.Prelude
import Database.EventStore.Internal.Settings
import Database.EventStore.Internal.Stream
import Database.EventStore.Internal.Types

--------------------------------------------------------------------------------
-- | Batch read from a regular stream operation.
readStreamEvents
  :: Settings
  -> Exec
  -> ReadDirection
  -> Text
  -> Int64
  -> Int32
  -> Bool
  -> Maybe Credentials
  -> IO (Async (ReadResult EventNumber StreamSlice))
readStreamEvents :: Settings
-> Exec
-> ReadDirection
-> Text
-> Int64
-> Int32
-> Bool
-> Maybe Credentials
-> IO (Async (ReadResult EventNumber StreamSlice))
readStreamEvents Settings{Bool
Maybe Text
Maybe TLSSettings
Maybe Credentials
LogType
NominalDiffTime
LoggerFilter
MonitoringBackend
Retry
s_defaultUserCredentials :: Settings -> Maybe Credentials
s_defaultConnectionName :: Settings -> Maybe Text
s_monitoring :: Settings -> MonitoringBackend
s_operationRetry :: Settings -> Retry
s_operationTimeout :: Settings -> NominalDiffTime
s_loggerDetailed :: Settings -> Bool
s_loggerFilter :: Settings -> LoggerFilter
s_loggerType :: Settings -> LogType
s_ssl :: Settings -> Maybe TLSSettings
s_reconnect_delay :: Settings -> NominalDiffTime
s_retry :: Settings -> Retry
s_requireMaster :: Settings -> Bool
s_heartbeatTimeout :: Settings -> NominalDiffTime
s_heartbeatInterval :: Settings -> NominalDiffTime
s_defaultUserCredentials :: Maybe Credentials
s_defaultConnectionName :: Maybe Text
s_monitoring :: MonitoringBackend
s_operationRetry :: Retry
s_operationTimeout :: NominalDiffTime
s_loggerDetailed :: Bool
s_loggerFilter :: LoggerFilter
s_loggerType :: LogType
s_ssl :: Maybe TLSSettings
s_reconnect_delay :: NominalDiffTime
s_retry :: Retry
s_requireMaster :: Bool
s_heartbeatTimeout :: NominalDiffTime
s_heartbeatInterval :: NominalDiffTime
..} Exec
exec ReadDirection
dir Text
stream Int64
st Int32
cnt Bool
tos Maybe Credentials
cred
  = do Mailbox
m <- IO Mailbox
forall (m :: * -> *). MonadBase IO m => m Mailbox
mailboxNew
       IO (ReadResult EventNumber StreamSlice)
-> IO (Async (StM IO (ReadResult EventNumber StreamSlice)))
forall (m :: * -> *) a.
MonadBaseControl IO m =>
m a -> m (Async (StM m a))
async (IO (ReadResult EventNumber StreamSlice)
 -> IO (Async (StM IO (ReadResult EventNumber StreamSlice))))
-> IO (ReadResult EventNumber StreamSlice)
-> IO (Async (StM IO (ReadResult EventNumber StreamSlice)))
forall a b. (a -> b) -> a -> b
$
         do let reqCmd :: Command
reqCmd =
                  case ReadDirection
dir of
                    ReadDirection
Forward  -> Command
readStreamEventsForwardCmd
                    ReadDirection
Backward -> Command
readStreamEventsBackwardCmd

                req :: Request
req = Text -> Int64 -> Int32 -> Bool -> Bool -> Request
newRequest Text
stream Int64
st Int32
cnt Bool
tos Bool
s_requireMaster
            Package
pkg <- Command -> Maybe Credentials -> Request -> IO Package
forall msg (m :: * -> *).
(Encode msg, MonadIO m) =>
Command -> Maybe Credentials -> msg -> m Package
createPkg Command
reqCmd Maybe Credentials
cred Request
req
            Exec -> Transmit -> IO ()
forall p a (m :: * -> *).
(Pub p, Typeable a, MonadIO m) =>
p -> a -> m ()
publishWith Exec
exec (Mailbox -> Lifetime -> Package -> Transmit
Transmit Mailbox
m Lifetime
OneTime Package
pkg)
            Either OperationError Response
outcome <- Mailbox -> IO (Either OperationError Response)
forall (m :: * -> *) resp.
(MonadBase IO m, Decode resp) =>
Mailbox -> m (Either OperationError resp)
mailboxReadDecoded Mailbox
m
            case Either OperationError Response
outcome of
              Left OperationError
e
                -> OperationError -> IO (ReadResult EventNumber StreamSlice)
forall (m :: * -> *) e a. (MonadThrow m, Exception e) => e -> m a
throw OperationError
e
              Right Response
resp
                -> let r :: FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
r     = Field 2 (RequiredField (Always (Enumeration Result)))
-> FieldType
     (Field 2 (RequiredField (Always (Enumeration Result))))
forall a. HasField a => a -> FieldType a
getField (Field 2 (RequiredField (Always (Enumeration Result)))
 -> FieldType
      (Field 2 (RequiredField (Always (Enumeration Result)))))
-> Field 2 (RequiredField (Always (Enumeration Result)))
-> FieldType
     (Field 2 (RequiredField (Always (Enumeration Result))))
forall a b. (a -> b) -> a -> b
$ Response -> Required 2 (Enumeration Result)
_result Response
resp
                       es :: FieldType (Repeated 1 (Message ResolvedIndexedEvent))
es    = Repeated 1 (Message ResolvedIndexedEvent)
-> FieldType (Repeated 1 (Message ResolvedIndexedEvent))
forall a. HasField a => a -> FieldType a
getField (Repeated 1 (Message ResolvedIndexedEvent)
 -> FieldType (Repeated 1 (Message ResolvedIndexedEvent)))
-> Repeated 1 (Message ResolvedIndexedEvent)
-> FieldType (Repeated 1 (Message ResolvedIndexedEvent))
forall a b. (a -> b) -> a -> b
$ Response -> Repeated 1 (Message ResolvedIndexedEvent)
_events Response
resp
                       evts :: [ResolvedEvent]
evts  = (ResolvedIndexedEvent -> ResolvedEvent)
-> [ResolvedIndexedEvent] -> [ResolvedEvent]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ResolvedIndexedEvent -> ResolvedEvent
newResolvedEvent [ResolvedIndexedEvent]
FieldType (Repeated 1 (Message ResolvedIndexedEvent))
es
                       err :: FieldType (Field 7 (OptionalField (Last (Value Text))))
err   = Field 7 (OptionalField (Last (Value Text)))
-> FieldType (Field 7 (OptionalField (Last (Value Text))))
forall a. HasField a => a -> FieldType a
getField (Field 7 (OptionalField (Last (Value Text)))
 -> FieldType (Field 7 (OptionalField (Last (Value Text)))))
-> Field 7 (OptionalField (Last (Value Text)))
-> FieldType (Field 7 (OptionalField (Last (Value Text))))
forall a b. (a -> b) -> a -> b
$ Response -> Optional 7 (Value Text)
_error Response
resp
                       eos :: FieldType (Field 5 (RequiredField (Always (Value Bool))))
eos   = Field 5 (RequiredField (Always (Value Bool)))
-> FieldType (Field 5 (RequiredField (Always (Value Bool))))
forall a. HasField a => a -> FieldType a
getField (Field 5 (RequiredField (Always (Value Bool)))
 -> FieldType (Field 5 (RequiredField (Always (Value Bool)))))
-> Field 5 (RequiredField (Always (Value Bool)))
-> FieldType (Field 5 (RequiredField (Always (Value Bool))))
forall a b. (a -> b) -> a -> b
$ Response -> Required 5 (Value Bool)
_endOfStream Response
resp
                       nxt :: FieldType (Field 3 (RequiredField (Always (Value Int64))))
nxt   = Field 3 (RequiredField (Always (Value Int64)))
-> FieldType (Field 3 (RequiredField (Always (Value Int64))))
forall a. HasField a => a -> FieldType a
getField (Field 3 (RequiredField (Always (Value Int64)))
 -> FieldType (Field 3 (RequiredField (Always (Value Int64)))))
-> Field 3 (RequiredField (Always (Value Int64)))
-> FieldType (Field 3 (RequiredField (Always (Value Int64))))
forall a b. (a -> b) -> a -> b
$ Response -> Required 3 (Value Int64)
_nextNumber Response
resp
                       found :: StreamSlice
found =
                           if [ResolvedEvent] -> Bool
forall mono. MonoFoldable mono => mono -> Bool
null [ResolvedEvent]
evts Bool -> Bool -> Bool
&& Bool
FieldType (Field 5 (RequiredField (Always (Value Bool))))
eos
                           then StreamSlice
forall t. Slice t
SliceEndOfStream
                           else [ResolvedEvent] -> Maybe EventNumber -> StreamSlice
forall t. [ResolvedEvent] -> Maybe t -> Slice t
Slice [ResolvedEvent]
evts (if Bool
FieldType (Field 5 (RequiredField (Always (Value Bool))))
eos then Maybe EventNumber
forall a. Maybe a
Nothing else EventNumber -> Maybe EventNumber
forall a. a -> Maybe a
Just (EventNumber -> Maybe EventNumber)
-> EventNumber -> Maybe EventNumber
forall a b. (a -> b) -> a -> b
$ Int64 -> EventNumber
EventNumber Int64
FieldType (Field 3 (RequiredField (Always (Value Int64))))
nxt) in
                   case FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
r of
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
NO_STREAM -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure ReadResult EventNumber StreamSlice
forall a. ReadResult EventNumber a
ReadNoStream
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
STREAM_DELETED -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (ReadResult EventNumber StreamSlice
 -> IO (ReadResult EventNumber StreamSlice))
-> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall a b. (a -> b) -> a -> b
$ StreamName -> ReadResult EventNumber StreamSlice
forall a. StreamName -> ReadResult EventNumber a
ReadStreamDeleted (StreamName -> ReadResult EventNumber StreamSlice)
-> StreamName -> ReadResult EventNumber StreamSlice
forall a b. (a -> b) -> a -> b
$ Text -> StreamName
StreamName Text
stream
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
NOT_MODIFIED -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure ReadResult EventNumber StreamSlice
forall t a. ReadResult t a
ReadNotModified
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
ERROR -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe Text -> ReadResult EventNumber StreamSlice
forall t a. Maybe Text -> ReadResult t a
ReadError Maybe Text
FieldType (Field 7 (OptionalField (Last (Value Text))))
err)
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
ACCESS_DENIED -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (ReadResult EventNumber StreamSlice
 -> IO (ReadResult EventNumber StreamSlice))
-> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall a b. (a -> b) -> a -> b
$ StreamName -> ReadResult EventNumber StreamSlice
forall t a. StreamId t -> ReadResult t a
ReadAccessDenied (StreamName -> ReadResult EventNumber StreamSlice)
-> StreamName -> ReadResult EventNumber StreamSlice
forall a b. (a -> b) -> a -> b
$ Text -> StreamName
StreamName Text
stream
                     FieldType (Field 2 (RequiredField (Always (Enumeration Result))))
SUCCESS -> ReadResult EventNumber StreamSlice
-> IO (ReadResult EventNumber StreamSlice)
forall (f :: * -> *) a. Applicative f => a -> f a
pure (StreamSlice -> ReadResult EventNumber StreamSlice
forall a t. a -> ReadResult t a
ReadSuccess StreamSlice
found)