added directory cleanup

This commit is contained in:
ston1th 2022-03-17 21:10:23 +01:00
commit 74ab10358a
5 changed files with 87 additions and 22 deletions

View file

@ -88,4 +88,5 @@ func main() {
} }
filesystem.Close() filesystem.Close()
log.Info("cachefs shutdown completed") log.Info("cachefs shutdown completed")
os.Exit(0)
} }

View file

@ -8,6 +8,8 @@ import (
"os" "os"
"sync" "sync"
"time" "time"
"github.com/go-logr/logr"
) )
var NoCacheEntry = errors.New("no cache entry") var NoCacheEntry = errors.New("no cache entry")
@ -42,6 +44,7 @@ func (dc *DirCache) Set(name string, dce *DirCacheEntry) {
} }
type Dir struct { type Dir struct {
log logr.Logger
f *os.File f *os.File
dc *DirCache dc *DirCache
} }
@ -54,6 +57,19 @@ func (f *Dir) Seek(_ int64, _ int) (int64, error) {
return 0, io.EOF return 0, io.EOF
} }
func readDir(f *os.File) (fi []os.FileInfo, err error) {
ent, err := f.ReadDir(0)
if err != nil {
return
}
fi = make([]os.FileInfo, len(ent))
for j, e := range ent {
i, _ := e.Info()
fi[j] = i
}
return
}
func (f *Dir) Readdir(_ int) (fi []os.FileInfo, err error) { func (f *Dir) Readdir(_ int) (fi []os.FileInfo, err error) {
i, err := f.Stat() i, err := f.Stat()
if err != nil { if err != nil {
@ -61,13 +77,17 @@ func (f *Dir) Readdir(_ int) (fi []os.FileInfo, err error) {
} }
name := f.f.Name() name := f.f.Name()
dce, err := f.dc.Get(name) dce, err := f.dc.Get(name)
log := f.log
if err == NoCacheEntry || i.ModTime().After(dce.mtime) { if err == NoCacheEntry || i.ModTime().After(dce.mtime) {
fi, err = f.f.Readdir(0) log.V(3).Info("cache miss")
fi, err = readDir(f.f)
if err != nil { if err != nil {
return return
} }
dce = &DirCacheEntry{i.ModTime(), fi} dce = &DirCacheEntry{i.ModTime(), fi}
f.dc.Set(name, dce) f.dc.Set(name, dce)
} else {
log.V(3).Info("cache hit")
} }
fi = dce.fi fi = dce.fi
return return

View file

@ -205,7 +205,7 @@ func (fs *FS) Open(name string) (f http.File, err error) {
offline = true offline = true
log.V(2).Info("dir offline mode", "path", dp) log.V(2).Info("dir offline mode", "path", dp)
} }
f = &Dir{f: sf, dc: fs.dc} f = &Dir{log: log.WithName("dir"), f: sf, dc: fs.dc}
return return
} }
md := fs.metadata(name, sfi.Size()) md := fs.metadata(name, sfi.Size())
@ -295,10 +295,32 @@ func (fs *FS) flusher(ctx context.Context) {
return return
case <-time.After(time.Minute * 5): case <-time.After(time.Minute * 5):
} }
fs.cleanupEmptyDirs()
fs.flushMetadata() fs.flushMetadata()
} }
} }
func (fs *FS) cleanupEmptyDirs() {
filepath.WalkDir(fs.dst, func(path string, d stdfs.DirEntry, err error) error {
if err == nil && d.IsDir() {
if path == fs.dst {
return nil
}
empty, err := dirEmpty(path)
if err != nil || !empty {
return nil
}
log := fs.log.WithValues("dir", path)
log.V(2).Info("removing empty dir")
err = os.Remove(path)
if err != nil {
log.Error(err, "failed to remove empty dir")
}
}
return nil
})
}
func (fs *FS) flushMetadata() { func (fs *FS) flushMetadata() {
log := fs.log log := fs.log
fs.mu.Lock() fs.mu.Lock()
@ -341,10 +363,11 @@ func dirEmpty(name string) (bool, error) {
if err != nil { if err != nil {
return false, err return false, err
} }
defer f.Close() _, err = f.ReadDir(1)
_, err = f.Readdir(1)
if err == io.EOF { if err == io.EOF {
f.Close()
return true, nil return true, nil
} }
f.Close()
return false, err return false, err
} }

View file

@ -5,8 +5,8 @@ package srv
import ( import (
"bytes" "bytes"
"encoding/xml" "encoding/xml"
"html/template"
"net/http" "net/http"
"sort"
"strings" "strings"
"cachefs/pkg/fs" "cachefs/pkg/fs"
@ -41,12 +41,22 @@ func (s *statusInterceptor) Status() int {
type dirContents struct { type dirContents struct {
AllPaths []string `xml:"a"` AllPaths []string `xml:"a"`
Dirs []string `xml:"-"` Dirs dirs `xml:"-"`
Files files `xml:"-"` Files files `xml:"-"`
} }
type dir struct {
Name template.HTML
URI template.HTML
}
type dirs []dir
func (dirs) Less(i, j dir) bool { return i.Name < j.Name }
type file struct { type file struct {
Name string Name template.HTML
URI template.HTML
Anchor string Anchor string
Status int Status int
} }
@ -80,25 +90,36 @@ func (r *responseInterceptor) Status() int {
return r.status return r.status
} }
func (r *responseInterceptor) GetPaths(path string, fs *fs.FS) (dir dirContents, err error) { func (r *responseInterceptor) GetPaths(path string, fs *fs.FS) (dc dirContents, err error) {
err = xml.Unmarshal(r.buf.Bytes(), &dir) buf := r.buf.Bytes()
buf = bytes.ReplaceAll(buf, []byte{'&'}, []byte("&#38;"))
err = xml.Unmarshal(buf, &dc)
if err != nil { if err != nil {
return return
} }
path = strings.TrimSuffix(path, "/") path = strings.TrimSuffix(path, "/")
for _, p := range dir.AllPaths { for _, p := range dc.AllPaths {
p = strings.ReplaceAll(p, "&amp;", "&")
p = strings.ReplaceAll(p, "&#39;", "'")
name := template.HTML(p)
full := path + "/" + p
uri := template.HTML(full)
if p[len(p)-1] == '/' { if p[len(p)-1] == '/' {
dir.Dirs = append(dir.Dirs, p) dc.Dirs = append(dc.Dirs, dir{
Name: name,
URI: uri,
})
} else { } else {
dir.Files = append(dir.Files, file{ dc.Files = append(dc.Files, file{
Name: p, Name: name,
URI: uri,
Anchor: anchor(p), Anchor: anchor(p),
Status: fs.CacheStatus(path + "/" + p), Status: fs.CacheStatus(full),
}) })
} }
} }
sort.Strings(dir.Dirs) slices.SortFunc(dc.Dirs, dc.Dirs.Less)
slices.SortFunc(dir.Files, dir.Files.Less) slices.SortFunc(dc.Files, dc.Files.Less)
return return
} }

View file

@ -24,13 +24,13 @@ var (
</pre> </pre>
<table> <table>
<tr><th style="width:100%;">Path</th><th>Options</th></tr> <tr><th style="width:100%;">Path</th><th>Options</th></tr>
<tr><td><a href="../">../</a></td><td></td></tr> <tr><td><a href="../">../</a></td><td>[dir]</td></tr>
{{range $s := .Dirs -}} {{range $s := .Dirs -}}
<tr><td><a href="{{$s}}">{{$s}}</a></td><td></td></tr> <tr><td><a href="{{$s.URI}}">{{$s.Name}}</a></td><td>[dir]</td></tr>
{{end -}} {{end -}}
{{range $s := .Files -}} {{range $s := .Files -}}
<tr><td><a id="{{$s.Anchor}}" href="{{$s.Name}}">{{$s.Name}}</a></td> <tr><td><a id="{{$s.Anchor}}" href="{{$s.URI}}">{{$s.Name}}</a></td>
<td><a href="{{$s.Name}}?o=v">[v]</a>&nbsp;<a href="{{$s.Name}}?o=n">[n]</a>&nbsp;<a href="{{$s.Name}}?o=p">[p]</a>&nbsp;<a href="{{$s.Name}}?o=s">[s]</a>&nbsp;{{if ne $s.Status -1}}{{$s.Status}}%{{end}}</td></tr> <td><a href="{{$s.URI}}?o=v">[v]</a>&nbsp;<a href="{{$s.URI}}?o=n">[n]</a>&nbsp;<a href="{{$s.URI}}?o=p">[p]</a>&nbsp;<a href="{{$s.URI}}?o=s">[s]</a>&nbsp;{{if ne $s.Status -1}}{{$s.Status}}%{{end}}</td></tr>
{{end -}} {{end -}}
</table> </table>
</article> </article>