mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-11 16:47:32 +00:00
fix: add grpc server and gateway
This commit is contained in:
52
pkg/admin/api/grpc/gateway.go
Normal file
52
pkg/admin/api/grpc/gateway.go
Normal 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
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
}
|
@@ -1,6 +1,8 @@
|
|||||||
package grpc
|
package grpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
|
||||||
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
||||||
gs := grpc.NewServer()
|
gs := grpc.NewServer(
|
||||||
|
grpc.UnaryInterceptor(
|
||||||
|
grpc_middleware.ChainUnaryServer(
|
||||||
|
grpc_utils.ErrorHandler(),
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
RegisterAdminServiceServer(gs, s)
|
RegisterAdminServiceServer(gs, s)
|
||||||
return gs, nil
|
return gs, nil
|
||||||
}
|
}
|
||||||
|
52
pkg/auth/api/grpc/gateway.go
Normal file
52
pkg/auth/api/grpc/gateway.go
Normal 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
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
}
|
@@ -1,6 +1,8 @@
|
|||||||
package grpc
|
package grpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
|
||||||
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
||||||
gs := grpc.NewServer()
|
gs := grpc.NewServer(
|
||||||
|
grpc.UnaryInterceptor(
|
||||||
|
grpc_middleware.ChainUnaryServer(
|
||||||
|
grpc_utils.ErrorHandler(),
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
RegisterAuthServiceServer(gs, s)
|
RegisterAuthServiceServer(gs, s)
|
||||||
return gs, nil
|
return gs, nil
|
||||||
}
|
}
|
||||||
|
52
pkg/management/api/grpc/gateway.go
Normal file
52
pkg/management/api/grpc/gateway.go
Normal 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
|
||||||
|
}),
|
||||||
|
}
|
||||||
|
}
|
@@ -1,6 +1,8 @@
|
|||||||
package grpc
|
package grpc
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
grpc_utils "github.com/caos/zitadel/internal/api/grpc"
|
||||||
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
||||||
"google.golang.org/grpc"
|
"google.golang.org/grpc"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -28,7 +30,13 @@ func (s *Server) GRPCPort() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
func (s *Server) GRPCServer() (*grpc.Server, error) {
|
||||||
gs := grpc.NewServer()
|
gs := grpc.NewServer(
|
||||||
|
grpc.UnaryInterceptor(
|
||||||
|
grpc_middleware.ChainUnaryServer(
|
||||||
|
grpc_utils.ErrorHandler(),
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
RegisterManagementServiceServer(gs, s)
|
RegisterManagementServiceServer(gs, s)
|
||||||
return gs, nil
|
return gs, nil
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user