baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

lint

Browse source
This commit is contained in:
Michael Yang 2024-01-29 17:01:31 -08:00
parent bea007deb7
commit 6a4b994433
2 changed files with 5 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
server/download.go
View file

@ -188,7 +188,7 @@ func (b *blobDownload) Run(ctx context.Context, requestURL *url.URL, opts *regis
continue
}
g.Go(func() error {
g.Go(inner, func() error {
var err error
for try := 0; try < maxRetries; try++ {
w := io.NewOffsetWriter(file, part.StartsAt())
@ -238,7 +238,6 @@ func (b *blobDownload) Run(ctx context.Context, requestURL *url.URL, opts *regis
}
b.done = true
return
}
func (b *blobDownload) downloadChunk(ctx context.Context, requestURL *url.URL, w io.Writer, part *blobDownloadPart, opts *registryOptions) error {
@ -411,7 +410,6 @@ func downloadBlob(ctx context.Context, opts downloadOpts) error {
type LimitGroup struct {
*errgroup.Group
context.Context
Semaphore *semaphore.Weighted
weight, max_weight int64
@ -421,17 +419,16 @@ func NewLimitGroup(ctx context.Context, n int64) (*LimitGroup, context.Context)
g, ctx := errgroup.WithContext(ctx)
return &LimitGroup{
Group: g,
Context: ctx,
Semaphore: semaphore.NewWeighted(n),
weight: n,
max_weight: n,
}, ctx
}
func (g *LimitGroup) Go(fn func() error) {
func (g *LimitGroup) Go(ctx context.Context, fn func() error) {
weight := g.weight
g.Semaphore.Acquire(g.Context, weight)
if g.Context.Err() != nil {
_ = g.Semaphore.Acquire(ctx, weight)
if ctx.Err() != nil {
return
}

2
server/upload.go
View file

@ -172,7 +172,7 @@ func (b *blobUpload) Run(ctx context.Context, opts *registryOptions) {
select {
case <-inner.Done():
case requestURL := <-b.nextURL:
g.Go(func() error {
g.Go(inner, func() error {
var err error
for try := 0; try < maxRetries; try++ {
err = b.uploadPart(inner, http.MethodPatch, requestURL, part, opts)