mirror of https://github.com/voidlizard/hbs2
88 lines
2.4 KiB
Haskell
88 lines
2.4 KiB
Haskell
{-# OPTIONS_GHC -fno-warn-orphans #-}
|
|
{-# Language UndecidableInstances #-}
|
|
module HBS2.Peer.RPC.Internal.Storage
|
|
( module HBS2.Peer.RPC.API.Storage
|
|
, module HBS2.Peer.RPC.Class
|
|
)
|
|
where
|
|
|
|
|
|
import HBS2.Prelude.Plated
|
|
import HBS2.Actors.Peer.Types
|
|
import HBS2.Data.Types.Refs (HashRef(..),refMetaData)
|
|
import HBS2.Storage
|
|
import HBS2.Peer.RPC.Class
|
|
import HBS2.Peer.RPC.API.Storage
|
|
|
|
import HBS2.Net.Proto.Service
|
|
import HBS2.System.Logger.Simple
|
|
|
|
import Data.Functor
|
|
|
|
-- type StorageContext m = (MonadIO m, HasStorage m)
|
|
type StorageContext m = (MonadIO m, HasStorage m)
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageHasBlock where
|
|
|
|
handleMethod href = do
|
|
-- debug $ "rpc.storage.hasBlock" <+> pretty href
|
|
sto <- getStorage
|
|
liftIO $ hasBlock sto (fromHashRef href)
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageGetBlock where
|
|
|
|
handleMethod href = do
|
|
-- debug $ "rpc.storage.getBlock" <+> pretty href
|
|
sto <- getStorage
|
|
liftIO $ getBlock sto (fromHashRef href)
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageEnqueueBlock where
|
|
|
|
handleMethod lbs = do
|
|
-- debug $ "rpc.storage.enqueueBlock"
|
|
sto <- getStorage
|
|
liftIO $ enqueueBlock sto lbs <&> fmap HashRef
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStoragePutBlock where
|
|
|
|
handleMethod lbs = do
|
|
-- debug $ "rpc.storage.putBlock"
|
|
sto <- getStorage
|
|
liftIO $ putBlock sto lbs <&> fmap HashRef
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageDelBlock where
|
|
|
|
handleMethod href = do
|
|
-- debug $ "rpc.storage.delBlock" <+> pretty href
|
|
sto <- getStorage
|
|
liftIO $ delBlock sto (fromHashRef href)
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageGetChunk where
|
|
|
|
handleMethod (h,o,s) = do
|
|
sto <- getStorage
|
|
liftIO $ getChunk sto (fromHashRef h) o s
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageGetRef where
|
|
|
|
handleMethod ref = do
|
|
trace $ "rpc.storage.getRef" <+> pretty ref
|
|
sto <- getStorage
|
|
liftIO $ getRef sto ref <&> fmap HashRef
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageUpdateRef where
|
|
|
|
handleMethod (ref, val) = do
|
|
trace $ "rpc.storage.updateRef" <+> pretty ref <+> pretty (refMetaData ref)
|
|
sto <- getStorage
|
|
liftIO $ updateRef sto ref (fromHashRef val)
|
|
|
|
instance (StorageContext m) => HandleMethod m RpcStorageDelRef where
|
|
|
|
handleMethod ref = do
|
|
trace $ "rpc.storage.delRef" <+> pretty ref
|
|
sto <- getStorage
|
|
liftIO $ delRef sto ref
|
|
|
|
|