extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-06-26 04:31:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

rough wiring for the grpc gateway (#221)

Browse Source
This commit is contained in:
Michael Quigley 2024-10-01 13:27:26 -04:00
parent c3fafec2d9
commit b787c6c07f
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
agent/agent.go
View File

@ -1,6 +1,8 @@
package agent package agent
import ( import (
"context"
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
"github.com/openziti/zrok/agent/agentGrpc" "github.com/openziti/zrok/agent/agentGrpc"
"github.com/openziti/zrok/agent/proctree" "github.com/openziti/zrok/agent/proctree"
"github.com/openziti/zrok/environment/env_core" "github.com/openziti/zrok/environment/env_core"
@ -8,7 +10,9 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/sirupsen/logrus" "github.com/sirupsen/logrus"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"net" "net"
"net/http"
"os" "os"
) )
@ -44,7 +48,6 @@ func (a *Agent) Run() error {
if err := proctree.Init("zrok Agent"); err != nil { if err := proctree.Init("zrok Agent"); err != nil {
return err return err
} }
go a.manager()
agentSocket, err := a.root.AgentSocket() agentSocket, err := a.root.AgentSocket()
if err != nil { if err != nil {
@ -56,6 +59,9 @@ func (a *Agent) Run() error {
} }
a.agentSocket = agentSocket a.agentSocket = agentSocket
go a.manager()
go a.gateway()
srv := grpc.NewServer() srv := grpc.NewServer()
agentGrpc.RegisterAgentServer(srv, &agentGrpcImpl{agent: a}) agentGrpc.RegisterAgentServer(srv, &agentGrpcImpl{agent: a})
if err := srv.Serve(l); err != nil { if err := srv.Serve(l); err != nil {
@ -81,6 +87,23 @@ func (a *Agent) Shutdown() {
} }
} }
func (a *Agent) gateway() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
mux := runtime.NewServeMux()
opts := []grpc.DialOption{grpc.WithTransportCredentials(insecure.NewCredentials())}
endpoint := "unix:" + a.agentSocket
logrus.Infof("endpoint: %v", endpoint)
if err := agentGrpc.RegisterAgentHandlerFromEndpoint(ctx, mux, "unix:"+a.agentSocket, opts); err != nil {
logrus.Fatalf("unable to register gateway: %v", err)
}
if err := http.ListenAndServe(":8888", mux); err != nil {
logrus.Error(err)
}
}
func (a *Agent) manager() { func (a *Agent) manager() {
logrus.Info("started") logrus.Info("started")
defer logrus.Warn("exited") defer logrus.Warn("exited")