mirror of
https://github.com/usememos/memos.git
synced 2024-12-23 03:02:10 +03:00
fix: concurrent counter operates (#1706)
Co-authored-by: Athurg Feng <athurg@gooth.org>
This commit is contained in:
parent
1e4a867a9a
commit
0cea5ebaeb
@ -13,6 +13,7 @@ import (
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"github.com/disintegration/imaging"
|
||||
@ -501,7 +502,7 @@ func replacePathTemplate(path string, filename string) string {
|
||||
return path
|
||||
}
|
||||
|
||||
var availableGeneratorAmount = 32
|
||||
var availableGeneratorAmount int32 = 32
|
||||
|
||||
func getOrGenerateThumbnailImage(srcBlob []byte, dstPath string) ([]byte, error) {
|
||||
if _, err := os.Stat(dstPath); err != nil {
|
||||
@ -509,12 +510,12 @@ func getOrGenerateThumbnailImage(srcBlob []byte, dstPath string) ([]byte, error)
|
||||
return nil, errors.Wrap(err, "failed to check thumbnail image stat")
|
||||
}
|
||||
|
||||
if availableGeneratorAmount <= 0 {
|
||||
if atomic.LoadInt32(&availableGeneratorAmount) <= 0 {
|
||||
return nil, errors.New("not enough available generator amount")
|
||||
}
|
||||
availableGeneratorAmount--
|
||||
atomic.AddInt32(&availableGeneratorAmount, -1)
|
||||
defer func() {
|
||||
availableGeneratorAmount++
|
||||
atomic.AddInt32(&availableGeneratorAmount, 1)
|
||||
}()
|
||||
|
||||
reader := bytes.NewReader(srcBlob)
|
||||
|
Loading…
Reference in New Issue
Block a user