summaryrefslogtreecommitdiff
path: root/frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
diff options
context:
space:
mode:
authoraltaf-creator <dev@altafcreator.com>2025-11-09 11:15:19 +0800
committeraltaf-creator <dev@altafcreator.com>2025-11-09 11:15:19 +0800
commit8eff962cab608341a6f2fedc640a0e32d96f26e2 (patch)
tree05534d1a720ddc3691d346c69b4972555820a061 /frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
pain
Diffstat (limited to 'frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts')
-rw-r--r--frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts1022
1 files changed, 1022 insertions, 0 deletions
diff --git a/frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts b/frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
new file mode 100644
index 0000000..3032c6e
--- /dev/null
+++ b/frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
@@ -0,0 +1,1022 @@
+/*
+ * Copyright 2019 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 { EventEmitter } from 'events';
+import * as http2 from 'http2';
+import { Duplex, Readable, Writable } from 'stream';
+import * as zlib from 'zlib';
+
+import {
+ Status,
+ DEFAULT_MAX_SEND_MESSAGE_LENGTH,
+ DEFAULT_MAX_RECEIVE_MESSAGE_LENGTH,
+ LogVerbosity,
+} from './constants';
+import { Deserialize, Serialize } from './make-client';
+import { Metadata } from './metadata';
+import { StreamDecoder } from './stream-decoder';
+import { ObjectReadable, ObjectWritable } from './object-stream';
+import { ChannelOptions } from './channel-options';
+import * as logging from './logging';
+import { StatusObject, PartialStatusObject } from './call-interface';
+import { Deadline } from './deadline';
+import { getErrorCode, getErrorMessage } from './error';
+
+const TRACER_NAME = 'server_call';
+
+function trace(text: string): void {
+ logging.trace(LogVerbosity.DEBUG, TRACER_NAME, text);
+}
+
+interface DeadlineUnitIndexSignature {
+ [name: string]: number;
+}
+
+const GRPC_ACCEPT_ENCODING_HEADER = 'grpc-accept-encoding';
+const GRPC_ENCODING_HEADER = 'grpc-encoding';
+const GRPC_MESSAGE_HEADER = 'grpc-message';
+const GRPC_STATUS_HEADER = 'grpc-status';
+const GRPC_TIMEOUT_HEADER = 'grpc-timeout';
+const DEADLINE_REGEX = /(\d{1,8})\s*([HMSmun])/;
+const deadlineUnitsToMs: DeadlineUnitIndexSignature = {
+ H: 3600000,
+ M: 60000,
+ S: 1000,
+ m: 1,
+ u: 0.001,
+ n: 0.000001,
+};
+const defaultCompressionHeaders = {
+ // TODO(cjihrig): Remove these encoding headers from the default response
+ // once compression is integrated.
+ [GRPC_ACCEPT_ENCODING_HEADER]: 'identity,deflate,gzip',
+ [GRPC_ENCODING_HEADER]: 'identity',
+};
+const defaultResponseHeaders = {
+ [http2.constants.HTTP2_HEADER_STATUS]: http2.constants.HTTP_STATUS_OK,
+ [http2.constants.HTTP2_HEADER_CONTENT_TYPE]: 'application/grpc+proto',
+};
+const defaultResponseOptions = {
+ waitForTrailers: true,
+} as http2.ServerStreamResponseOptions;
+
+export type ServerStatusResponse = Partial<StatusObject>;
+
+export type ServerErrorResponse = ServerStatusResponse & Error;
+
+export type ServerSurfaceCall = {
+ cancelled: boolean;
+ readonly metadata: Metadata;
+ getPeer(): string;
+ sendMetadata(responseMetadata: Metadata): void;
+ getDeadline(): Deadline;
+ getPath(): string;
+} & EventEmitter;
+
+export type ServerUnaryCall<RequestType, ResponseType> = ServerSurfaceCall & {
+ request: RequestType;
+};
+export type ServerReadableStream<RequestType, ResponseType> =
+ ServerSurfaceCall & ObjectReadable<RequestType>;
+export type ServerWritableStream<RequestType, ResponseType> =
+ ServerSurfaceCall &
+ ObjectWritable<ResponseType> & {
+ request: RequestType;
+ end: (metadata?: Metadata) => void;
+ };
+export type ServerDuplexStream<RequestType, ResponseType> = ServerSurfaceCall &
+ ObjectReadable<RequestType> &
+ ObjectWritable<ResponseType> & { end: (metadata?: Metadata) => void };
+
+export class ServerUnaryCallImpl<RequestType, ResponseType>
+ extends EventEmitter
+ implements ServerUnaryCall<RequestType, ResponseType>
+{
+ cancelled: boolean;
+
+ constructor(
+ private call: Http2ServerCallStream<RequestType, ResponseType>,
+ public metadata: Metadata,
+ public request: RequestType
+ ) {
+ super();
+ this.cancelled = false;
+ this.call.setupSurfaceCall(this);
+ }
+
+ getPeer(): string {
+ return this.call.getPeer();
+ }
+
+ sendMetadata(responseMetadata: Metadata): void {
+ this.call.sendMetadata(responseMetadata);
+ }
+
+ getDeadline(): Deadline {
+ return this.call.getDeadline();
+ }
+
+ getPath(): string {
+ return this.call.getPath();
+ }
+}
+
+export class ServerReadableStreamImpl<RequestType, ResponseType>
+ extends Readable
+ implements ServerReadableStream<RequestType, ResponseType>
+{
+ cancelled: boolean;
+
+ constructor(
+ private call: Http2ServerCallStream<RequestType, ResponseType>,
+ public metadata: Metadata,
+ public deserialize: Deserialize<RequestType>,
+ encoding: string
+ ) {
+ super({ objectMode: true });
+ this.cancelled = false;
+ this.call.setupSurfaceCall(this);
+ this.call.setupReadable(this, encoding);
+ }
+
+ _read(size: number) {
+ if (!this.call.consumeUnpushedMessages(this)) {
+ return;
+ }
+
+ this.call.resume();
+ }
+
+ getPeer(): string {
+ return this.call.getPeer();
+ }
+
+ sendMetadata(responseMetadata: Metadata): void {
+ this.call.sendMetadata(responseMetadata);
+ }
+
+ getDeadline(): Deadline {
+ return this.call.getDeadline();
+ }
+
+ getPath(): string {
+ return this.call.getPath();
+ }
+}
+
+export class ServerWritableStreamImpl<RequestType, ResponseType>
+ extends Writable
+ implements ServerWritableStream<RequestType, ResponseType>
+{
+ cancelled: boolean;
+ private trailingMetadata: Metadata;
+
+ constructor(
+ private call: Http2ServerCallStream<RequestType, ResponseType>,
+ public metadata: Metadata,
+ public serialize: Serialize<ResponseType>,
+ public request: RequestType
+ ) {
+ super({ objectMode: true });
+ this.cancelled = false;
+ this.trailingMetadata = new Metadata();
+ this.call.setupSurfaceCall(this);
+
+ this.on('error', err => {
+ this.call.sendError(err);
+ this.end();
+ });
+ }
+
+ getPeer(): string {
+ return this.call.getPeer();
+ }
+
+ sendMetadata(responseMetadata: Metadata): void {
+ this.call.sendMetadata(responseMetadata);
+ }
+
+ getDeadline(): Deadline {
+ return this.call.getDeadline();
+ }
+
+ getPath(): string {
+ return this.call.getPath();
+ }
+
+ _write(
+ chunk: ResponseType,
+ encoding: string,
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
+ callback: (...args: any[]) => void
+ ) {
+ try {
+ const response = this.call.serializeMessage(chunk);
+
+ if (!this.call.write(response)) {
+ this.call.once('drain', callback);
+ return;
+ }
+ } catch (err) {
+ this.emit('error', {
+ details: getErrorMessage(err),
+ code: Status.INTERNAL,
+ });
+ }
+
+ callback();
+ }
+
+ _final(callback: Function): void {
+ this.call.sendStatus({
+ code: Status.OK,
+ details: 'OK',
+ metadata: this.trailingMetadata,
+ });
+ callback(null);
+ }
+
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
+ end(metadata?: any) {
+ if (metadata) {
+ this.trailingMetadata = metadata;
+ }
+
+ return super.end();
+ }
+}
+
+export class ServerDuplexStreamImpl<RequestType, ResponseType>
+ extends Duplex
+ implements ServerDuplexStream<RequestType, ResponseType>
+{
+ cancelled: boolean;
+ /* This field appears to be unsued, but it is actually used in _final, which is assiged from
+ * ServerWritableStreamImpl.prototype._final below. */
+ // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+ // @ts-ignore noUnusedLocals
+ private trailingMetadata: Metadata;
+
+ constructor(
+ private call: Http2ServerCallStream<RequestType, ResponseType>,
+ public metadata: Metadata,
+ public serialize: Serialize<ResponseType>,
+ public deserialize: Deserialize<RequestType>,
+ encoding: string
+ ) {
+ super({ objectMode: true });
+ this.cancelled = false;
+ this.trailingMetadata = new Metadata();
+ this.call.setupSurfaceCall(this);
+ this.call.setupReadable(this, encoding);
+
+ this.on('error', err => {
+ this.call.sendError(err);
+ this.end();
+ });
+ }
+
+ getPeer(): string {
+ return this.call.getPeer();
+ }
+
+ sendMetadata(responseMetadata: Metadata): void {
+ this.call.sendMetadata(responseMetadata);
+ }
+
+ getDeadline(): Deadline {
+ return this.call.getDeadline();
+ }
+
+ getPath(): string {
+ return this.call.getPath();
+ }
+
+ // eslint-disable-next-line @typescript-eslint/no-explicit-any
+ end(metadata?: any) {
+ if (metadata) {
+ this.trailingMetadata = metadata;
+ }
+
+ return super.end();
+ }
+}
+
+ServerDuplexStreamImpl.prototype._read =
+ ServerReadableStreamImpl.prototype._read;
+ServerDuplexStreamImpl.prototype._write =
+ ServerWritableStreamImpl.prototype._write;
+ServerDuplexStreamImpl.prototype._final =
+ ServerWritableStreamImpl.prototype._final;
+
+// Unary response callback signature.
+export type sendUnaryData<ResponseType> = (
+ error: ServerErrorResponse | ServerStatusResponse | null,
+ value?: ResponseType | null,
+ trailer?: Metadata,
+ flags?: number
+) => void;
+
+// User provided handler for unary calls.
+export type handleUnaryCall<RequestType, ResponseType> = (
+ call: ServerUnaryCall<RequestType, ResponseType>,
+ callback: sendUnaryData<ResponseType>
+) => void;
+
+// User provided handler for client streaming calls.
+export type handleClientStreamingCall<RequestType, ResponseType> = (
+ call: ServerReadableStream<RequestType, ResponseType>,
+ callback: sendUnaryData<ResponseType>
+) => void;
+
+// User provided handler for server streaming calls.
+export type handleServerStreamingCall<RequestType, ResponseType> = (
+ call: ServerWritableStream<RequestType, ResponseType>
+) => void;
+
+// User provided handler for bidirectional streaming calls.
+export type handleBidiStreamingCall<RequestType, ResponseType> = (
+ call: ServerDuplexStream<RequestType, ResponseType>
+) => void;
+
+export type HandleCall<RequestType, ResponseType> =
+ | handleUnaryCall<RequestType, ResponseType>
+ | handleClientStreamingCall<RequestType, ResponseType>
+ | handleServerStreamingCall<RequestType, ResponseType>
+ | handleBidiStreamingCall<RequestType, ResponseType>;
+
+export interface UnaryHandler<RequestType, ResponseType> {
+ func: handleUnaryCall<RequestType, ResponseType>;
+ serialize: Serialize<ResponseType>;
+ deserialize: Deserialize<RequestType>;
+ type: HandlerType;
+ path: string;
+}
+
+export interface ClientStreamingHandler<RequestType, ResponseType> {
+ func: handleClientStreamingCall<RequestType, ResponseType>;
+ serialize: Serialize<ResponseType>;
+ deserialize: Deserialize<RequestType>;
+ type: HandlerType;
+ path: string;
+}
+
+export interface ServerStreamingHandler<RequestType, ResponseType> {
+ func: handleServerStreamingCall<RequestType, ResponseType>;
+ serialize: Serialize<ResponseType>;
+ deserialize: Deserialize<RequestType>;
+ type: HandlerType;
+ path: string;
+}
+
+export interface BidiStreamingHandler<RequestType, ResponseType> {
+ func: handleBidiStreamingCall<RequestType, ResponseType>;
+ serialize: Serialize<ResponseType>;
+ deserialize: Deserialize<RequestType>;
+ type: HandlerType;
+ path: string;
+}
+
+export type Handler<RequestType, ResponseType> =
+ | UnaryHandler<RequestType, ResponseType>
+ | ClientStreamingHandler<RequestType, ResponseType>
+ | ServerStreamingHandler<RequestType, ResponseType>
+ | BidiStreamingHandler<RequestType, ResponseType>;
+
+export type HandlerType = 'bidi' | 'clientStream' | 'serverStream' | 'unary';
+
+// Internal class that wraps the HTTP2 request.
+export class Http2ServerCallStream<
+ RequestType,
+ ResponseType
+> extends EventEmitter {
+ cancelled = false;
+ deadlineTimer: NodeJS.Timeout | null = null;
+ private statusSent = false;
+ private deadline: Deadline = Infinity;
+ private wantTrailers = false;
+ private metadataSent = false;
+ private canPush = false;
+ private isPushPending = false;
+ private bufferedMessages: Array<Buffer | null> = [];
+ private messagesToPush: Array<RequestType | null> = [];
+ private maxSendMessageSize: number = DEFAULT_MAX_SEND_MESSAGE_LENGTH;
+ private maxReceiveMessageSize: number = DEFAULT_MAX_RECEIVE_MESSAGE_LENGTH;
+
+ constructor(
+ private stream: http2.ServerHttp2Stream,
+ private handler: Handler<RequestType, ResponseType>,
+ options: ChannelOptions
+ ) {
+ super();
+
+ this.stream.once('error', (err: ServerErrorResponse) => {
+ /* We need an error handler to avoid uncaught error event exceptions, but
+ * there is nothing we can reasonably do here. Any error event should
+ * have a corresponding close event, which handles emitting the cancelled
+ * event. And the stream is now in a bad state, so we can't reasonably
+ * expect to be able to send an error over it. */
+ });
+
+ this.stream.once('close', () => {
+ trace(
+ 'Request to method ' +
+ this.handler?.path +
+ ' stream closed with rstCode ' +
+ this.stream.rstCode
+ );
+
+ if (!this.statusSent) {
+ this.cancelled = true;
+ this.emit('cancelled', 'cancelled');
+ this.emit('streamEnd', false);
+ this.sendStatus({
+ code: Status.CANCELLED,
+ details: 'Cancelled by client',
+ metadata: null,
+ });
+ if (this.deadlineTimer) clearTimeout(this.deadlineTimer);
+ }
+ });
+
+ this.stream.on('drain', () => {
+ this.emit('drain');
+ });
+
+ if ('grpc.max_send_message_length' in options) {
+ this.maxSendMessageSize = options['grpc.max_send_message_length']!;
+ }
+ if ('grpc.max_receive_message_length' in options) {
+ this.maxReceiveMessageSize = options['grpc.max_receive_message_length']!;
+ }
+ }
+
+ private checkCancelled(): boolean {
+ /* In some cases the stream can become destroyed before the close event
+ * fires. That creates a race condition that this check works around */
+ if (this.stream.destroyed || this.stream.closed) {
+ this.cancelled = true;
+ }
+ return this.cancelled;
+ }
+
+ private getDecompressedMessage(
+ message: Buffer,
+ encoding: string
+ ): Buffer | Promise<Buffer> { const messageContents = message.subarray(5);
+ if (encoding === 'identity') {
+ return messageContents;
+ } else if (encoding === 'deflate' || encoding === 'gzip') {
+ let decompresser: zlib.Gunzip | zlib.Deflate;
+ if (encoding === 'deflate') {
+ decompresser = zlib.createInflate();
+ } else {
+ decompresser = zlib.createGunzip();
+ }
+ return new Promise((resolve, reject) => {
+ let totalLength = 0
+ const messageParts: Buffer[] = [];
+ decompresser.on('data', (chunk: Buffer) => {
+ messageParts.push(chunk);
+ totalLength += chunk.byteLength;
+ if (this.maxReceiveMessageSize !== -1 && totalLength > this.maxReceiveMessageSize) {
+ decompresser.destroy();
+ reject({
+ code: Status.RESOURCE_EXHAUSTED,
+ details: `Received message that decompresses to a size larger than ${this.maxReceiveMessageSize}`
+ });
+ }
+ });
+ decompresser.on('end', () => {
+ resolve(Buffer.concat(messageParts));
+ });
+ decompresser.write(messageContents);
+ decompresser.end();
+ });
+ } else {
+ return Promise.reject({
+ code: Status.UNIMPLEMENTED,
+ details: `Received message compressed with unsupported encoding "${encoding}"`,
+ });
+ }
+ }
+
+ sendMetadata(customMetadata?: Metadata) {
+ if (this.checkCancelled()) {
+ return;
+ }
+
+ if (this.metadataSent) {
+ return;
+ }
+
+ this.metadataSent = true;
+ const custom = customMetadata ? customMetadata.toHttp2Headers() : null;
+ // TODO(cjihrig): Include compression headers.
+ const headers = {
+ ...defaultResponseHeaders,
+ ...defaultCompressionHeaders,
+ ...custom,
+ };
+ this.stream.respond(headers, defaultResponseOptions);
+ }
+
+ receiveMetadata(headers: http2.IncomingHttpHeaders) {
+ const metadata = Metadata.fromHttp2Headers(headers);
+
+ if (logging.isTracerEnabled(TRACER_NAME)) {
+ trace(
+ 'Request to ' +
+ this.handler.path +
+ ' received headers ' +
+ JSON.stringify(metadata.toJSON())
+ );
+ }
+
+ // TODO(cjihrig): Receive compression metadata.
+
+ const timeoutHeader = metadata.get(GRPC_TIMEOUT_HEADER);
+
+ if (timeoutHeader.length > 0) {
+ const match = timeoutHeader[0].toString().match(DEADLINE_REGEX);
+
+ if (match === null) {
+ const err = new Error('Invalid deadline') as ServerErrorResponse;
+ err.code = Status.OUT_OF_RANGE;
+ this.sendError(err);
+ return metadata;
+ }
+
+ const timeout = (+match[1] * deadlineUnitsToMs[match[2]]) | 0;
+
+ const now = new Date();
+ this.deadline = now.setMilliseconds(now.getMilliseconds() + timeout);
+ this.deadlineTimer = setTimeout(handleExpiredDeadline, timeout, this);
+ metadata.remove(GRPC_TIMEOUT_HEADER);
+ }
+
+ // Remove several headers that should not be propagated to the application
+ metadata.remove(http2.constants.HTTP2_HEADER_ACCEPT_ENCODING);
+ metadata.remove(http2.constants.HTTP2_HEADER_TE);
+ metadata.remove(http2.constants.HTTP2_HEADER_CONTENT_TYPE);
+ metadata.remove('grpc-accept-encoding');
+
+ return metadata;
+ }
+
+ receiveUnaryMessage(encoding: string): Promise<RequestType> {
+ return new Promise((resolve, reject) => {
+ const { stream } = this;
+
+ let receivedLength = 0;
+
+ // eslint-disable-next-line @typescript-eslint/no-this-alias
+ const call = this;
+ const body: Buffer[] = [];
+ const limit = this.maxReceiveMessageSize;
+
+ this.stream.on('data', onData);
+ this.stream.on('end', onEnd);
+ this.stream.on('error', onEnd);
+
+ function onData(chunk: Buffer) {
+ receivedLength += chunk.byteLength;
+
+ if (limit !== -1 && receivedLength > limit) {
+ stream.removeListener('data', onData);
+ stream.removeListener('end', onEnd);
+ stream.removeListener('error', onEnd);
+
+ reject({
+ code: Status.RESOURCE_EXHAUSTED,
+ details: `Received message larger than max (${receivedLength} vs. ${limit})`,
+ });
+ return;
+ }
+
+ body.push(chunk);
+ }
+
+ function onEnd(err?: Error) {
+ stream.removeListener('data', onData);
+ stream.removeListener('end', onEnd);
+ stream.removeListener('error', onEnd);
+
+ if (err !== undefined) {
+ reject({ code: Status.INTERNAL, details: err.message });
+ return;
+ }
+
+ if (receivedLength === 0) {
+ reject({
+ code: Status.INTERNAL,
+ details: 'received empty unary message',
+ });
+ return;
+ }
+
+ call.emit('receiveMessage');
+
+ const requestBytes = Buffer.concat(body, receivedLength);
+ const compressed = requestBytes.readUInt8(0) === 1;
+ const compressedMessageEncoding = compressed ? encoding : 'identity';
+ const decompressedMessage = call.getDecompressedMessage(
+ requestBytes,
+ compressedMessageEncoding
+ );
+
+ if (Buffer.isBuffer(decompressedMessage)) {
+ resolve(
+ call.deserializeMessageWithInternalError(decompressedMessage)
+ );
+ return;
+ }
+
+ decompressedMessage.then(
+ decompressed =>
+ resolve(call.deserializeMessageWithInternalError(decompressed)),
+ (err: any) =>
+ reject(
+ err.code
+ ? err
+ : {
+ code: Status.INTERNAL,
+ details: `Received "grpc-encoding" header "${encoding}" but ${encoding} decompression failed`,
+ }
+ )
+ );
+ }
+ });
+ }
+
+ private async deserializeMessageWithInternalError(buffer: Buffer) {
+ try {
+ return this.deserializeMessage(buffer);
+ } catch (err) {
+ throw {
+ details: getErrorMessage(err),
+ code: Status.INTERNAL,
+ };
+ }
+ }
+
+ serializeMessage(value: ResponseType) {
+ const messageBuffer = this.handler.serialize(value);
+
+ // TODO(cjihrig): Call compression aware serializeMessage().
+ const byteLength = messageBuffer.byteLength;
+ const output = Buffer.allocUnsafe(byteLength + 5);
+ output.writeUInt8(0, 0);
+ output.writeUInt32BE(byteLength, 1);
+ messageBuffer.copy(output, 5);
+ return output;
+ }
+
+ deserializeMessage(bytes: Buffer) {
+ return this.handler.deserialize(bytes);
+ }
+
+ async sendUnaryMessage(
+ err: ServerErrorResponse | ServerStatusResponse | null,
+ value?: ResponseType | null,
+ metadata?: Metadata | null,
+ flags?: number
+ ) {
+ if (this.checkCancelled()) {
+ return;
+ }
+
+ if (metadata === undefined) {
+ metadata = null;
+ }
+
+ if (err) {
+ if (!Object.prototype.hasOwnProperty.call(err, 'metadata') && metadata) {
+ err.metadata = metadata;
+ }
+ this.sendError(err);
+ return;
+ }
+
+ try {
+ const response = this.serializeMessage(value!);
+
+ this.write(response);
+ this.sendStatus({ code: Status.OK, details: 'OK', metadata });
+ } catch (err) {
+ this.sendError({
+ details: getErrorMessage(err),
+ code: Status.INTERNAL,
+ });
+ }
+ }
+
+ sendStatus(statusObj: PartialStatusObject) {
+ this.emit('callEnd', statusObj.code);
+ this.emit('streamEnd', statusObj.code === Status.OK);
+ if (this.checkCancelled()) {
+ return;
+ }
+
+ trace(
+ 'Request to method ' +
+ this.handler?.path +
+ ' ended with status code: ' +
+ Status[statusObj.code] +
+ ' details: ' +
+ statusObj.details
+ );
+
+ if (this.deadlineTimer) clearTimeout(this.deadlineTimer);
+
+ if (this.stream.headersSent) {
+ if (!this.wantTrailers) {
+ this.wantTrailers = true;
+ this.stream.once('wantTrailers', () => {
+ const trailersToSend = {
+ [GRPC_STATUS_HEADER]: statusObj.code,
+ [GRPC_MESSAGE_HEADER]: encodeURI(statusObj.details),
+ ...statusObj.metadata?.toHttp2Headers(),
+ };
+
+ this.stream.sendTrailers(trailersToSend);
+ this.statusSent = true;
+ });
+ this.stream.end();
+ }
+ } else {
+ // Trailers-only response
+ const trailersToSend = {
+ [GRPC_STATUS_HEADER]: statusObj.code,
+ [GRPC_MESSAGE_HEADER]: encodeURI(statusObj.details),
+ ...defaultResponseHeaders,
+ ...statusObj.metadata?.toHttp2Headers(),
+ };
+ this.stream.respond(trailersToSend, { endStream: true });
+ this.statusSent = true;
+ }
+ }
+
+ sendError(error: ServerErrorResponse | ServerStatusResponse) {
+ const status: PartialStatusObject = {
+ code: Status.UNKNOWN,
+ details: 'message' in error ? error.message : 'Unknown Error',
+ metadata:
+ 'metadata' in error && error.metadata !== undefined
+ ? error.metadata
+ : null,
+ };
+
+ if (
+ 'code' in error &&
+ typeof error.code === 'number' &&
+ Number.isInteger(error.code)
+ ) {
+ status.code = error.code;
+
+ if ('details' in error && typeof error.details === 'string') {
+ status.details = error.details!;
+ }
+ }
+
+ this.sendStatus(status);
+ }
+
+ write(chunk: Buffer) {
+ if (this.checkCancelled()) {
+ return;
+ }
+
+ if (
+ this.maxSendMessageSize !== -1 &&
+ chunk.length > this.maxSendMessageSize
+ ) {
+ this.sendError({
+ code: Status.RESOURCE_EXHAUSTED,
+ details: `Sent message larger than max (${chunk.length} vs. ${this.maxSendMessageSize})`,
+ });
+ return;
+ }
+
+ this.sendMetadata();
+ this.emit('sendMessage');
+ return this.stream.write(chunk);
+ }
+
+ resume() {
+ this.stream.resume();
+ }
+
+ setupSurfaceCall(call: ServerSurfaceCall) {
+ this.once('cancelled', reason => {
+ call.cancelled = true;
+ call.emit('cancelled', reason);
+ });
+
+ this.once('callEnd', status => call.emit('callEnd', status));
+ }
+
+ setupReadable(
+ readable:
+ | ServerReadableStream<RequestType, ResponseType>
+ | ServerDuplexStream<RequestType, ResponseType>,
+ encoding: string
+ ) {
+ const decoder = new StreamDecoder(this.maxReceiveMessageSize);
+
+ let readsDone = false;
+
+ let pendingMessageProcessing = false;
+
+ let pushedEnd = false;
+
+ const maybePushEnd = async () => {
+ if (!pushedEnd && readsDone && !pendingMessageProcessing) {
+ pushedEnd = true;
+ await this.pushOrBufferMessage(readable, null);
+ }
+ };
+
+ this.stream.on('data', async (data: Buffer) => {
+ let messages: Buffer[];
+ try {
+ messages = decoder.write(data);
+ } catch (e) {
+ this.sendError({
+ code: Status.RESOURCE_EXHAUSTED,
+ details: (e as Error).message
+ });
+ return;
+ }
+
+ pendingMessageProcessing = true;
+ this.stream.pause();
+ for (const message of messages) {
+ this.emit('receiveMessage');
+
+ const compressed = message.readUInt8(0) === 1;
+ const compressedMessageEncoding = compressed ? encoding : 'identity';
+ let decompressedMessage: Buffer;
+ try {
+ decompressedMessage = await this.getDecompressedMessage(
+ message,
+ compressedMessageEncoding
+ );
+ } catch (e) {
+ this.sendError(e as Partial<StatusObject>);
+ return;
+ }
+
+ // Encountered an error with decompression; it'll already have been propogated back
+ // Just return early
+ if (!decompressedMessage) return;
+
+ await this.pushOrBufferMessage(readable, decompressedMessage);
+ }
+ pendingMessageProcessing = false;
+ this.stream.resume();
+ await maybePushEnd();
+ });
+
+ this.stream.once('end', async () => {
+ readsDone = true;
+ await maybePushEnd();
+ });
+ }
+
+ consumeUnpushedMessages(
+ readable:
+ | ServerReadableStream<RequestType, ResponseType>
+ | ServerDuplexStream<RequestType, ResponseType>
+ ): boolean {
+ this.canPush = true;
+
+ while (this.messagesToPush.length > 0) {
+ const nextMessage = this.messagesToPush.shift();
+ const canPush = readable.push(nextMessage);
+
+ if (nextMessage === null || canPush === false) {
+ this.canPush = false;
+ break;
+ }
+ }
+
+ return this.canPush;
+ }
+
+ private async pushOrBufferMessage(
+ readable:
+ | ServerReadableStream<RequestType, ResponseType>
+ | ServerDuplexStream<RequestType, ResponseType>,
+ messageBytes: Buffer | null
+ ): Promise<void> {
+ if (this.isPushPending) {
+ this.bufferedMessages.push(messageBytes);
+ } else {
+ await this.pushMessage(readable, messageBytes);
+ }
+ }
+
+ private async pushMessage(
+ readable:
+ | ServerReadableStream<RequestType, ResponseType>
+ | ServerDuplexStream<RequestType, ResponseType>,
+ messageBytes: Buffer | null
+ ) {
+ if (messageBytes === null) {
+ trace('Received end of stream');
+ if (this.canPush) {
+ readable.push(null);
+ } else {
+ this.messagesToPush.push(null);
+ }
+
+ return;
+ }
+
+ trace('Received message of length ' + messageBytes.length);
+
+ this.isPushPending = true;
+
+ try {
+ const deserialized = await this.deserializeMessage(messageBytes);
+
+ if (this.canPush) {
+ if (!readable.push(deserialized)) {
+ this.canPush = false;
+ this.stream.pause();
+ }
+ } else {
+ this.messagesToPush.push(deserialized);
+ }
+ } catch (error) {
+ // Ignore any remaining messages when errors occur.
+ this.bufferedMessages.length = 0;
+ let code = getErrorCode(error);
+ if (code === null || code < Status.OK || code > Status.UNAUTHENTICATED) {
+ code = Status.INTERNAL;
+ }
+
+ readable.emit('error', {
+ details: getErrorMessage(error),
+ code: code,
+ });
+ }
+
+ this.isPushPending = false;
+
+ if (this.bufferedMessages.length > 0) {
+ await this.pushMessage(
+ readable,
+ this.bufferedMessages.shift() as Buffer | null
+ );
+ }
+ }
+
+ 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';
+ }
+ }
+
+ getDeadline(): Deadline {
+ return this.deadline;
+ }
+
+ getPath(): string {
+ return this.handler.path;
+ }
+}
+
+/* eslint-disable @typescript-eslint/no-explicit-any */
+type UntypedServerCall = Http2ServerCallStream<any, any>;
+
+function handleExpiredDeadline(call: UntypedServerCall) {
+ const err = new Error('Deadline exceeded') as ServerErrorResponse;
+ err.code = Status.DEADLINE_EXCEEDED;
+
+ call.sendError(err);
+ call.cancelled = true;
+ call.emit('cancelled', 'deadline');
+}