extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-02-22 05:01:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

adjust proxy to retrieve configs (#12)

Browse Source
This commit is contained in:
Michael Quigley 2022-08-15 15:18:40 -04:00
parent 01ffe50889
commit 83ffb025e5
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
proxy/proxy.go
View File

@ -25,6 +25,7 @@ func Run(cfg *Config) error {
if err != nil {
return errors.Wrap(err, "error loading config")
}
zCfg.ConfigTypes = []string{"zrok.auth.v1"}
zCtx := ziti.NewContextWithConfig(zCfg)
zDialCtx := ZitiDialContext{Context: zCtx}
zTransport := http.DefaultTransport.(*http.Transport).Clone()
@ -54,15 +55,20 @@ type ZitiDialContext struct {
}
func (self *ZitiDialContext) Dial(_ context.Context, _ string, addr string) (net.Conn, error) {
service := strings.Split(addr, ":")[0] // ignore :port (we get passed 'host:port')
_, found := self.Context.GetService(service)
svcName := strings.Split(addr, ":")[0] // ignore :port (we get passed 'host:port')
svc, found := self.Context.GetService(svcName)
if !found {
logrus.Infof("service '%v' not cached; refreshing", service)
logrus.Infof("service '%v' not cached; refreshing", svcName)
if err := self.Context.RefreshServices(); err != nil {
return nil, errors.Wrap(err, "error refreshing services")
}
svc, found = self.Context.GetService(svcName)
if !found {
return nil, errors.Errorf("no such service '%v'", svcName)
}
}
return self.Context.Dial(service)
logrus.Info(svc.Configs)
return self.Context.Dial(svcName)
}
type ProxyServiceResolver interface {