Skip to content

Commit 87285fc

Browse files
chore: implement PeerstoreProvider#GetUnstakedPeerstore()
1 parent dd542c0 commit 87285fc

File tree

3 files changed

+26
-36
lines changed

3 files changed

+26
-36
lines changed

p2p/providers/peerstore_provider/persistence/provider.go

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
package persistence
22

33
import (
4+
"fmt"
5+
46
"github.com/pokt-network/pocket/p2p/providers/peerstore_provider"
57
typesP2P "github.com/pokt-network/pocket/p2p/types"
68
"github.com/pokt-network/pocket/shared/modules"
@@ -58,5 +60,16 @@ func (persistencePSP *persistencePeerstoreProvider) GetStakedPeerstoreAtHeight(h
5860

5961
// GetStakedPeerstoreAtHeight implements the respective `PeerstoreProvider` interface method.
6062
func (persistencePSP *persistencePeerstoreProvider) GetUnstakedPeerstore() (typesP2P.Peerstore, error) {
61-
return peerstore_provider.GetUnstakedPeerstore(persistencePSP.GetBus())
63+
// TECHDEBT(#810, #811): use `bus.GetUnstakedActorRouter()` once it's available.
64+
unstakedActorRouterMod, err := persistencePSP.GetBus().GetModulesRegistry().GetModule(typesP2P.UnstakedActorRouterSubmoduleName)
65+
if err != nil {
66+
return nil, err
67+
}
68+
69+
unstakedActorRouter, ok := unstakedActorRouterMod.(typesP2P.Router)
70+
if !ok {
71+
return nil, fmt.Errorf("unexpected unstaked actor router submodule type: %T", unstakedActorRouterMod)
72+
}
73+
74+
return unstakedActorRouter.GetPeerstore(), nil
6275
}

p2p/providers/peerstore_provider/rpc/provider.go

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,18 @@ func (rpcPSP *rpcPeerstoreProvider) GetStakedPeerstoreAtHeight(height uint64) (t
9292
}
9393

9494
func (rpcPSP *rpcPeerstoreProvider) GetUnstakedPeerstore() (typesP2P.Peerstore, error) {
95-
return peerstore_provider.GetUnstakedPeerstore(rpcPSP.GetBus())
95+
// TECHDEBT(#810, #811): use `bus.GetUnstakedActorRouter()` once it's available.
96+
unstakedActorRouterMod, err := rpcPSP.GetBus().GetModulesRegistry().GetModule(typesP2P.UnstakedActorRouterSubmoduleName)
97+
if err != nil {
98+
return nil, err
99+
}
100+
101+
unstakedActorRouter, ok := unstakedActorRouterMod.(typesP2P.Router)
102+
if !ok {
103+
return nil, fmt.Errorf("unexpected unstaked actor router submodule type: %T", unstakedActorRouterMod)
104+
}
105+
106+
return unstakedActorRouter.GetPeerstore(), nil
96107
}
97108

98109
func (rpcPSP *rpcPeerstoreProvider) initRPCClient() {

p2p/providers/peerstore_provider/unstaked.go

Lines changed: 0 additions & 34 deletions
This file was deleted.

0 commit comments

Comments
 (0)