rudi
/
headless9
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update logging

This commit is contained in:
Rudi 2022-10-23 01:51:50 -04:00
parent d35d6d9ea4
commit 5106303181
Signed by: rudi
GPG Key ID: EF64F3CBD1A1EBDD
1 changed files with 34 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
main.go
View File

@ -13,7 +13,6 @@ import (
)
var (
daemon = false
serviceFile = "/adm/services"
services = make(map[string]Service)
controlSocket = "/adm/headless9/ctl/headless9.ctl"
@ -23,12 +22,16 @@ var (
func main() {
flag.Parse()
if flag.NArg() == 0 {
logFile, err := os.Create(fmt.Sprintf("%+v/%+v.log", logPath, "headless9"))
if err != nil {
panic(err)
}
log.SetOutput(logFile)
log.Println(DAEMON_START)
daemon = true
runDaemon()
}
if flag.NArg() != 2 {
fmt.Println(CMD_SYNTAX)
log.Println(CMD_SYNTAX)
return
}
verb := flag.Args()[0]
@ -37,41 +40,33 @@ func main() {
f := getCtl()
_, err := fmt.Fprintf(f, "%+v %+v", verb, service)
if err != nil {
fmt.Printf(CTL_UNABLE_WRITE, controlSocket)
log.Fatalf(CTL_UNABLE_WRITE, controlSocket)
}
f.Close()
err = watchFile(controlSocket)
if err != nil {
log.Println(err)
}
fmt.Println(strings.Join(sysTail(10, controlSocket), "\n"))
fmt.Println(sysTail(10, controlSocket))
}
func getCtl() *os.File {
f, err := os.OpenFile(controlSocket, os.O_TRUNC, 0660)
if err != nil {
if daemon {
log.Printf(CTL_NOT_OPEN, controlSocket)
} else {
fmt.Printf(CTL_NOT_OPEN, controlSocket)
}
log.Fatalf(CTL_NOT_OPEN, controlSocket)
}
err = f.Truncate(0)
if err != nil {
if daemon {
log.Printf(CTL_NOT_CLEAR, controlSocket)
} else {
fmt.Printf(CTL_NOT_CLEAR, controlSocket)
}
log.Fatalf(CTL_NOT_CLEAR, controlSocket)
}
_, err = f.Seek(0, 0)
if err != nil {
if daemon {
log.Printf(CTL_NOT_REWOUND, controlSocket)
} else {
fmt.Printf(CTL_NOT_REWOUND, controlSocket)
}
log.Fatalf(CTL_NOT_REWOUND, controlSocket)
}
f, err = os.OpenFile(controlSocket, os.O_RDWR, 0660)
if err != nil {
log.Fatalf(CTL_NOT_OPEN, controlSocket)
}
return f
}
@ -142,18 +137,23 @@ func processCommand(cmd string) error {
verb := parts[0]
svc := parts[1]
start := time.Now()
messages := 0
if verb == "restart" || verb == "stop" {
f, err := os.OpenFile(fmt.Sprintf("/proc/%d/ctl", services[svc].ProcessHandle.Pid), os.O_WRONLY, 0660)
if err != nil {
log.Printf("")
log.Printf(DAEMON_CTL_FAILED, err, time.Since(start))
messages++
}
defer f.Close()
_, err = f.WriteString("kill")
if err != nil {
log.Printf(DAEMON_CTL_FAILED, err, time.Since(start))
messages++
} else {
delete(services, svc)
log.Printf(DAEMON_CTL_PROCESSED, svc, time.Since(start))
messages++
}
}
if verb == "restart" || verb == "start" {
@ -168,10 +168,22 @@ func processCommand(cmd string) error {
}
log.Printf(DAEMON_SVC_MISSING, svcArgs[0])
messages++
go execCommand(svcArgs[0], svcArgs[1:]...)
}
}
ctlOut := ""
if verb == "status" {
ctlOut = strings.Join(sysTail(10, fmt.Sprintf("%+v/%+v.log", logPath, svc)), "\n")
} else {
ctlOut = strings.Join(sysTail(messages, fmt.Sprintf("%+v/%+v.log", logPath, svc)), "\n")
}
f := getCtl()
_, err := fmt.Fprintf(f, "%+v", ctlOut)
if err != nil {
log.Printf(CTL_UNABLE_WRITE, controlSocket)
}
f.Close()
return nil
}
@ -264,7 +276,6 @@ func PanicSafe(a ...interface{}) {
func sysTail(count int, path string) []string {
c := exec.Command("tail", fmt.Sprintf("-%d", count+1), path)
output, _ := c.Output()
//log.Printf("SysTail call output: %+v\nEND", string(output))
lines := strings.Split(string(output), "\n")
return lines[:len(lines)-1]