文件缓存增加自动限速/提升本地缓存数据库写入和查询速度

internal/caches/max_open_files.go

@@ -0,0 +1,83 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package caches
+
+import (
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
+	"github.com/TeaOSLab/EdgeNode/internal/goman"
+	"sync/atomic"
+	"time"
+)
+
+const (
+	minOpenFilesValue int32 = 2
+	maxOpenFilesValue int32 = 65535
+
+	modSlow int32 = 1
+	modFast int32 = 2
+)
+
+type MaxOpenFiles struct {
+	step         int32
+	maxOpenFiles int32
+	ptr          *int32
+	ticker       *time.Ticker
+	mod          int32
+}
+
+func NewMaxOpenFiles(step int32) *MaxOpenFiles {
+	if step <= 0 {
+		step = 2
+	}
+	var f = &MaxOpenFiles{
+		step:         step,
+		maxOpenFiles: 2,
+	}
+	if teaconst.DiskIsFast {
+		f.maxOpenFiles = 32
+	}
+	f.ptr = &f.maxOpenFiles
+	f.ticker = time.NewTicker(1 * time.Second)
+	goman.New(func() {
+		for range f.ticker.C {
+			var mod = atomic.LoadInt32(&f.mod)
+			switch mod {
+			case modSlow:
+				// we decrease more quickly, with more steps
+				if atomic.AddInt32(f.ptr, -step*2) <= 0 {
+					atomic.StoreInt32(f.ptr, minOpenFilesValue)
+				}
+			case modFast:
+				if atomic.AddInt32(f.ptr, step) >= maxOpenFilesValue {
+					atomic.StoreInt32(f.ptr, maxOpenFilesValue)
+				}
+			}
+
+			// reset mod
+			atomic.StoreInt32(&f.mod, 0)
+		}
+	})
+	return f
+}
+
+func (this *MaxOpenFiles) Fast() {
+	if atomic.LoadInt32(&this.mod) == 0 {
+		this.mod = modFast
+	}
+}
+
+func (this *MaxOpenFiles) Slow() {
+	atomic.StoreInt32(&this.mod, modSlow)
+}
+
+func (this *MaxOpenFiles) Max() int32 {
+	if atomic.LoadInt32(&this.mod) == modSlow {
+		return 0
+	}
+
+	var v = atomic.LoadInt32(this.ptr)
+	if v <= minOpenFilesValue {
+		return minOpenFilesValue
+	}
+	return v
+}