TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2024-12-04 23:45:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add grpc server and gateway

Browse Source
This commit is contained in:
Fabiennne 2020-03-24 11:20:45 +01:00
parent c20dd3cb3f
commit f2a9fa8972
6 changed files with 183 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
pkg/admin/api/grpc/gateway.go Normal file
View File

@ -0,0 +1,52 @@
package grpc
import (
"github.com/caos/utils/api/grpc/server"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"strings"
)
type GatewayConfig struct {
Port string
GRPCEndpoint string
CustomHeaders []string
}
type Gateway struct {
grpcEndpoint string
port string
cutomHeaders []string
}
func StartGateway(conf GatewayConfig) *Gateway {
return &Gateway{
grpcEndpoint: conf.GRPCEndpoint,
port: conf.Port,
cutomHeaders: conf.CustomHeaders,
}
}
func (gw *Gateway) Gateway() server.GatewayFunc {
return RegisterAdminServiceHandlerFromEndpoint
}
func (gw *Gateway) GRPCEndpoint() string {
return ":" + gw.grpcEndpoint
}
func (gw *Gateway) GatewayPort() string {
return gw.port
}
func (gw *Gateway) GatewayServeMuxOptions() []runtime.ServeMuxOption {
return []runtime.ServeMuxOption{
runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
for _, customHeader := range gw.cutomHeaders {
if strings.HasPrefix(strings.ToLower(header), customHeader) {
return header, true
}
}
return header, false
}),
}
}

10
pkg/admin/api/grpc/server.go
View File

@ -1,6 +1,8 @@
package grpc
import (
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
"google.golang.org/grpc"
)
@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
}
func (s *Server) GRPCServer() (*grpc.Server, error) {
gs := grpc.NewServer()
gs := grpc.NewServer(
grpc.UnaryInterceptor(
grpc_middleware.ChainUnaryServer(
grpc_utils.ErrorHandler(),
),
),
)
RegisterAdminServiceServer(gs, s)
return gs, nil
}

52
pkg/auth/api/grpc/gateway.go Normal file
View File

@ -0,0 +1,52 @@
package grpc
import (
"github.com/caos/utils/api/grpc/server"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"strings"
)
type GatewayConfig struct {
Port string
GRPCEndpoint string
CustomHeaders []string
}
type Gateway struct {
grpcEndpoint string
port string
cutomHeaders []string
}
func StartGateway(conf GatewayConfig) *Gateway {
return &Gateway{
grpcEndpoint: conf.GRPCEndpoint,
port: conf.Port,
cutomHeaders: conf.CustomHeaders,
}
}
func (gw *Gateway) Gateway() server.GatewayFunc {
return RegisterAuthServiceHandlerFromEndpoint
}
func (gw *Gateway) GRPCEndpoint() string {
return ":" + gw.grpcEndpoint
}
func (gw *Gateway) GatewayPort() string {
return gw.port
}
func (gw *Gateway) GatewayServeMuxOptions() []runtime.ServeMuxOption {
return []runtime.ServeMuxOption{
runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
for _, customHeader := range gw.cutomHeaders {
if strings.HasPrefix(strings.ToLower(header), customHeader) {
return header, true
}
}
return header, false
}),
}
}

10
pkg/auth/api/grpc/server.go
View File

@ -1,6 +1,8 @@
package grpc
import (
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
"google.golang.org/grpc"
)
@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
}
func (s *Server) GRPCServer() (*grpc.Server, error) {
gs := grpc.NewServer()
gs := grpc.NewServer(
grpc.UnaryInterceptor(
grpc_middleware.ChainUnaryServer(
grpc_utils.ErrorHandler(),
),
),
)
RegisterAuthServiceServer(gs, s)
return gs, nil
}

52
pkg/management/api/grpc/gateway.go Normal file
View File

@ -0,0 +1,52 @@
package grpc
import (
"github.com/caos/utils/api/grpc/server"
"github.com/grpc-ecosystem/grpc-gateway/runtime"
"strings"
)
type GatewayConfig struct {
Port string
GRPCEndpoint string
CustomHeaders []string
}
type Gateway struct {
grpcEndpoint string
port string
cutomHeaders []string
}
func StartGateway(conf GatewayConfig) *Gateway {
return &Gateway{
grpcEndpoint: conf.GRPCEndpoint,
port: conf.Port,
cutomHeaders: conf.CustomHeaders,
}
}
func (gw *Gateway) Gateway() server.GatewayFunc {
return RegisterManagementServiceHandlerFromEndpoint
}
func (gw *Gateway) GRPCEndpoint() string {
return ":" + gw.grpcEndpoint
}
func (gw *Gateway) GatewayPort() string {
return gw.port
}
func (gw *Gateway) GatewayServeMuxOptions() []runtime.ServeMuxOption {
return []runtime.ServeMuxOption{
runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
for _, customHeader := range gw.cutomHeaders {
if strings.HasPrefix(strings.ToLower(header), customHeader) {
return header, true
}
}
return header, false
}),
}
}

10
pkg/management/api/grpc/server.go
View File

@ -1,6 +1,8 @@
package grpc
import (
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
"google.golang.org/grpc"
)
@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
}
func (s *Server) GRPCServer() (*grpc.Server, error) {
gs := grpc.NewServer()
gs := grpc.NewServer(
grpc.UnaryInterceptor(
grpc_middleware.ChainUnaryServer(
grpc_utils.ErrorHandler(),
),
),
)
RegisterManagementServiceServer(gs, s)
return gs, nil
}