pmux/main.go
2022-06-18 19:52:04 -06:00

48 lines
909 B
Go

package main
import (
"context"
"flag"
"fmt"
"io/ioutil"
"os"
"os/signal"
"syscall"
"github.com/cryptic-io/pmux/pmuxlib"
"gopkg.in/yaml.v2"
)
func main() {
cfgPath := flag.String("c", "./pmux.yml", "Path to config yaml file")
flag.Parse()
cfgB, err := ioutil.ReadFile(*cfgPath)
if err != nil {
panic(fmt.Sprintf("couldn't read cfg file at %q: %v", *cfgPath, err))
}
var cfg pmuxlib.Config
if err := yaml.Unmarshal(cfgB, &cfg); err != nil {
panic(fmt.Sprintf("couldn't parse cfg file: %v", err))
}
ctx, cancel := context.WithCancel(context.Background())
go func() {
sigCh := make(chan os.Signal, 2)
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
<-sigCh
cancel()
<-sigCh
fmt.Fprintln(os.Stderr, "forcefully exiting pmux process, there may be zombie child processes being left behind, good luck!")
os.Stderr.Sync()
os.Exit(1)
}()
pmuxlib.Run(ctx, cfg)
}