add-go-fuse-to-inject-filesystem-error

Signed-off-by: qiffang <[email protected]>

Author: qiffang

compact_test.go

@@ -16,11 +16,15 @@ package tsdb
 import (
 	"context"
 	"fmt"
+	"github.com/hanwen/go-fuse/fuse"
+	testutil2 "github.com/qiffang/tsdb/testutil"
 	"io/ioutil"
 	"math"
 	"os"
 	"path"
 	"path/filepath"
+	"strings"
+	"syscall"
 	"testing"
 	"time"
 
@@ -1058,3 +1062,87 @@ func TestDeleteCompactionBlockAfterFailedReload(t *testing.T) {
 		})
 	}
 }
+
+func TestCreateBlockWithHook(t *testing.T) {
+	//init acgtion
+	original := filepath.Join(string(filepath.Separator), "tmp", fmt.Sprintf("dev-%d", time.Now().Unix()))
+	mountpoint := filepath.Join(string(filepath.Separator), "tmp", fmt.Sprintf("mountpoint-%d", time.Now().Unix()))
+	server := newFuseServer(t, original, mountpoint)
+	//remember to call unmount after you do not use it
+	defer cleanUp(server, mountpoint, original)
+
+	//normal logic
+	createBlock(t, mountpoint, genSeries(1, 1, 200, 300))
+
+	dir, _ := ioutil.ReadDir(mountpoint)
+	testutil.Equals(t, 0, len(dir))
+}
+
+func TestOpenBlockWithHook(t *testing.T) {
+	//init action
+	original := filepath.Join(string(filepath.Separator), "tmp", fmt.Sprintf("dev-%d", time.Now().Unix()))
+	mountpoint := filepath.Join(string(filepath.Separator), "tmp", fmt.Sprintf("mountpoint-%d", time.Now().Unix()))
+	path := createBlock(t, original, genSeries(1, 1, 200, 300))
+	_, file := filepath.Split(path)
+	server := newFuseServer(t, original, mountpoint)
+	//remember to call unmount after you do not use it
+	defer cleanUp(server, mountpoint, original)
+
+	//normal logic
+	OpenBlock(nil, filepath.Join(mountpoint, file), nil)
+	dir, _  := ioutil.ReadDir(filepath.Join(mountpoint, file))
+
+	testutil.Equals(t, true, len(dir) > 0)
+	hasTempFile := false
+	for _, info := range dir {
+		if strings.HasSuffix(info.Name(), "tmp") {
+			hasTempFile = true
+			break
+		}
+	}
+
+	testutil.Equals(t, true, hasTempFile)
+}
+
+func newFuseServer(t *testing.T, original, mountpoint string)(*fuse.Server){
+	createDirIfAbsent(original)
+	createDirIfAbsent(mountpoint)
+	fs, err :=  testutil2.NewHookFs(original, mountpoint, &TestRenameHook{})
+	testutil.Ok(t, err)
+	server, err := fs.NewServe()
+	if err != nil {
+		fmt.Printf("start server failed, %v \n", err)
+	}
+	testutil.Ok(t, err)
+	go func(){
+		fs.Start(server)
+	}()
+
+	return server
+}
+
+func cleanUp(server *fuse.Server, mountpoint string, original string) {
+	server.Unmount()
+	syscall.Unmount(mountpoint, -1)
+
+	os.RemoveAll(mountpoint)
+	os.RemoveAll(original)
+	fmt.Println("Done")
+}
+
+func createDirIfAbsent(name string) {
+	_, err := os.Stat(name)
+	if err != nil {
+		os.Mkdir(name, os.ModePerm)
+	}
+}
+
+type TestRenameHook struct {}
+
+func (h *TestRenameHook) PreRename(oldPatgh string, newPath string) (hooked bool, err error) {
+	fmt.Printf("renamed file from %s to %s \n", oldPatgh, newPath)
+	return true, syscall.EIO
+}
+func (h *TestRenameHook) PostRename(oldPatgh string, newPath string) (hooked bool, err error) {
+	return false, nil
+}

fileutil/preallocate_darwin.go

@@ -34,7 +34,7 @@ func preallocFixed(f *os.File, sizeInBytes int64) error {
 		Length:  sizeInBytes}
 	p := unsafe.Pointer(fstore)
 	_, _, errno := syscall.Syscall(syscall.SYS_FCNTL, f.Fd(), uintptr(syscall.F_PREALLOCATE), uintptr(p))
-	if errno == 0 || errno == syscall.ENOTSUP {
+	if errno == 0 || errno == syscall.ENOTSUP || errno == syscall.EINVAL {
 		return nil
 	}
 	return errno