JakeHillion/storj
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

storagenode/piecestore: small cleanup

Browse Source 
* normalize ExistsCheckWorkers flag to avoid setting incorrect values
* wait for limiter to finish if context was canceled

Change-Id: I688a395bd958cd09233605fc264d43f91ec45ed1
This commit is contained in:
Michal Niewrzal 2022-12-19 08:53:10 +01:00
parent 1c309a0318
commit d1d617d654
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
storagenode/piecestore/endpoint.go
View File

@ -213,6 +213,10 @@ func (endpoint *Endpoint) Exists(
return &pb.ExistsResponse{}, nil
}
if endpoint.config.ExistsCheckWorkers < 1 {
endpoint.config.ExistsCheckWorkers = 1
}
limiter := sync2.NewLimiter(endpoint.config.ExistsCheckWorkers)
var mu sync.Mutex
@ -240,7 +244,8 @@ func (endpoint *Endpoint) Exists(
}
})
if !ok {
return &pb.ExistsResponse{}, rpcstatus.Wrap(rpcstatus.Internal, ctx.Err())
limiter.Wait()
return nil, rpcstatus.Wrap(rpcstatus.Canceled, ctx.Err())
}
}
@ -248,7 +253,7 @@ func (endpoint *Endpoint) Exists(
return &pb.ExistsResponse{
Missing: missing,
}, rpcstatus.Wrap(rpcstatus.Internal, err)
}, nil
}
// Upload handles uploading a piece on piece store.