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-16 19:08:29 +0800
committeraltaf-creator <dev@altafcreator.com>2025-11-16 19:08:29 +0800
commit434aa8343fdcbb4d5002f934979913c099489bee (patch)
tree55bab4ec5a6151be57797d34f61faf5ea744471b /frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
parent893c388d4e99442a36005e5971a87730623f946e (diff)
sdk, del
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, 0 insertions, 1022 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
deleted file mode 100644
index 3032c6e..0000000
--- a/frontend-old/node_modules/@grpc/grpc-js/src/server-call.ts
+++ /dev/null
@@ -1,1022 +0,0 @@
-/*
- * 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');
-}