mirror of
https://github.com/swc-project/swc.git
synced 2024-11-28 11:13:43 +03:00
226 lines
5.3 KiB
TypeScript
226 lines
5.3 KiB
TypeScript
// Loaded from https://raw.githubusercontent.com/nats-io/nats.deno/v1.0.0-11/nats-base-client/util.ts
|
|
|
|
|
|
/*
|
|
* Copyright 2018-2020 The NATS 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.
|
|
*/
|
|
//@ts-ignore
|
|
import { DataBuffer } from "./databuffer.ts";
|
|
import { ErrorCode, NatsError } from "./error.ts";
|
|
import { TD } from "./encoders.ts";
|
|
|
|
export const CR_LF = "\r\n";
|
|
export const CR_LF_LEN = CR_LF.length;
|
|
export const CRLF = DataBuffer.fromAscii(CR_LF);
|
|
export const CR = new Uint8Array(CRLF)[0]; // 13
|
|
export const LF = new Uint8Array(CRLF)[1]; // 10
|
|
|
|
export function isUint8Array(a: any): boolean {
|
|
return a instanceof Uint8Array;
|
|
}
|
|
|
|
export function protoLen(ba: Uint8Array): number {
|
|
for (let i = 0; i < ba.length; i++) {
|
|
let n = i + 1;
|
|
if (ba.byteLength > n && ba[i] === CR && ba[n] === LF) {
|
|
return n + 1;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
export function extractProtocolMessage(a: Uint8Array): string {
|
|
// protocol messages are ascii, so Uint8Array
|
|
let len = protoLen(a);
|
|
if (len) {
|
|
let ba = new Uint8Array(a);
|
|
let out = ba.slice(0, len);
|
|
return TD.decode(out);
|
|
}
|
|
return "";
|
|
}
|
|
|
|
export function extend(a: any, ...b: any[]): any {
|
|
for (let i = 0; i < b.length; i++) {
|
|
let o = b[i];
|
|
Object.keys(o).forEach(function (k) {
|
|
a[k] = o[k];
|
|
});
|
|
}
|
|
return a;
|
|
}
|
|
|
|
export function settle(a: any[]): Promise<any[]> {
|
|
if (Array.isArray(a)) {
|
|
return Promise.resolve(a).then(_settle);
|
|
} else {
|
|
return Promise.reject(
|
|
new TypeError("argument requires an array of promises"),
|
|
);
|
|
}
|
|
}
|
|
|
|
function _settle(a: any[]): Promise<any> {
|
|
return Promise.all(a.map((p) => {
|
|
return Promise.resolve(p).then(_resolve, _resolve);
|
|
}));
|
|
}
|
|
|
|
function _resolve(r: any): any {
|
|
return r;
|
|
}
|
|
|
|
export interface Pending {
|
|
pending: number;
|
|
write: (c: number) => void;
|
|
wrote: (c: number) => void;
|
|
err: (err: Error) => void;
|
|
close: () => void;
|
|
promise: () => Promise<any>;
|
|
resolved: boolean;
|
|
done: boolean;
|
|
}
|
|
|
|
export function pending(): Pending {
|
|
const v = {} as Pending;
|
|
const promise = new Promise((resolve) => {
|
|
v.promise = () => {
|
|
return promise;
|
|
};
|
|
v.write = (c: number) => {
|
|
if (v.resolved) {
|
|
return;
|
|
}
|
|
v.pending += c;
|
|
};
|
|
v.wrote = (c: number) => {
|
|
if (v.resolved) {
|
|
return;
|
|
}
|
|
v.pending -= c;
|
|
if (v.done && 0 >= v.pending) {
|
|
resolve();
|
|
}
|
|
};
|
|
v.close = () => {
|
|
v.done = true;
|
|
if (v.pending === 0) {
|
|
resolve();
|
|
}
|
|
};
|
|
v.err = () => {
|
|
v.pending = 0;
|
|
v.resolved = true;
|
|
v.close();
|
|
};
|
|
});
|
|
return v;
|
|
}
|
|
|
|
export function render(frame: Uint8Array): string {
|
|
const cr = "␍";
|
|
const lf = "␊";
|
|
return TD.decode(frame)
|
|
.replace(/\n/g, lf)
|
|
.replace(/\r/g, cr);
|
|
}
|
|
|
|
export interface Timeout<T> extends Promise<T> {
|
|
cancel: () => void;
|
|
}
|
|
|
|
export function timeout<T>(ms: number): Timeout<T> {
|
|
let methods;
|
|
let timer: number;
|
|
const p = new Promise((resolve, reject) => {
|
|
let cancel = (): void => {
|
|
if (timer) {
|
|
clearTimeout(timer);
|
|
}
|
|
};
|
|
methods = { cancel };
|
|
// node is not a number
|
|
// @ts-ignore
|
|
timer = setTimeout(() => {
|
|
reject(NatsError.errorForCode(ErrorCode.TIMEOUT));
|
|
}, ms);
|
|
});
|
|
// noinspection JSUnusedAssignment
|
|
return Object.assign(p, methods) as Timeout<T>;
|
|
}
|
|
|
|
export function delay<T>(ms: number = 0, value?: T): Promise<T> {
|
|
return new Promise((resolve) => {
|
|
setTimeout(() => {
|
|
resolve(value);
|
|
}, ms);
|
|
});
|
|
}
|
|
|
|
export interface Deferred<T> extends Promise<T> {
|
|
resolve: (value?: T | PromiseLike<T>) => void;
|
|
//@ts-ignore
|
|
reject: (reason?: any) => void;
|
|
}
|
|
|
|
export function deferred<T>(): Deferred<T> {
|
|
let methods = {};
|
|
const p = new Promise<T>((resolve, reject): void => {
|
|
methods = { resolve, reject };
|
|
});
|
|
return Object.assign(p, methods) as Deferred<T>;
|
|
}
|
|
|
|
export function shuffle(a: any[]): any[] {
|
|
for (let i = a.length - 1; i > 0; i--) {
|
|
const j = Math.floor(Math.random() * (i + 1));
|
|
[a[i], a[j]] = [a[j], a[i]];
|
|
}
|
|
return a;
|
|
}
|
|
|
|
export class Perf {
|
|
timers: Map<string, number>;
|
|
measures: Map<string, number>;
|
|
|
|
constructor() {
|
|
this.timers = new Map();
|
|
this.measures = new Map();
|
|
}
|
|
|
|
mark(key: string) {
|
|
this.timers.set(key, Date.now());
|
|
}
|
|
|
|
measure(key: string, startKey: string, endKey: string) {
|
|
const s = this.timers.get(startKey);
|
|
if (s === undefined) {
|
|
throw new Error(`${startKey} is not defined`);
|
|
}
|
|
const e = this.timers.get(endKey);
|
|
if (e === undefined) {
|
|
throw new Error(`${endKey} is not defined`);
|
|
}
|
|
this.measures.set(key, e - s);
|
|
}
|
|
|
|
getEntries(): { name: string; duration: number }[] {
|
|
const values: { name: string; duration: number }[] = [];
|
|
this.measures.forEach((v, k) => {
|
|
values.push({ name: k, duration: v });
|
|
});
|
|
return values;
|
|
}
|
|
}
|