optimize: add cache for asset path

common/assets/assets.go

@@ -15,9 +15,54 @@ import (
 	"path/filepath"
 	"runtime"
 	"strings"
+	"sync"
+	"time"
 )
 
-func GetLocationAsset(log *logrus.Logger, filename string) (path string, err error) {
+const CacheTimeout = 5 * time.Second
+
+type CacheItem struct {
+	Filename string
+	Path     string
+
+	CacheDeadline time.Time
+}
+
+type LocationFinder struct {
+	mu sync.Mutex
+	m  map[string]CacheItem
+}
+
+func NewLocationFinder() *LocationFinder {
+	return &LocationFinder{
+		mu: sync.Mutex{},
+		m:  map[string]CacheItem{},
+	}
+}
+
+func (c *LocationFinder) GetLocationAsset(log *logrus.Logger, filename string) (path string, err error) {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	if item, ok := c.m[filename]; ok && time.Now().Before(item.CacheDeadline) {
+		return item.Path, nil
+	}
+	defer func() {
+		if err == nil {
+			c.m[filename] = CacheItem{
+				Filename:      filename,
+				Path:          path,
+				CacheDeadline: time.Now().Add(CacheTimeout),
+			}
+			time.AfterFunc(CacheTimeout, func() {
+				c.mu.Lock()
+				defer c.mu.Unlock()
+				if item, ok := c.m[filename]; ok && time.Now().After(item.CacheDeadline) {
+					delete(c.m, filename)
+				}
+			})
+		}
+	}()
+
 	folder := "dae"
 	location := os.Getenv("DAE_LOCATION_ASSET")
 	// check if DAE_LOCATION_ASSET is set
@@ -76,3 +121,5 @@ func GetLocationAsset(log *logrus.Logger, filename string) (path string, err err
 		}
 	}
 }
+
+var DefaultLocationFinder = NewLocationFinder()

component/routing/optimizer.go

@@ -154,7 +154,7 @@ func (o *DatReaderOptimizer) loadGeoSite(filename string, code string) (params [
 	if !strings.HasSuffix(filename, ".dat") {
 		filename += ".dat"
 	}
-	filePath, err := assets.GetLocationAsset(o.Logger, filename)
+	filePath, err := assets.DefaultLocationFinder.GetLocationAsset(o.Logger, filename)
 	if err != nil {
 		o.Logger.Debugf("Failed to read geosite \"%v:%v\": %v", filename, code, err)
 		return nil, err
@@ -199,7 +199,7 @@ func (o *DatReaderOptimizer) loadGeoIp(filename string, code string) (params []*
 	if !strings.HasSuffix(filename, ".dat") {
 		filename += ".dat"
 	}
-	filePath, err := assets.GetLocationAsset(o.Logger, filename)
+	filePath, err := assets.DefaultLocationFinder.GetLocationAsset(o.Logger, filename)
 	if err != nil {
 		o.Logger.Debugf("Failed to read geoip \"%v:%v\": %v", filename, code, err)
 		return nil, err