feat: add data presence check before batch update in utils
This commit is contained in:
@@ -2,11 +2,12 @@ package model
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"github.com/bytedance/gopkg/util/gopool"
|
|
||||||
"gorm.io/gorm"
|
|
||||||
"one-api/common"
|
"one-api/common"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/bytedance/gopkg/util/gopool"
|
||||||
|
"gorm.io/gorm"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -48,6 +49,22 @@ func addNewRecord(type_ int, id int, value int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func batchUpdate() {
|
func batchUpdate() {
|
||||||
|
// check if there's any data to update
|
||||||
|
hasData := false
|
||||||
|
for i := 0; i < BatchUpdateTypeCount; i++ {
|
||||||
|
batchUpdateLocks[i].Lock()
|
||||||
|
if len(batchUpdateStores[i]) > 0 {
|
||||||
|
hasData = true
|
||||||
|
batchUpdateLocks[i].Unlock()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
batchUpdateLocks[i].Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !hasData {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
common.SysLog("batch update started")
|
common.SysLog("batch update started")
|
||||||
for i := 0; i < BatchUpdateTypeCount; i++ {
|
for i := 0; i < BatchUpdateTypeCount; i++ {
|
||||||
batchUpdateLocks[i].Lock()
|
batchUpdateLocks[i].Lock()
|
||||||
|
|||||||
Reference in New Issue
Block a user