Add admin interface

Author: murgatroid99

packages/grpc-js/src/admin.ts

@@ -0,0 +1,39 @@
+/*
+ * Copyright 2021 gRPC authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+import { ServiceDefinition } from "./make-client";
+import { Server, UntypedServiceImplementation } from "./server";
+
+interface GetServiceDefinition {
+  (): ServiceDefinition;
+}
+
+interface GetHandlers {
+  (): UntypedServiceImplementation;
+}
+
+const registeredAdminServices: {getServiceDefinition: GetServiceDefinition, getHandlers: GetHandlers}[] = [];
+
+export function registerAdminService(getServiceDefinition: GetServiceDefinition, getHandlers: GetHandlers) {
+  registeredAdminServices.push({getServiceDefinition, getHandlers});
+}
+
+export function addAdminServicesToServer(server: Server): void {
+  for (const {getServiceDefinition, getHandlers} of registeredAdminServices) {
+    server.addService(getServiceDefinition(), getHandlers());
+  }
+}

packages/grpc-js/src/channelz.ts

@@ -50,6 +50,7 @@ import { GetServerSocketsResponse } from "./generated/grpc/channelz/v1/GetServer
 import { ChannelzDefinition, ChannelzHandlers } from "./generated/grpc/channelz/v1/Channelz";
 import { ProtoGrpcType as ChannelzProtoGrpcType } from "./generated/channelz";
 import type { loadSync } from '@grpc/proto-loader';
+import { registerAdminService } from "./admin";
 
 export type TraceSeverity = 'CT_UNKNOWN' | 'CT_INFO' | 'CT_WARNING' | 'CT_ERROR';
 
@@ -713,4 +714,8 @@ export function getChannelzServiceDefinition(): ChannelzDefinition {
   }) as unknown as ChannelzProtoGrpcType;
   loadedChannelzDefinition = loadedProto.grpc.channelz.v1.Channelz.service;
   return loadedChannelzDefinition;
+}
+
+export function setup() {
+  registerAdminService(getChannelzServiceDefinition, getChannelzHandlers);
 }

packages/grpc-js/src/experimental.ts

@@ -32,3 +32,4 @@ export {
 export { Call as CallStream } from './call-stream';
 export { Filter, BaseFilter, FilterFactory } from './filter';
 export { FilterStackFactory } from './filter-stack';
+export { registerAdminService } from './admin';

packages/grpc-js/src/index.ts

@@ -249,6 +249,13 @@ export { GrpcObject } from './make-client';
 
 export { ChannelOptions } from './channel-options';
 
+export {
+  getChannelzServiceDefinition,
+  getChannelzHandlers
+} from './channelz';
+
+export { addAdminServicesToServer } from './admin';
+
 import * as experimental from './experimental';
 export { experimental };
 
@@ -257,11 +264,13 @@ import * as resolver_uds from './resolver-uds';
 import * as resolver_ip from './resolver-ip';
 import * as load_balancer_pick_first from './load-balancer-pick-first';
 import * as load_balancer_round_robin from './load-balancer-round-robin';
+import * as channelz from './channelz';
 
 (() => {
   resolver_dns.setup();
   resolver_uds.setup();
   resolver_ip.setup();
   load_balancer_pick_first.setup();
   load_balancer_round_robin.setup();
+  channelz.setup();
 })();