Merge branch 'disk-cache'

cmd/reflector.go

@@ -18,12 +18,15 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var reflectorCmdCacheDir string
+
 func init() {
 	var cmd = &cobra.Command{
 		Use:   "reflector",
 		Short: "Run reflector server",
 		Run:   reflectorCmd,
 	}
+	cmd.Flags().StringVar(&reflectorCmdCacheDir, "cache", "", "Enable disk cache for blobs. Store them in this directory")
 	rootCmd.AddCommand(cmd)
 }
 
@@ -60,6 +63,14 @@ func reflectorCmd(cmd *cobra.Command, args []string) {
 		}
 	}
 
+	if reflectorCmdCacheDir != "" {
+		err = os.MkdirAll(reflectorCmdCacheDir, os.ModePerm)
+		if err != nil {
+			log.Fatal(err)
+		}
+		blobStore = store.NewCachingBlobStore(blobStore, store.NewDiskBlobStore(reflectorCmdCacheDir, 2))
+	}
+
 	peerServer := peer.NewServer(blobStore)
 	err = peerServer.Start(":5567")
 	if err != nil {

store/disk.go

@@ -4,9 +4,14 @@ import (
 	"io/ioutil"
 	"os"
 	"path"
+	"sort"
+	"syscall"
+	"time"
 
 	"github.com/lbryio/lbry.go/v2/extras/errors"
 	"github.com/lbryio/lbry.go/v2/stream"
+
+	log "github.com/sirupsen/logrus"
 )
 
 // DiskBlobStore stores blobs on a local disk
@@ -17,11 +22,15 @@ type DiskBlobStore struct {
 	prefixLength int
 
 	initialized     bool
+	lastChecked     time.Time
+	diskCleanupBusy chan bool
 }
 
 // NewDiskBlobStore returns an initialized file disk store pointer.
 func NewDiskBlobStore(dir string, prefixLength int) *DiskBlobStore {
-	return &DiskBlobStore{blobDir: dir, prefixLength: prefixLength}
+	dbs := DiskBlobStore{blobDir: dir, prefixLength: prefixLength, diskCleanupBusy: make(chan bool, 1)}
+	dbs.diskCleanupBusy <- true
+	return &dbs
 }
 
 func (d *DiskBlobStore) dir(hash string) string {
@@ -31,6 +40,21 @@ func (d *DiskBlobStore) dir(hash string) string {
 	return path.Join(d.blobDir, hash[:d.prefixLength])
 }
 
+// GetUsedSpace returns a value between 0 and 1, with 0 being completely empty and 1 being full, for the disk that holds the provided path
+func (d *DiskBlobStore) getUsedSpace() (float32, error) {
+	var stat syscall.Statfs_t
+	err := syscall.Statfs(d.blobDir, &stat)
+	if err != nil {
+		return 0, err
+	}
+	// Available blocks * size per block = available space in bytes
+	all := stat.Blocks * uint64(stat.Bsize)
+	free := stat.Bfree * uint64(stat.Bsize)
+	used := all - free
+
+	return float32(used) / float32(all), nil
+}
+
 func (d *DiskBlobStore) path(hash string) string {
 	return path.Join(d.dir(hash), hash)
 }
@@ -99,6 +123,14 @@ func (d *DiskBlobStore) Put(hash string, blob stream.Blob) error {
 	if err != nil {
 		return err
 	}
+	select {
+	case <-d.diskCleanupBusy:
+		if time.Since(d.lastChecked) > 5*time.Minute {
+			go d.ensureDiskSpace()
+		}
+	default:
+		break
+	}
 
 	return ioutil.WriteFile(d.path(hash), blob, 0644)
 }
@@ -125,3 +157,81 @@ func (d *DiskBlobStore) Delete(hash string) error {
 
 	return os.Remove(d.path(hash))
 }
+
+func (d *DiskBlobStore) ensureDiskSpace() {
+	defer func() {
+		d.lastChecked = time.Now()
+		d.diskCleanupBusy <- true
+	}()
+
+	used, err := d.getUsedSpace()
+	if err != nil {
+		log.Errorln(err.Error())
+		return
+	}
+	log.Infof("disk usage: %.2f%%\n", used*100)
+	if used > 0.90 {
+		log.Infoln("over 0.90, cleaning up")
+		err = d.WipeOldestBlobs()
+		if err != nil {
+			log.Errorln(err.Error())
+			return
+		}
+		log.Infoln("Done cleaning up")
+	}
+}
+
+func (d *DiskBlobStore) WipeOldestBlobs() (err error) {
+	dirs, err := ioutil.ReadDir(d.blobDir)
+	if err != nil {
+		return err
+	}
+	type datedFile struct {
+		Atime    time.Time
+		File     *os.FileInfo
+		FullPath string
+	}
+	datedFiles := make([]datedFile, 0, 500)
+	for _, dir := range dirs {
+		if dir.IsDir() {
+			files, err := ioutil.ReadDir(d.blobDir + "/" + dir.Name())
+			if err != nil {
+				return err
+			}
+			for _, file := range files {
+				if file.Mode().IsRegular() && !file.IsDir() {
+					datedFiles = append(datedFiles, datedFile{
+						Atime:    atime(file),
+						File:     &file,
+						FullPath: d.blobDir + "/" + dir.Name() + "/" + file.Name(),
+					})
+				}
+			}
+		}
+	}
+
+	sort.Slice(datedFiles, func(i, j int) bool {
+		return datedFiles[i].Atime.Before(datedFiles[j].Atime)
+	})
+	//delete the first 50000 blobs
+	for i, df := range datedFiles {
+		if i >= 50000 {
+			break
+		}
+		log.Infoln(df.FullPath)
+		log.Infoln(df.Atime.String())
+		err = os.Remove(df.FullPath)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func timespecToTime(ts syscall.Timespec) time.Time {
+	return time.Unix(int64(ts.Sec), int64(ts.Nsec))
+}
+
+func atime(fi os.FileInfo) time.Time {
+	return timespecToTime(fi.Sys().(*syscall.Stat_t).Atim)
+}