pmux/main.go

48 lines
909 B
Go
Raw Normal View History

2021-09-21 22:36:50 +00:00
package main
import (
"context"
"flag"
"fmt"
"io/ioutil"
"os"
"os/signal"
"syscall"
"github.com/cryptic-io/pmux/pmuxlib"
2021-09-21 22:36:50 +00:00
"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
2021-09-21 22:36:50 +00:00
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)
2021-09-21 22:36:50 +00:00
signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)
<-sigCh
2021-09-21 22:36:50 +00:00
cancel()
<-sigCh
fmt.Fprintln(os.Stderr, "forcefully exiting pmux process, there may be zombie child processes being left behind, good luck!")
os.Stderr.Sync()
2021-09-21 22:36:50 +00:00
os.Exit(1)
}()
pmuxlib.Run(ctx, cfg)
2021-09-21 22:36:50 +00:00
}