Files

120 lines
3.1 KiB
TypeScript

import { bench, describe } from 'vitest';
import { Reader, Writer, router } from '../../plugin/index.ts';
import {
Book,
Order,
Ticker,
buildBook,
buildOrder,
buildTicker,
} from './payloads.ts';
const ticker = buildTicker();
const order = buildOrder();
const book = buildBook(1000);
// Pre-allocated pooled Writers (sized generously so we don't measure grow()).
const wTicker = new Writer(256);
const wOrder = new Writer(256);
const wBook = new Writer(64 * 1024);
// Pre-encoded buffers for decode benches.
const tickerJSON = JSON.stringify(ticker);
const orderJSON = JSON.stringify(order);
const bookJSON = JSON.stringify(book);
const tickerBin = Ticker.encode(ticker);
const orderBin = Order.encode(order);
const bookBin = Book.encode(book);
// One-time payload-size print on module load.
// eslint-disable-next-line no-console
console.log(
'\n--- payload sizes ---\n' +
`ticker | json: ${tickerJSON.length}b bin: ${tickerBin.length}b (${((tickerBin.length / tickerJSON.length) * 100).toFixed(0)}%)\n` +
`order | json: ${orderJSON.length}b bin: ${orderBin.length}b (${((orderBin.length / orderJSON.length) * 100).toFixed(0)}%)\n` +
`book | json: ${bookJSON.length}b bin: ${bookBin.length}b (${((bookBin.length / bookJSON.length) * 100).toFixed(0)}%)\n`,
);
describe('encode ticker (5 fields)', () => {
bench('JSON.stringify', () => {
JSON.stringify(ticker);
});
bench('codec.encodeInto (pooled)', () => {
wTicker.reset();
Ticker.encodeInto(ticker, wTicker);
});
});
describe('encode order (10 fields + bitset)', () => {
bench('JSON.stringify', () => {
JSON.stringify(order);
});
bench('codec.encodeInto (pooled)', () => {
wOrder.reset();
Order.encodeInto(order, wOrder);
});
});
describe('encode book (1000 levels)', () => {
bench('JSON.stringify', () => {
JSON.stringify(book);
});
bench('codec.encodeInto (pooled)', () => {
wBook.reset();
Book.encodeInto(book, wBook);
});
});
describe('decode ticker', () => {
bench('JSON.parse', () => {
JSON.parse(tickerJSON);
});
bench('codec.decodeFrom', () => {
const r = new Reader(tickerBin);
Ticker.decodeFrom(r);
});
});
describe('decode order', () => {
bench('JSON.parse', () => {
JSON.parse(orderJSON);
});
bench('codec.decodeFrom', () => {
const r = new Reader(orderBin);
Order.decodeFrom(r);
});
});
describe('decode book (1000 levels)', () => {
bench('JSON.parse', () => {
JSON.parse(bookJSON);
});
bench('codec.decodeFrom', () => {
const r = new Reader(bookBin);
Book.decodeFrom(r);
});
});
describe('roundtrip ticker', () => {
bench('JSON', () => {
JSON.parse(JSON.stringify(ticker));
});
bench('codec (pooled)', () => {
wTicker.reset();
Ticker.encodeInto(ticker, wTicker);
const r = new Reader(wTicker.bytes());
Ticker.decodeFrom(r);
});
});
describe('framed ticker via router', () => {
const proto = router(Ticker, Order, Book);
bench('JSON', () => {
JSON.parse(JSON.stringify(ticker));
});
bench('router encode + decode (framed)', () => {
proto.decode(proto.encode(ticker, Ticker));
});
});