pmt/db/worker.go

package db

import (
	"flag"
	"fmt"
	"log"
	"os"
	"os/exec"
	"time"
)

// 移除不需要的参数
func strip(slice []string, remove string) []string {
	for i := 0; i < len(slice); {
		if slice[i] == remove {
			if i != len(slice)-1 {
				slice = append(slice[:i], slice[i+1:]...)
			} else {
				slice = slice[:i]
			}
		} else {
			i++
		}
	}
	return slice
}

func subProcess(args []string) *exec.Cmd {
	cmd := exec.Command(args[0], args[1:]...)
	cmd.Stdin = os.Stdin
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	err := cmd.Start()
	if err != nil {
		fmt.Fprintf(os.Stderr, "[-] Error: %s\n", err)
	}
	return cmd
}

func main() {
	daemon := flag.Bool("daemon", false, "run in daemon")
	forever := flag.Bool("forever", false, "run forever")
	flag.Parse()

	// 启用守护模式
	if *daemon {
		subProcess(strip(os.Args, "-daemon"))
		fmt.Printf("[*] Daemon running in pid: %d PPID: %d\n", os.Getpid(), os.Getppid())
		os.Exit(0)
	}

	if *forever {
		for {
			cmd := subProcess(strip(os.Args, "-forever"))
			fmt.Printf("[*] Forever running in pid: %d PPID: %d\n", os.Getpid(), os.Getppid())
			if err := cmd.Wait(); err != nil {
				fmt.Println(err)
				os.Exit(1)
			}
		}
	}

	fmt.Printf("[*] Service running in pid: %d PPID: %d\n", os.Getpid(), os.Getppid())
	fp, _ := os.OpenFile("./dosomething.log", os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
	log.SetOutput(fp)
	for {
		log.Printf("DoSomething running in pid: %d PPID: %d\n", os.Getpid(), os.Getppid())
		time.Sleep(time.Second * 5)
	}
}

func run() {

}