From 33a4c85f89591c36cb7001d604ce1f62b65e2db6 Mon Sep 17 00:00:00 2001
From: Michael Lumish <mlumish@google.com>
Date: Tue, 4 Aug 2020 13:03:08 -0700
Subject: [PATCH] grpc-js: Implement getPeer on the client and server

---
 packages/grpc-js/src/call-stream.ts |  2 +-
 packages/grpc-js/src/call.ts        |  8 ++++----
 packages/grpc-js/src/server-call.ts | 21 +++++++++++++++++----
 3 files changed, 22 insertions(+), 9 deletions(-)

diff --git a/packages/grpc-js/src/call-stream.ts b/packages/grpc-js/src/call-stream.ts
index 825342a5..2c9f01cb 100644
--- a/packages/grpc-js/src/call-stream.ts
+++ b/packages/grpc-js/src/call-stream.ts
@@ -630,7 +630,7 @@ export class Http2CallStream implements Call {
   }
 
   getPeer(): string {
-    throw new Error('Not yet implemented');
+    return this.subchannel?.getAddress() ?? this.channel.getTarget();
   }
 
   getMethod(): string {
diff --git a/packages/grpc-js/src/call.ts b/packages/grpc-js/src/call.ts
index 0d88ef15..cfe37ecf 100644
--- a/packages/grpc-js/src/call.ts
+++ b/packages/grpc-js/src/call.ts
@@ -93,7 +93,7 @@ export class ClientUnaryCallImpl extends EventEmitter
   }
 
   getPeer(): string {
-    return this.call?.getPeer() ?? '';
+    return this.call?.getPeer() ?? 'unknown';
   }
 }
 
@@ -109,7 +109,7 @@ export class ClientReadableStreamImpl<ResponseType> extends Readable
   }
 
   getPeer(): string {
-    return this.call?.getPeer() ?? '';
+    return this.call?.getPeer() ?? 'unknown';
   }
 
   _read(_size: number): void {
@@ -129,7 +129,7 @@ export class ClientWritableStreamImpl<RequestType> extends Writable
   }
 
   getPeer(): string {
-    return this.call?.getPeer() ?? '';
+    return this.call?.getPeer() ?? 'unknown';
   }
 
   _write(chunk: RequestType, encoding: string, cb: WriteCallback) {
@@ -164,7 +164,7 @@ export class ClientDuplexStreamImpl<RequestType, ResponseType> extends Duplex
   }
 
   getPeer(): string {
-    return this.call?.getPeer() ?? '';
+    return this.call?.getPeer() ?? 'unknown';
   }
 
   _read(_size: number): void {
diff --git a/packages/grpc-js/src/server-call.ts b/packages/grpc-js/src/server-call.ts
index 82d88517..bd992773 100644
--- a/packages/grpc-js/src/server-call.ts
+++ b/packages/grpc-js/src/server-call.ts
@@ -112,7 +112,7 @@ export class ServerUnaryCallImpl<RequestType, ResponseType> extends EventEmitter
   }
 
   getPeer(): string {
-    throw new Error('not implemented yet');
+    return this.call.getPeer();
   }
 
   sendMetadata(responseMetadata: Metadata): void {
@@ -145,7 +145,7 @@ export class ServerReadableStreamImpl<RequestType, ResponseType>
   }
 
   getPeer(): string {
-    throw new Error('not implemented yet');
+    return this.call.getPeer();
   }
 
   sendMetadata(responseMetadata: Metadata): void {
@@ -178,7 +178,7 @@ export class ServerWritableStreamImpl<RequestType, ResponseType>
   }
 
   getPeer(): string {
-    throw new Error('not implemented yet');
+    return this.call.getPeer();
   }
 
   sendMetadata(responseMetadata: Metadata): void {
@@ -249,7 +249,7 @@ export class ServerDuplexStreamImpl<RequestType, ResponseType> extends Duplex
   }
 
   getPeer(): string {
-    throw new Error('not implemented yet');
+    return this.call.getPeer();
   }
 
   sendMetadata(responseMetadata: Metadata): void {
@@ -738,6 +738,19 @@ export class Http2ServerCallStream<
       );
     }
   }
+
+  getPeer(): string {
+    const socket = this.stream.session.socket;
+    if (socket.remoteAddress) {
+      if (socket.remotePort) {
+        return `${socket.remoteAddress}:${socket.remotePort}`;
+      } else {
+        return socket.remoteAddress;
+      }
+    } else {
+      return 'unknown';
+    }
+  }
 }
 
 /* eslint-disable @typescript-eslint/no-explicit-any */