File tree Expand file tree Collapse file tree 3 files changed +26
-36
lines changed
p2p/providers/peerstore_provider Expand file tree Collapse file tree 3 files changed +26
-36
lines changed Original file line number Diff line number Diff line change 1
1
package persistence
2
2
3
3
import (
4
+ "fmt"
5
+
4
6
"github.com/pokt-network/pocket/p2p/providers/peerstore_provider"
5
7
typesP2P "github.com/pokt-network/pocket/p2p/types"
6
8
"github.com/pokt-network/pocket/shared/modules"
@@ -58,5 +60,16 @@ func (persistencePSP *persistencePeerstoreProvider) GetStakedPeerstoreAtHeight(h
58
60
59
61
// GetStakedPeerstoreAtHeight implements the respective `PeerstoreProvider` interface method.
60
62
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
62
75
}
Original file line number Diff line number Diff line change @@ -92,7 +92,18 @@ func (rpcPSP *rpcPeerstoreProvider) GetStakedPeerstoreAtHeight(height uint64) (t
92
92
}
93
93
94
94
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
96
107
}
97
108
98
109
func (rpcPSP * rpcPeerstoreProvider ) initRPCClient () {
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments