TrueCloudLab/distribution
13
0
Fork 3
Code Issues 3 Pull requests Projects Releases 2 Packages Wiki Activity Actions

* NewStore ctx argument for logger extraction

Browse source 
* Add to WaitGroup on limiter `acquire()`
Signed-off-by: Andrey Voronkov <voronkovaa@gmail.com>
This commit is contained in:
Andrey Voronkov 2023-03-20 15:07:54 +03:00
parent 0ffef9d8f2
commit a555c9cbbe
2 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
registry/storage/registry.go
View file

@ -203,7 +203,7 @@ func (repo *repository) Named() reference.Named {
}
func (repo *repository) Tags(ctx context.Context) distribution.TagService {
tags := NewStore(repo, repo.registry.blobStore)
tags := NewStore(ctx, repo, repo.registry.blobStore)
return tags
}

9
registry/storage/tagstore.go
View file

@ -9,9 +9,9 @@ import (
"sync"
"github.com/distribution/distribution/v3"
dcontext "github.com/distribution/distribution/v3/context"
storagedriver "github.com/distribution/distribution/v3/registry/storage/driver"
"github.com/opencontainers/go-digest"
"github.com/sirupsen/logrus"
)
var _ distribution.TagService = &TagStore{}
@ -27,11 +27,12 @@ type TagStore struct {
lookupConcurrencyFactor int
}
func NewStore(repository *repository, blobStore *blobStore) *TagStore {
func NewStore(ctx context.Context, repository *repository, blobStore *blobStore) *TagStore {
logger := dcontext.GetLogger(ctx)
lookupConcurrencyFactor, err := strconv.Atoi(os.Getenv("STORAGE_TAGSTORE_LOOKUP_CONCURRENCY"))
if err != nil {
lookupConcurrencyFactor = 64
logrus.Infof("TagStore: STORAGE_TAGSTORE_LOOKUP_CONCURRENCY is not set. Using default %d as lookup concurrency factor", lookupConcurrencyFactor)
logger.Infof("TagStore: STORAGE_TAGSTORE_LOOKUP_CONCURRENCY is not set. Using default %d as lookup concurrency factor", lookupConcurrencyFactor)
}
return &TagStore{
repository: repository,
@ -171,11 +172,11 @@ func (ts *TagStore) Lookup(ctx context.Context, desc distribution.Descriptor) ([
outputChan := make(chan string, allTagsCount)
waitGroup := sync.WaitGroup{}
waitGroup.Add(allTagsCount)
limiter := make(chan struct{}, ts.lookupConcurrencyFactor)
acquire := func() {
limiter <- struct{}{}
waitGroup.Add(1)
}
release := func() {
<-limiter