1
0
mirror of https://github.com/astaxie/beego.git synced 2024-11-06 01:30:53 +00:00

set static file cahce limit:file size & file count

This commit is contained in:
jianzhiyao 2020-05-09 17:57:00 +08:00
parent af19822293
commit 0307c8b110

View File

@ -105,8 +105,15 @@ type serveContentReader struct {
*bytes.Reader
}
const (
//max file size to cache,default: 3m
MaxCacheFileSize int = 1024 * 1024 * 2
//max file count to cache,default: 100
MaxCacheFileCount int = 100
)
var (
staticFileLruCache, _ = lru.New(1000)
staticFileLruCache, _ = lru.New(MaxCacheFileCount)
lruLock sync.RWMutex
)
@ -139,7 +146,9 @@ func openFile(filePath string, fi os.FileInfo, acceptEncoding string) (bool, str
return false, "", nil, nil, err
}
mapFile = &serveContentHolder{data: bufferWriter.Bytes(), modTime: fi.ModTime(), size: int64(bufferWriter.Len()), originSize: fi.Size(), encoding: n}
staticFileLruCache.Add(mapKey, mapFile)
if isOk(mapFile, fi) {
staticFileLruCache.Add(mapKey, mapFile)
}
}
reader := &serveContentReader{Reader: bytes.NewReader(mapFile.data)}
@ -149,6 +158,8 @@ func openFile(filePath string, fi os.FileInfo, acceptEncoding string) (bool, str
func isOk(s *serveContentHolder, fi os.FileInfo) bool {
if s == nil {
return false
} else if s.size > int64(MaxCacheFileSize) {
return false
}
return s.modTime == fi.ModTime() && s.originSize == fi.Size()
}