feat: register lease/cluster/maintanance grpc gateway handlers (#31)

api7 · Feb 1, 2024 · 0343aa7 · 0343aa7
1 parent 155ec8b
commit 0343aa7
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/pkg/adapter/server.go b/pkg/adapter/server.go
@@ -128,12 +128,21 @@ func (a *adapter) registerGateway(addr string) (*gatewayruntime.ServeMux, error)
 		return nil, err
 	}
 	gwmux := gatewayruntime.NewServeMux()
-	if err := etcdservergw.RegisterKVHandler(a.ctx, gwmux, grpcConn); err != nil {
-		return nil, err
+
+	handlers := []func(context.Context, *gatewayruntime.ServeMux, *grpc.ClientConn) error{
+		etcdservergw.RegisterWatchHandler,
+		etcdservergw.RegisterKVHandler,
+		etcdservergw.RegisterLeaseHandler,
+		etcdservergw.RegisterClusterHandler,
+		etcdservergw.RegisterMaintenanceHandler,
 	}
-	if err := etcdservergw.RegisterWatchHandler(a.ctx, gwmux, grpcConn); err != nil {
-		return nil, err
+
+	for i := range handlers {
+		if err := handlers[i](a.ctx, gwmux, grpcConn); err != nil {
+			return nil, err
+		}
 	}
+
 	go func() {
 		<-a.ctx.Done()
 		if err := grpcConn.Close(); err != nil {