usememos/memos
1
1
Fork 0
mirror of https://github.com/usememos/memos.git synced 2024-12-23 11:11:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix: concurrent counter operates (#1706)

Browse Source 
Co-authored-by: Athurg Feng <athurg@gooth.org>
This commit is contained in:
Athurg Gooth 2023-05-22 11:08:49 +08:00 committed by GitHub
parent 1e4a867a9a
commit 0cea5ebaeb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/resource.go
View File

@ -13,6 +13,7 @@ import (
"regexp" "regexp"
"strconv" "strconv"
"strings" "strings"
"sync/atomic"
"time" "time"
"github.com/disintegration/imaging" "github.com/disintegration/imaging"
@ -501,7 +502,7 @@ func replacePathTemplate(path string, filename string) string {
return path return path
} }
var availableGeneratorAmount = 32 var availableGeneratorAmount int32 = 32
func getOrGenerateThumbnailImage(srcBlob []byte, dstPath string) ([]byte, error) { func getOrGenerateThumbnailImage(srcBlob []byte, dstPath string) ([]byte, error) {
if _, err := os.Stat(dstPath); err != nil { 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") 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") return nil, errors.New("not enough available generator amount")
} }
availableGeneratorAmount-- atomic.AddInt32(&availableGeneratorAmount, -1)
defer func() { defer func() {
availableGeneratorAmount++ atomic.AddInt32(&availableGeneratorAmount, 1)
}() }()
reader := bytes.NewReader(srcBlob) reader := bytes.NewReader(srcBlob)