mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
chore: minor performance improvements
This commit is contained in:
parent
80252ff701
commit
e14f8c1463
14
main.go
14
main.go
@ -247,14 +247,12 @@ func main() {
|
|||||||
// https://github.com/golang/go/issues/43431
|
// https://github.com/golang/go/issues/43431
|
||||||
embeddedStatic, _ := fs.Sub(staticFiles, "static")
|
embeddedStatic, _ := fs.Sub(staticFiles, "static")
|
||||||
static := conf.ChooseFS("static", embeddedStatic)
|
static := conf.ChooseFS("static", embeddedStatic)
|
||||||
|
fileServer := gzipped.FileServer(fsutils.NewExistsHttpFS(
|
||||||
fileServer := gzipped.FileServer(fsutils.NewExistsHttpFs(
|
fsutils.NewExistsFS(
|
||||||
fsutils.ExistsFS{
|
fsutils.NeuteredFileSystem{
|
||||||
Fs: fsutils.NeuteredFileSystem{
|
FS: static,
|
||||||
Fs: static,
|
}).WithCache(!config.IsDev()),
|
||||||
},
|
))
|
||||||
}),
|
|
||||||
)
|
|
||||||
|
|
||||||
router.PathPrefix("/contribute.json").Handler(fileServer)
|
router.PathPrefix("/contribute.json").Handler(fileServer)
|
||||||
router.PathPrefix("/assets").Handler(fileServer)
|
router.PathPrefix("/assets").Handler(fileServer)
|
||||||
|
@ -1,35 +1,59 @@
|
|||||||
package fs
|
package fs
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/patrickmn/go-cache"
|
||||||
"io/fs"
|
"io/fs"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func NewExistsFS(fs fs.FS) ExistsFS {
|
||||||
|
return ExistsFS{
|
||||||
|
FS: fs,
|
||||||
|
cache: cache.New(cache.NoExpiration, cache.NoExpiration),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type ExistsFS struct {
|
type ExistsFS struct {
|
||||||
Fs fs.FS
|
fs.FS
|
||||||
|
UseCache bool
|
||||||
|
cache *cache.Cache
|
||||||
|
}
|
||||||
|
|
||||||
|
func (efs ExistsFS) WithCache(withCache bool) ExistsFS {
|
||||||
|
efs.UseCache = withCache
|
||||||
|
return efs
|
||||||
}
|
}
|
||||||
|
|
||||||
func (efs ExistsFS) Exists(name string) bool {
|
func (efs ExistsFS) Exists(name string) bool {
|
||||||
_, err := fs.Stat(efs.Fs, name)
|
if efs.UseCache {
|
||||||
return err == nil
|
if result, ok := efs.cache.Get(name); ok {
|
||||||
|
return result.(bool)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err := fs.Stat(efs.FS, name)
|
||||||
|
result := err == nil
|
||||||
|
if efs.UseCache {
|
||||||
|
efs.cache.SetDefault(name, result)
|
||||||
|
}
|
||||||
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
func (efs ExistsFS) Open(name string) (fs.File, error) {
|
func (efs ExistsFS) Open(name string) (fs.File, error) {
|
||||||
return efs.Fs.Open(name)
|
return efs.FS.Open(name)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ---
|
// ---
|
||||||
|
|
||||||
type ExistsHttpFS struct {
|
type ExistsHttpFS struct {
|
||||||
Fs ExistsFS
|
ExistsFS
|
||||||
httpFs http.FileSystem
|
httpFs http.FileSystem
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewExistsHttpFs(fs ExistsFS) ExistsHttpFS {
|
func NewExistsHttpFS(fs ExistsFS) ExistsHttpFS {
|
||||||
return ExistsHttpFS{
|
return ExistsHttpFS{
|
||||||
Fs: fs,
|
ExistsFS: fs,
|
||||||
httpFs: http.FS(fs),
|
httpFs: http.FS(fs),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -37,7 +61,7 @@ func (ehfs ExistsHttpFS) Exists(name string) bool {
|
|||||||
if strings.HasPrefix(name, "/") {
|
if strings.HasPrefix(name, "/") {
|
||||||
name = name[1:]
|
name = name[1:]
|
||||||
}
|
}
|
||||||
return ehfs.Fs.Exists(name)
|
return ehfs.ExistsFS.Exists(name)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ehfs ExistsHttpFS) Open(name string) (http.File, error) {
|
func (ehfs ExistsHttpFS) Open(name string) (http.File, error) {
|
||||||
|
@ -8,11 +8,11 @@ import (
|
|||||||
// https://www.alexedwards.net/blog/disable-http-fileserver-directory-listings
|
// https://www.alexedwards.net/blog/disable-http-fileserver-directory-listings
|
||||||
|
|
||||||
type NeuteredFileSystem struct {
|
type NeuteredFileSystem struct {
|
||||||
Fs fs.FS
|
fs.FS
|
||||||
}
|
}
|
||||||
|
|
||||||
func (nfs NeuteredFileSystem) Open(path string) (fs.File, error) {
|
func (nfs NeuteredFileSystem) Open(path string) (fs.File, error) {
|
||||||
f, err := nfs.Fs.Open(path)
|
f, err := nfs.FS.Open(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@ -20,7 +20,7 @@ func (nfs NeuteredFileSystem) Open(path string) (fs.File, error) {
|
|||||||
s, err := f.Stat()
|
s, err := f.Stat()
|
||||||
if s.IsDir() {
|
if s.IsDir() {
|
||||||
index := filepath.Join(path, "index.html")
|
index := filepath.Join(path, "index.html")
|
||||||
if _, err := nfs.Fs.Open(index); err != nil {
|
if _, err := nfs.FS.Open(index); err != nil {
|
||||||
closeErr := f.Close()
|
closeErr := f.Close()
|
||||||
if closeErr != nil {
|
if closeErr != nil {
|
||||||
return nil, closeErr
|
return nil, closeErr
|
||||||
|
Loading…
Reference in New Issue
Block a user