cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

plumb context in tvx

Browse Source
This commit is contained in:
vyzo 2021-12-13 15:03:54 +02:00 committed by Aayush Rajasekaran
parent 773c20a81c
commit 7377cf727e
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cmd/tvx/stores.go
View File

@ -113,14 +113,14 @@ func (pb *proxyingBlockstore) FinishTracing() map[cid.Cid]struct{} {
return ret
}
func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
func (pb *proxyingBlockstore) Get(ctx context.Context, cid cid.Cid) (blocks.Block, error) {
pb.lk.Lock()
if pb.tracing {
pb.traced[cid] = struct{}{}
}
pb.lk.Unlock()
if block, err := pb.Blockstore.Get(cid); err == nil {
if block, err := pb.Blockstore.Get(ctx, cid); err == nil {
return block, err
}
@ -134,7 +134,7 @@ func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
return nil, err
}
err = pb.Blockstore.Put(block)
err = pb.Blockstore.Put(ctx, block)
if err != nil {
return nil, err
}
@ -142,16 +142,16 @@ func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
return block, nil
}
func (pb *proxyingBlockstore) Put(block blocks.Block) error {
func (pb *proxyingBlockstore) Put(ctx context.Context, block blocks.Block) error {
pb.lk.Lock()
if pb.tracing {
pb.traced[block.Cid()] = struct{}{}
}
pb.lk.Unlock()
return pb.Blockstore.Put(block)
return pb.Blockstore.Put(ctx, block)
}
func (pb *proxyingBlockstore) PutMany(blocks []blocks.Block) error {
func (pb *proxyingBlockstore) PutMany(ctx context.Context, blocks []blocks.Block) error {
pb.lk.Lock()
if pb.tracing {
for _, b := range blocks {
@ -159,5 +159,5 @@ func (pb *proxyingBlockstore) PutMany(blocks []blocks.Block) error {
}
}
pb.lk.Unlock()
return pb.Blockstore.PutMany(blocks)
return pb.Blockstore.PutMany(ctx, blocks)
}