From 9a4fb14ea0674c09892157220560236d69b8218e Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Fri, 10 May 2019 17:08:15 -0700
Subject: [PATCH] weedfuse: daemonize

---
 weed/command/mount_std.go         |  6 +++++
 weed/command/weedfuse/weedfuse.go | 37 ++++++++++++++++++++++++++++++-
 2 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/weed/command/mount_std.go b/weed/command/mount_std.go
index 66aa613d5..36c1f97a3 100644
--- a/weed/command/mount_std.go
+++ b/weed/command/mount_std.go
@@ -14,6 +14,7 @@ import (
 
 	"github.com/chrislusf/seaweedfs/weed/security"
 	"github.com/chrislusf/seaweedfs/weed/server"
+	"github.com/jacobsa/daemonize"
 	"github.com/spf13/viper"
 
 	"github.com/chrislusf/seaweedfs/weed/filesys"
@@ -107,6 +108,7 @@ func RunMount(filer, filerMountRootPath, dir, collection, replication, dataCente
 	c, err := fuse.Mount(dir, options...)
 	if err != nil {
 		glog.Fatal(err)
+		daemonize.SignalOutcome(err)
 		return false
 	}
 
@@ -118,6 +120,7 @@ func RunMount(filer, filerMountRootPath, dir, collection, replication, dataCente
 	filerGrpcAddress, err := parseFilerGrpcAddress(filer)
 	if err != nil {
 		glog.Fatal(err)
+		daemonize.SignalOutcome(err)
 		return false
 	}
 
@@ -126,6 +129,8 @@ func RunMount(filer, filerMountRootPath, dir, collection, replication, dataCente
 		mountRoot = mountRoot[0 : len(mountRoot)-1]
 	}
 
+	daemonize.SignalOutcome(nil)
+
 	err = fs.Serve(c, filesys.NewSeaweedFileSystem(&filesys.Option{
 		FilerGrpcAddress:   filerGrpcAddress,
 		GrpcDialOption:     security.LoadClientTLS(viper.Sub("grpc"), "client"),
@@ -151,6 +156,7 @@ func RunMount(filer, filerMountRootPath, dir, collection, replication, dataCente
 	<-c.Ready
 	if err := c.MountError; err != nil {
 		glog.Fatal(err)
+		daemonize.SignalOutcome(err)
 	}
 
 	return true
diff --git a/weed/command/weedfuse/weedfuse.go b/weed/command/weedfuse/weedfuse.go
index 3b7789eb1..9e9c0505c 100644
--- a/weed/command/weedfuse/weedfuse.go
+++ b/weed/command/weedfuse/weedfuse.go
@@ -7,10 +7,14 @@ import (
 	"strings"
 
 	"github.com/chrislusf/seaweedfs/weed/command"
+	"github.com/chrislusf/seaweedfs/weed/glog"
+	"github.com/kardianos/osext"
+	"github.com/jacobsa/daemonize"
 )
 
 var (
-	options = flag.String("o", "", "comma separated options rw,uid=xxx,gid=xxx")
+	options      = flag.String("o", "", "comma separated options rw,uid=xxx,gid=xxx")
+	isForeground = flag.Bool("foreground", false, "starts as a daemon")
 )
 
 func main() {
@@ -25,6 +29,11 @@ func main() {
 
 	maybeSetupPath()
 
+	if !*isForeground {
+		startAsDaemon()
+		return
+	}
+
 	parts := strings.SplitN(device, "/", 2)
 	filer, filerPath := parts[0], parts[1]
 
@@ -47,3 +56,29 @@ func maybeSetupPath() {
 		os.Setenv("PATH", "/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin")
 	}
 }
+
+func startAsDaemon() {
+
+	// adapted from gcsfuse
+
+	// Find the executable.
+	var path string
+	path, err := osext.Executable()
+	if err != nil {
+		glog.Fatalf("osext.Executable: %v", err)
+	}
+
+	// Set up arguments. Be sure to use foreground mode.
+	args := append([]string{"-foreground"}, os.Args[1:]...)
+
+	// Pass along PATH so that the daemon can find fusermount on Linux.
+	env := []string{
+		fmt.Sprintf("PATH=%s", os.Getenv("PATH")),
+	}
+
+	err = daemonize.Run(path, args, env, os.Stdout)
+	if err != nil {
+		glog.Fatalf("daemonize.Run: %v", err)
+	}
+
+}