58 lines
1,008 B
Go
58 lines
1,008 B
Go
|
package writecachebadger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
"sync/atomic"
|
||
|
|
|
||
|
|
"github.com/dgraph-io/badger/v4"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (c *cache) estimateCacheSize() uint64 {
|
||
|
|
onDiskSize, _ := c.db.EstimateSize(nil)
|
||
|
|
c.metrics.SetEstimateSize(onDiskSize, 0)
|
||
|
|
return onDiskSize
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *cache) incSizeDB(sz uint64) uint64 {
|
||
|
|
return sz + c.maxObjectSize
|
||
|
|
}
|
||
|
|
|
||
|
|
type counters struct {
|
||
|
|
cDB atomic.Uint64
|
||
|
|
}
|
||
|
|
|
||
|
|
func (x *counters) IncDB() {
|
||
|
|
x.cDB.Add(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (x *counters) DecDB() {
|
||
|
|
x.cDB.Add(math.MaxUint64)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (x *counters) DB() uint64 {
|
||
|
|
return x.cDB.Load()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *cache) initCounters() error {
|
||
|
|
var inDB uint64
|
||
|
|
err := c.db.View(func(tx *badger.Txn) error {
|
||
|
|
opts := badger.DefaultIteratorOptions
|
||
|
|
opts.PrefetchValues = false
|
||
|
|
it := tx.NewIterator(opts)
|
||
|
|
defer it.Close()
|
||
|
|
for it.Rewind(); it.Valid(); it.Next() {
|
||
|
|
inDB++
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("could not read write-cache DB counter: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
c.objCounters.cDB.Store(inDB)
|
||
|
|
c.metrics.SetActualCounters(inDB, 0)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|