... and it works!

This commit is contained in:
Dmitry Zuikov 2023-01-17 06:58:01 +03:00
parent d509fad3bc
commit c97a7c1bb3
3 changed files with 42 additions and 16 deletions

View File

@ -109,6 +109,7 @@ library
, stm , stm
, stm-chans , stm-chans
, text , text
, transformers
, uniplate , uniplate
hs-source-dirs: lib hs-source-dirs: lib

View File

@ -3,25 +3,33 @@
module HBS2.Net.Peer where module HBS2.Net.Peer where
import HBS2.Prelude import HBS2.Prelude
import HBS2.Prelude.Plated
import HBS2.Net.Proto import HBS2.Net.Proto
import HBS2.Net.Messaging import HBS2.Net.Messaging
import HBS2.Clock
import Lens.Micro.Platform import Lens.Micro.Platform
import Data.ByteString.Lazy ( ByteString ) import Data.ByteString.Lazy ( ByteString )
import Data.Foldable import Data.Foldable
import Control.Monad.Reader import Control.Monad.Reader
import Data.Map qualified as Map import Data.Map qualified as Map
import Data.Proxy
import GHC.TypeLits import GHC.TypeLits
import Codec.Serialise qualified as S import Control.Monad.Trans.Maybe
import Codec.Serialise hiding (encode,decode)
data AnyMessage e = AnyMessage Integer (Encoded e) data AnyMessage e = AnyMessage Integer (Encoded e)
deriving stock (Generic)
data EngineEnv e = forall bus . (Messaging bus e (AnyMessage e)) => instance Serialise (Encoded e) => Serialise (AnyMessage e)
data EngineEnv e = forall bus . ( Messaging bus e ByteString
, Serialise (Encoded e)
) =>
EngineEnv EngineEnv
{ _peer :: Maybe (Peer e) { _peer :: Maybe (Peer e)
, _self :: Peer e , _self :: Peer e
, bus :: bus , bus :: bus
} }
makeLenses 'EngineEnv makeLenses 'EngineEnv
@ -66,7 +74,8 @@ instance (MonadIO m, HasProtocol e p) => Request e p (EngineM e m) where
let proto = protoId @e @p (Proxy @p) let proto = protoId @e @p (Proxy @p)
ask >>= \case ask >>= \case
EngineEnv { _self = s, bus = b} -> do EngineEnv { _self = s, bus = b} -> do
liftIO $ sendTo b (To p) (From s) (AnyMessage proto (encode msg)) let bs = serialise (AnyMessage @e proto (encode msg))
liftIO $ sendTo b (To p) (From s) bs
instance (MonadIO m, HasProtocol e p) => Response e p (EngineM e m) where instance (MonadIO m, HasProtocol e p) => Response e p (EngineM e m) where
response resp = do response resp = do
@ -77,12 +86,14 @@ instance (MonadIO m, HasProtocol e p) => Response e p (EngineM e m) where
, _self = s , _self = s
, bus = b , bus = b
} ) -> do } ) -> do
liftIO $ sendTo b (To p) (From s) (AnyMessage proto (encode resp)) let bs = serialise (AnyMessage @e proto (encode resp))
liftIO $ sendTo b (To p) (From s) bs
_ -> pure () _ -> pure ()
newEnv :: forall e bus m . ( Monad m newEnv :: forall e bus m . ( Monad m
, Messaging bus e (AnyMessage e) , Messaging bus e ByteString
, Serialise (Encoded e)
) )
=> Peer e => Peer e
-> bus -> bus
@ -91,7 +102,12 @@ newEnv :: forall e bus m . ( Monad m
newEnv p pipe = pure $ EngineEnv Nothing p pipe newEnv p pipe = pure $ EngineEnv Nothing p pipe
runPeer :: MonadIO m => EngineEnv e -> [AnyProtocol e (EngineM e m)] -> m a runPeer :: forall e m a . ( MonadIO m
)
=> EngineEnv e
-> [AnyProtocol e (EngineM e m)]
-> m a
runPeer env@(EngineEnv {bus = pipe}) hh = do runPeer env@(EngineEnv {bus = pipe}) hh = do
let me = env ^. self let me = env ^. self
@ -105,12 +121,20 @@ runPeer env@(EngineEnv {bus = pipe}) hh = do
forever $ do forever $ do
messages <- receive pipe (To me) messages <- receive pipe (To me)
for_ messages $ \(From pip, AnyMessage n msg) -> local (set peer (Just pip)) do for_ messages $ \(From pip, bs) -> do
case Map.lookup n disp of case deserialiseOrFail @(AnyMessage e) bs of
Nothing -> pure ()
Just (AnyProtocol { protoDecode = decoder Left _-> pure () -- liftIO $ print "failed to deserialise"
, handle = h
}) -> maybe (pure ()) h (decoder msg) Right (AnyMessage n msg) -> do
local (set peer (Just pip)) do
case Map.lookup n disp of
Nothing -> pure ()
Just (AnyProtocol { protoDecode = decoder
, handle = h
}) -> maybe (pure ()) h (decoder msg)

View File

@ -34,7 +34,6 @@ instance HasProtocol Fake (PeekPoke Fake) where
decode = readMay decode = readMay
encode = show encode = show
pingPongHandler :: forall e m . ( MonadIO m pingPongHandler :: forall e m . ( MonadIO m
, Response e (PingPong e) m , Response e (PingPong e) m
, HasProtocol e (PingPong e) , HasProtocol e (PingPong e)
@ -90,6 +89,8 @@ testUniqProtoId = do
, makeResponse peekPokeHandler , makeResponse peekPokeHandler
] ]
void $ waitAnyCatchCancel [pip1, pip2] (_, e) <- waitAnyCatchCancel [pip1, pip2]
print e