LumeWeb
/
libs5-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: create provider store interface for use in hash query

This commit is contained in:
Derrick Hammer 2024-01-24 02:32:49 -05:00
parent 9b464e0932
commit d734e1a89b
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
3 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
interfaces/node.go
View File

@ -29,4 +29,6 @@ type Node interface {
GetMetadataByCID(cid *encoding.CID) (metadata.Metadata, error) GetMetadataByCID(cid *encoding.CID) (metadata.Metadata, error)
WaitOnConnectedPeers() WaitOnConnectedPeers()
ConnectionTracker() *sync.WaitGroup ConnectionTracker() *sync.WaitGroup
SetProviderStore(store ProviderStore)
ProviderStore() ProviderStore
} }

11
interfaces/provider_store.go Normal file
View File

@ -0,0 +1,11 @@
package interfaces
import (
"git.lumeweb.com/LumeWeb/libs5-go/encoding"
"git.lumeweb.com/LumeWeb/libs5-go/types"
)
type ProviderStore interface {
CanProvide(hash *encoding.Multihash, kind []types.StorageLocationType) bool
Provide(hash *encoding.Multihash, kind []types.StorageLocationType) (StorageLocation, error)
}

9
node/node.go
View File

@ -34,6 +34,7 @@ type NodeImpl struct {
services interfaces.Services services interfaces.Services
httpClient *resty.Client httpClient *resty.Client
connections sync.WaitGroup connections sync.WaitGroup
providerStore interfaces.ProviderStore
} }
func (n *NodeImpl) NetworkId() string { func (n *NodeImpl) NetworkId() string {
@ -333,3 +334,11 @@ func (n *NodeImpl) WaitOnConnectedPeers() {
func (n *NodeImpl) ConnectionTracker() *sync.WaitGroup { func (n *NodeImpl) ConnectionTracker() *sync.WaitGroup {
return &n.connections return &n.connections
} }
func (n *NodeImpl) SetProviderStore(store interfaces.ProviderStore) {
n.providerStore = store
}
func (n *NodeImpl) ProviderStore() interfaces.ProviderStore {
return n.providerStore
}