feat(local): thumbnail token bucket smooth migration (#7425)

* feat(local): allow to migrate static token buckets * improve(local): token bucket migration boundary handling
2024-11-01 20:58:53 +08:00
parent 4955d8cec8
commit 34a148c83d
2 changed files with 37 additions and 3 deletions
--- a/drivers/local/driver.go
+++ b/drivers/local/driver.go
@ -76,7 +76,7 @@ func (d *Local) Init(ctx context.Context) error {
 	if d.thumbConcurrency == 0 {
 		d.thumbTokenBucket = NewNopTokenBucket()
 	} else {
-		d.thumbTokenBucket = NewStaticTokenBucket(d.thumbConcurrency)
+		d.thumbTokenBucket = NewStaticTokenBucketWithMigration(d.thumbTokenBucket, d.thumbConcurrency)
 	}
 	return nil
 }