extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2024-11-25 09:33:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

shutdown hook

Browse Source
This commit is contained in:
Michael Quigley 2022-07-27 11:31:01 -04:00
parent cfa603ea0c
commit ad93c613b5
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 49 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
cmd/zrok/http.go
View File

@ -2,44 +2,63 @@ package main
import (
"github.com/openziti-test-kitchen/zrok/http"
"github.com/openziti-test-kitchen/zrok/rest_client_zrok"
"github.com/openziti-test-kitchen/zrok/rest_client_zrok/tunnel"
"github.com/openziti-test-kitchen/zrok/rest_model_zrok"
"github.com/openziti-test-kitchen/zrok/zrokdir"
"github.com/sirupsen/logrus"
"github.com/spf13/cobra"
"os"
"os/signal"
"syscall"
)
var httpCmd = &cobra.Command{
Use: "http <endpoint>",
Short: "Start an http terminator",
Args: cobra.ExactArgs(1),
Run: func(_ *cobra.Command, args []string) {
idCfg, err := zrokdir.IdentityConfigFile()
if err != nil {
panic(err)
}
cfg := &http.Config{
IdentityPath: idCfg,
EndpointAddress: args[0],
}
id, err := zrokdir.ReadIdentityId()
if err != nil {
panic(err)
}
zrok := newZrokClient()
req := tunnel.NewTunnelParams()
req.Body = &rest_model_zrok.TunnelRequest{
Endpoint: cfg.EndpointAddress,
Identity: id,
}
resp, err := zrok.Tunnel.Tunnel(req)
if err != nil {
panic(err)
}
cfg.Service = resp.Payload.Service
if err := http.Run(cfg); err != nil {
panic(err)
}
},
Run: handleHttp,
}
func handleHttp(_ *cobra.Command, args []string) {
idCfg, err := zrokdir.IdentityConfigFile()
if err != nil {
panic(err)
}
cfg := &http.Config{
IdentityPath: idCfg,
EndpointAddress: args[0],
}
id, err := zrokdir.ReadIdentityId()
if err != nil {
panic(err)
}
zrok := newZrokClient()
req := tunnel.NewTunnelParams()
req.Body = &rest_model_zrok.TunnelRequest{
Endpoint: cfg.EndpointAddress,
Identity: id,
}
resp, err := zrok.Tunnel.Tunnel(req)
if err != nil {
panic(err)
}
cfg.Service = resp.Payload.Service
c := make(chan os.Signal)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
<-c
cleanupHttp(cfg, zrok)
os.Exit(1)
}()
if err := http.Run(cfg); err != nil {
panic(err)
}
}
func cleanupHttp(cfg *http.Config, zrok *rest_client_zrok.Zrok) {
logrus.Infof("shutting down '%v'", cfg.Service)
}