go-sylixos/elvish
3
0
Fork 0
mirror of https://github.com/go-sylixos/elvish.git synced 2024-12-14 19:27:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Handle -cpuprofile and -log{,prefix} for all subprograms.

Browse Source
This commit is contained in:
Qi Xiao 2017-12-06 23:27:34 +00:00
parent fab99616fb
commit 6422519bdb
1 changed files with 22 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
program/program.go
View File

@ -68,10 +68,31 @@ func usage() {
}
func Main() int {
// Parse and check flags.
flag.Usage = usage
flag.Parse()
args := flag.Args()
// Handle flags common to all subprograms.
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
}
var err error
if *logpath != "" {
err = util.SetOutputFile(*logpath)
} else if *logpathprefix != "" {
err = util.SetOutputFile(*logpathprefix + strconv.Itoa(os.Getpid()))
}
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
return FindProgram(args).Main(args)
}
@ -106,33 +127,6 @@ func FindProgram(args []string) Program {
type SubPrograms struct{}
func (SubPrograms) Main(args []string) int {
// Flags common to all sub-programs: log and CPU profile.
if *isdaemon {
if *forked == 2 && *logpathprefix != "" {
// Honor logpathprefix.
pid := os.Getpid()
err := util.SetOutputFile(*logpathprefix + strconv.Itoa(pid))
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
} else {
util.SetOutputFile("/dev/stderr")
}
} else if *logpath != "" {
err := util.SetOutputFile(*logpath)
if err != nil {
fmt.Fprintln(os.Stderr, err)
}
}
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
}
// Pick a sub-program to run.
if *isdaemon {
d := daemon.Daemon{