TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-03 15:00:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化内存缓存限制

Browse Source
This commit is contained in:
GoEdgeLab
2023-10-01 14:11:48 +08:00
parent 486b972197
commit 4dc24c94b3
3 changed files with 56 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/caches/writer_memory.go
View File

@@ -20,12 +20,12 @@ type MemoryWriter struct {
hash uint64
item *MemoryItem
endFunc func()
endFunc func(valueItem *MemoryItem)
once sync.Once
}
func NewMemoryWriter(memoryStorage *MemoryStorage, key string, expiredAt int64, status int, isDirty bool, maxSize int64, endFunc func()) *MemoryWriter {
w := &MemoryWriter{
func NewMemoryWriter(memoryStorage *MemoryStorage, key string, expiredAt int64, status int, isDirty bool, maxSize int64, endFunc func(valueItem *MemoryItem)) *MemoryWriter {
var w = &MemoryWriter{
storage: memoryStorage,
key: key,
expiredAt: expiredAt,
@@ -39,6 +39,7 @@ func NewMemoryWriter(memoryStorage *MemoryStorage, key string, expiredAt int64,
maxSize: maxSize,
endFunc: endFunc,
}
w.hash = w.calculateHash(key)
return w
@@ -87,7 +88,7 @@ func (this *MemoryWriter) BodySize() int64 {
func (this *MemoryWriter) Close() error {
// 需要在Locker之外
defer this.once.Do(func() {
this.endFunc()
this.endFunc(this.item)
})
if this.item == nil {
@@ -102,7 +103,6 @@ func (this *MemoryWriter) Close() error {
select {
case this.storage.dirtyChan <- this.key:
default:
}
}
}
@@ -115,7 +115,7 @@ func (this *MemoryWriter) Close() error {
func (this *MemoryWriter) Discard() error {
// 需要在Locker之外
defer this.once.Do(func() {
this.endFunc()
this.endFunc(nil)
})
this.storage.locker.Lock()