Permalink
Branch: master
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
77 lines (63 sloc) 1.4 KB
package main
import (
"context"
"errors"
"log"
"os"
"os/signal"
"strconv"
"syscall"
"github.com/urcomputeringpal/kubevalidator/validator"
)
func runWithContext(ctx context.Context) error {
port, ok := os.LookupEnv("PORT")
if !ok {
port = "8080"
}
portInt, _ := strconv.Atoi(port)
webhookSecret, ok := os.LookupEnv("WEBHOOK_SECRET")
if !ok {
return errors.New("WEBHOOK_SECRET required")
}
appID, ok := os.LookupEnv("APP_ID")
if !ok {
return errors.New("APP_ID required")
}
appIDInt, _ := strconv.Atoi(appID)
privateKeyFile, ok := os.LookupEnv("PRIVATE_KEY_FILE")
if !ok {
return errors.New("PRIVATE_KEY_FILE required")
}
v := &validator.Server{
Port: portInt,
WebhookSecret: webhookSecret,
AppID: appIDInt,
PrivateKeyFile: privateKeyFile,
}
return v.Run(ctx)
}
func cancelOnInterrupt(ctx context.Context, f context.CancelFunc) {
term := make(chan os.Signal)
signal.Notify(term, os.Interrupt, syscall.SIGTERM)
for {
select {
case <-term:
log.Println("Received SIGTERM, exiting gracefully...")
f()
os.Exit(0)
case <-ctx.Done():
os.Exit(0)
}
}
}
func run() error {
ctx, cancelFunc := context.WithCancel(context.Background())
defer cancelFunc()
go cancelOnInterrupt(ctx, cancelFunc)
return runWithContext(ctx)
}
func main() {
if err := run(); err != nil && err != context.Canceled && err != context.DeadlineExceeded {
panic(err)
}
}