archive-gh-me/grpc-node
1
0
Fork 0
mirror of https://github.com/grpc/grpc-node.git synced 2025-12-08 18:23:54 +00:00
Code Issues Projects Releases Wiki Activity

rewrite CompressionAlgorithms to use numeric keys and export it

Browse Source
This commit is contained in:
Robert 2021-10-25 19:14:01 -07:00
parent 9d70f39b7e
commit 959503ec94
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
packages/grpc-js/src/compression-filter.ts
View File

@ -23,16 +23,16 @@ import { BaseFilter, Filter, FilterFactory } from './filter';
import { Metadata, MetadataValue } from './metadata';
import { ChannelOptions } from './channel-options';
const CompressionAlgorithms = {
'0': 'identity',
'1': 'deflate',
'2': 'gzip'
export const CompressionAlgorithms = {
0: 'identity',
1: 'deflate',
2: 'gzip'
} as const;
const CompressionAlgorithKeys = new Set(Object.keys(CompressionAlgorithms));
const isCompressionAlgorithmKey = (key: string | undefined): key is keyof typeof CompressionAlgorithms => {
return typeof key === 'string' && CompressionAlgorithKeys.has(key);
const isCompressionAlgorithmKey = (key: number | undefined): key is keyof typeof CompressionAlgorithms => {
return typeof key === 'number' && CompressionAlgorithKeys.has(key.toString());
}
type CompressionAlgorithm = (typeof CompressionAlgorithms)[keyof typeof CompressionAlgorithms];
@ -187,7 +187,7 @@ export class CompressionFilter extends BaseFilter implements Filter {
constructor(channelOptions: ChannelOptions) {
super();
const compressionAlgorithmKey = channelOptions['grpc.default_compression_algorithm']?.toString();
const compressionAlgorithmKey = channelOptions['grpc.default_compression_algorithm'];
if (isCompressionAlgorithmKey(compressionAlgorithmKey)) {
this.defaultCompressionAlgorithm = CompressionAlgorithms[compressionAlgorithmKey];
this.sendCompression = getCompressionHandler(this.defaultCompressionAlgorithm);

2
packages/grpc-js/src/index.ts
View File

@ -25,6 +25,7 @@ import {
import { CallCredentials, OAuth2Client } from './call-credentials';
import { Deadline, StatusObject } from './call-stream';
import { Channel, ChannelImplementation } from './channel';
import { CompressionAlgorithms } from './compression-filter';
import { ConnectivityState } from './connectivity-state';
import { ChannelCredentials } from './channel-credentials';
import {
@ -124,6 +125,7 @@ export {
Status as status,
ConnectivityState as connectivityState,
Propagate as propagate,
CompressionAlgorithms as compressionAlgorithms
// TODO: Other constants as well
};