Support cache on write

Signed-off-by: Changxin Miao <[email protected]>

cmd/flags.go

@@ -218,6 +218,10 @@ func dataCacheFlags() []cli.Flag {
 			Name:  "cache-partial-only",
 			Usage: "cache only random/small read",
 		},
+		&cli.BoolFlag{
+			Name:  "cache-on-write",
+			Usage: "cache blocks after uploading",
+		},
 		&cli.StringFlag{
 			Name:  "verify-cache-checksum",
 			Value: "full",

cmd/mount.go

@@ -352,6 +352,7 @@ func getChunkConf(c *cli.Context, format *meta.Format) *chunk.Config {
 		FreeSpace:         float32(c.Float64("free-space-ratio")),
 		CacheMode:         os.FileMode(cm),
 		CacheFullBlock:    !c.Bool("cache-partial-only"),
+		CacheOnWrite:      c.Bool("cache-on-write"),
 		CacheChecksum:     c.String("verify-cache-checksum"),
 		CacheEviction:     c.String("cache-eviction"),
 		CacheScanInterval: utils.Duration(c.String("cache-scan-interval")),

pkg/chunk/cached_store.go

@@ -391,7 +391,7 @@ func (store *cachedStore) upload(key string, block *Page, s *wSlice) error {
 		buf.Acquire()
 	}
 	defer buf.Release()
-	if sync && blen < store.conf.BlockSize {
+	if sync && (blen < store.conf.BlockSize || store.conf.CacheOnWrite) {
 		// block will be freed after written into disk
 		store.bcache.cache(key, block, false, false)
 	}
@@ -568,6 +568,7 @@ type Config struct {
 	GetTimeout        time.Duration
 	PutTimeout        time.Duration
 	CacheFullBlock    bool
+	CacheOnWrite      bool
 	BufferSize        uint64
 	Readahead         int
 	Prefetch          int