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

Merge pull request #4278 from filecoin-project/tvx/trace-puts

Browse Source 
tvx: trace puts to blockstore for inclusion in CAR.
This commit is contained in:
Łukasz Magiera 2020-10-14 12:08:06 +02:00 committed by GitHub
commit 4ad8d85b13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/tvx/stores.go
View File

@ -87,7 +87,7 @@ type proxyingBlockstore struct {
ctx context.Context ctx context.Context
api api.FullNode api api.FullNode
lk sync.RWMutex lk sync.Mutex
tracing bool tracing bool
traced map[cid.Cid]struct{} traced map[cid.Cid]struct{}
@ -113,11 +113,11 @@ func (pb *proxyingBlockstore) FinishTracing() map[cid.Cid]struct{} {
} }
func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) { func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
pb.lk.RLock() pb.lk.Lock()
if pb.tracing { if pb.tracing {
pb.traced[cid] = struct{}{} pb.traced[cid] = struct{}{}
} }
pb.lk.RUnlock() pb.lk.Unlock()
if block, err := pb.Blockstore.Get(cid); err == nil { if block, err := pb.Blockstore.Get(cid); err == nil {
return block, err return block, err
@ -140,3 +140,12 @@ func (pb *proxyingBlockstore) Get(cid cid.Cid) (blocks.Block, error) {
return block, nil return block, nil
} }
func (pb *proxyingBlockstore) Put(block blocks.Block) error {
pb.lk.Lock()
if pb.tracing {
pb.traced[block.Cid()] = struct{}{}
}
pb.lk.Unlock()
return pb.Blockstore.Put(block)
}