Question

我尝试添加必要的代码来执行我的应用程序，就像一个守护进程。我用了下一个项目：

github.com/sevlyar/go-daemon

我重写了它完成的示例代码： https://github.com/sevlyar/go-daemon/blob/master/sample/sample.go

package main

import (
    "bufio"
    "flag"
    "fmt"
    "io/ioutil"
    "os"
    "syscall"
    "time"

    "github.com/sevlyar/go-daemon"
)

var (
    signal = flag.String("s", "", `sdaemon -s ...
        quit -- graceful shutdown`)
)

var (
    stop = make(chan struct{})
    done = make(chan struct{})
)

func main() {
    flag.Parse()
    daemon.AddCommand(daemon.StringFlag(signal, "quit"), syscall.SIGQUIT, TermHandler)

    cntxt := &daemon.Context{
        PidFileName: "/var/run/sdaemon.pid",
        PidFilePerm: 0644,
        WorkDir:     "./",
        Umask:       027,
        Args:        []string{"[sdaemon]"},
    }
    if len(daemon.ActiveFlags()) > 0 {
        d, _ := cntxt.Search()
        daemon.SendCommands(d)
        return
    }
    d, err := cntxt.Reborn()
    if d != nil {
        return
    }
    if err != nil {
        os.Exit(1)
    }
    defer cntxt.Release()

    // Start daemon
    go Worker()

    err = daemon.ServeSignals()
    if err != nil {
        fmt.Printf("STOPPED!\n")
        return
    }
}

func Worker() {
    for {
        go Writer()
        if _, ok := <-stop; ok {
            break
        }
    }
    done <- struct{}{}
}

func TermHandler(sig os.Signal) error {
    stop <- struct{}{}
    if sig == syscall.SIGQUIT {
        <-done
    }
    return daemon.ErrStop
}

我添加了一个函数Writer()来读取文件，将文本保存为字符串并使用此字符串创建一个新文件。

func Writer() error {
    time.Sleep(time.Minute)

    f, _ := ioutil.ReadFile("$HOME/test")
    contents := string(f)

    fileHandle, _ := os.Create("$HOME/stest")
    writer := bufio.NewWriter(fileHandle)
    defer fileHandle.Close()
    fmt.Fprintln(writer, contents)
    writer.Flush()

    return nil
}

我没有处理golang中的通道这么好，我不知道为什么Worker()函数中的无限循环只执行一次......

你能帮我吗？

Answer 1

当您尝试检查Worker频道中是否有任何数据时，问题出在done功能中。 receive调用将阻塞，直到有值要读取，这样调用才会阻塞，直到您向进程发送信号为止。

接收运算符ok返回的第二个值不表示是否成功读取了值。它只是指示在尝试接收值时通道是否已关闭（如果是，则返回zero value，请参阅the specification）。

要检查频道中是否有值，您需要使用select语句，如下所示：

select {
        case v, ok <- stop:
            // We could read a value from the channel
        default:
            // No value could be read, but we didn't block
}

因此，您的Worker函数应该类似于：

func Worker() {
    for {
        time.Sleep(time.Minute)
        select {
        case <- stop:
            // Got a stop signal, stopping
            done <- struct{}{}
            return
        default:
            // No stop signal, continuing loop
        }

        go Writer()
    }
}

请注意，我已将Sleep从Writer函数移至Worker，否则您最终会遇到数千个并发Writer个例行程序...

守护进程只执行一次goroutine

1 个答案: