controller startup/setup (#12)

This commit is contained in:
Michael Quigley 2022-08-15 14:25:50 -04:00
parent 1f36af7cf5
commit 6652163d75
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
2 changed files with 54 additions and 0 deletions

View File

@ -29,6 +29,10 @@ func Run(cfg *Config) error {
api.TunnelTunnelHandler = newTunnelHandler(cfg)
api.TunnelUntunnelHandler = newUntunnelHandler(cfg)
if err := controllerStartup(cfg); err != nil {
return err
}
if v, err := store.Open(cfg.Store); err == nil {
str = v
} else {

50
controller/startup.go Normal file
View File

@ -0,0 +1,50 @@
package controller
import (
"context"
"github.com/openziti/edge/rest_management_api_client"
"github.com/openziti/edge/rest_management_api_client/config"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"time"
)
func controllerStartup(cfg *Config) error {
if err := inspectZiti(cfg); err != nil {
return err
}
return nil
}
func inspectZiti(cfg *Config) error {
logrus.Infof("inspecting ziti controller configuration")
edge, err := edgeClient(cfg.Ziti)
if err != nil {
return errors.Wrap(err, "error getting ziti edge client")
}
if err := ensureZrokAuthConfigType(edge); err != nil {
return err
}
return nil
}
func ensureZrokAuthConfigType(edge *rest_management_api_client.ZitiEdgeManagement) error {
filter := "name=\"zrok.auth.v1\""
limit := int64(100)
offset := int64(0)
listReq := &config.ListConfigTypesParams{
Filter: &filter,
Limit: &limit,
Offset: &offset,
Context: context.Background(),
}
listReq.SetTimeout(30 * time.Second)
listResp, err := edge.Config.ListConfigTypes(listReq, nil)
if err != nil {
return err
}
logrus.Infof("found %d zrok.auth.v1 config types", len(listResp.Payload.Data))
return nil
}