coosld/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2026-02-23 03:17:43 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

allow promises to fail

Browse Source 
License: MIT
Signed-off-by: Jeromy <jeromyj@gmail.com>
This commit is contained in:
Jeromy 2016-01-27 18:55:53 -08:00
parent 600c95eb53
commit d84dfafe69
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
merkledag/merkledag.go
View File

@ -192,6 +192,9 @@ func GetNodes(ctx context.Context, ds DAGService, keys []key.Key) []NodeGetter {
select {
case opt, ok := <-nodechan:
if !ok {
for _, p := range promises {
p.Fail(ErrNotFound)
}
return
}
@ -239,6 +242,7 @@ func newNodePromise(ctx context.Context) (NodeGetter, chan<- *Node) {
return &nodePromise{
recv: ch,
ctx: ctx,
err: make(chan error, 1),
}, ch
}
@ -246,6 +250,7 @@ type nodePromise struct {
cache *Node
recv <-chan *Node
ctx context.Context
err chan error
}
// NodeGetter provides a promise like interface for a dag Node
@ -254,6 +259,11 @@ type nodePromise struct {
// cached node.
type NodeGetter interface {
Get(context.Context) (*Node, error)
Fail(err error)
}
func (np *nodePromise) Fail(err error) {
np.err <- err
}
func (np *nodePromise) Get(ctx context.Context) (*Node, error) {
@ -268,6 +278,8 @@ func (np *nodePromise) Get(ctx context.Context) (*Node, error) {
return nil, np.ctx.Err()
case <-ctx.Done():
return nil, ctx.Err()
case err := <-np.err:
return nil, err
}
return np.cache, nil
}