Allow self-defined stage directory

FileHandleWriter.go

@@ -34,9 +34,8 @@ func NewFileHandleWriter(handle *FileHandle, newFile bool) (*FileHandleWriter, e
 		}
 		w.Close()
 	}
-	stageDir := "/var/hdfs-mount" // TODO: make configurable
 	if ok := os.MkdirAll(stageDir, 0700); ok != nil {
-		Error.Println("Failed to create stageDir /var/hdfs-mount, Error:", ok)
+		Error.Println("Failed to create stageDir", stageDir, ", Error:", ok)
 		return nil, ok
 	}
 	var err error

FileHandleWriter_test.go

@@ -3,13 +3,18 @@ package main
 import (
 	"bazil.org/fuse"
 	"errors"
+	"flag"
 	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/assert"
 	"io"
 	"os"
 	"testing"
 )
 
+func init() {
+	flag.StringVar(&stageDir, "stageDir", "/var/hdfs-mount", "set stage dir for test")
+}
+
 func TestWriteFile(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 	mockClock := &MockClock{}

main.go

@@ -22,6 +22,8 @@ var Usage = func() {
 	flag.PrintDefaults()
 }
 
+var stageDir string
+
 func main() {
 	sigs := make(chan os.Signal, 1)
 	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
@@ -37,6 +39,7 @@ func main() {
 	expandZips := flag.Bool("expandZips", false, "Enables automatic expansion of ZIP archives")
 	readOnly := flag.Bool("readOnly", false, "Enables mount with readonly")
 	logLevel := flag.Int("logLevel", 0, "logs to be printed. 0: only fatal/err logs; 1: +warning logs; 2: +info logs")
+	flag.StringVar(&stageDir, "stageDir", "/var/hdfs-mount", "stage directory for writing file")
 
 	flag.Usage = Usage
 	flag.Parse()