mirror of
https://github.com/ncarlier/webhookd.git
synced 2025-04-12 05:35:11 +00:00
113 lines
3.0 KiB
Go
113 lines
3.0 KiB
Go
package worker
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/ncarlier/webhookd/pkg/logger"
|
|
"github.com/ncarlier/webhookd/pkg/tools"
|
|
)
|
|
|
|
// ChanWriter is a simple writer to a channel of byte.
|
|
type ChanWriter struct {
|
|
ByteChan chan []byte
|
|
}
|
|
|
|
func (c *ChanWriter) Write(p []byte) (int, error) {
|
|
c.ByteChan <- p
|
|
return len(p), nil
|
|
}
|
|
|
|
var (
|
|
workingdir = os.Getenv("APP_WORKING_DIR")
|
|
)
|
|
|
|
func runScript(work *WorkRequest) (string, error) {
|
|
if workingdir == "" {
|
|
workingdir = os.TempDir()
|
|
}
|
|
|
|
logger.Info.Printf("Work %s#%d started...\n", work.Name, work.ID)
|
|
logger.Debug.Printf("Work %s#%d script: %s\n", work.Name, work.ID, work.Script)
|
|
logger.Debug.Printf("Work %s#%d parameter: %v\n", work.Name, work.ID, work.Args)
|
|
|
|
binary, err := exec.LookPath(work.Script)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Exec script with args...
|
|
cmd := exec.Command(binary, work.Payload)
|
|
// with env variables...
|
|
cmd.Env = append(os.Environ(), work.Args...)
|
|
// using a process group...
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
|
|
|
|
// Open the out file for writing
|
|
logFilename := path.Join(workingdir, fmt.Sprintf("%s_%d_%s.txt", tools.ToSnakeCase(work.Name), work.ID, time.Now().Format("20060102_1504")))
|
|
logFile, err := os.Create(logFilename)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer logFile.Close()
|
|
logger.Debug.Printf("Work %s#%d output to file: %s\n", work.Name, work.ID, logFilename)
|
|
|
|
wLogFile := bufio.NewWriter(logFile)
|
|
|
|
r, w := io.Pipe()
|
|
cmd.Stdout = w
|
|
cmd.Stderr = w
|
|
|
|
// Start the script...
|
|
err = cmd.Start()
|
|
if err != nil {
|
|
return logFilename, err
|
|
}
|
|
|
|
// Write script output to log file and the work message cahnnel.
|
|
go func(reader io.Reader) {
|
|
scanner := bufio.NewScanner(reader)
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
// writing to the work channel
|
|
if !work.IsTerminated() {
|
|
work.MessageChan <- []byte(line)
|
|
} else {
|
|
logger.Error.Printf("Work %s#%d is over. Unable to write more data into the channel: %s\n", work.Name, work.ID, line)
|
|
break
|
|
}
|
|
// writing to outfile
|
|
if _, err := wLogFile.WriteString(line + "\n"); err != nil {
|
|
logger.Error.Println("Error while writing into the log file:", logFilename, err)
|
|
break
|
|
}
|
|
}
|
|
if err := scanner.Err(); err != nil {
|
|
logger.Error.Printf("Work %s#%d unable to read script stdout: %v\n", work.Name, work.ID, err)
|
|
}
|
|
if err = wLogFile.Flush(); err != nil {
|
|
logger.Error.Println("Error while flushing the log file:", logFilename, err)
|
|
}
|
|
}(r)
|
|
|
|
timer := time.AfterFunc(time.Duration(work.Timeout)*time.Second, func() {
|
|
logger.Warning.Printf("Work %s#%d has timed out (%ds). Killing process #%d...\n", work.Name, work.ID, work.Timeout, cmd.Process.Pid)
|
|
syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL)
|
|
})
|
|
err = cmd.Wait()
|
|
timer.Stop()
|
|
work.Terminate()
|
|
if err != nil {
|
|
logger.Info.Printf("Work %s#%d done [ERROR]\n", work.Name, work.ID)
|
|
return logFilename, err
|
|
}
|
|
logger.Info.Printf("Work %s#%d done [SUCCESS]\n", work.Name, work.ID)
|
|
return logFilename, nil
|
|
}
|