mpmc
Version:
Async multi-producer, multi-consumer FIFO queue communication utilies
215 lines (214 loc) • 10.3 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (_) try {
if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
var __asyncValues = (this && this.__asyncValues) || function (o) {
if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
var m = o[Symbol.asyncIterator], i;
return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
};
import { channel, onceChannel } from ".";
var A = Symbol();
var B = Symbol();
var C = Symbol();
var D = Symbol();
test("onceChannel", function () { return __awaiter(void 0, void 0, void 0, function () {
var _a, resolve, promise, start, _b;
return __generator(this, function (_c) {
switch (_c.label) {
case 0:
_a = onceChannel(), resolve = _a[0], promise = _a[1];
setTimeout(resolve, 20, A);
start = Date.now();
_b = expect;
return [4 /*yield*/, promise];
case 1:
_b.apply(void 0, [_c.sent()]).toBe(A);
expect((Date.now() - start) / 1000).toBeCloseTo(0.02, 2);
return [2 /*return*/];
}
});
}); });
describe("channel", function () {
var sender;
var receiver;
beforeEach(function () {
var _a;
_a = channel(), sender = _a[0], receiver = _a[1];
});
test("recv order", function () { return __awaiter(void 0, void 0, void 0, function () {
var _a, _b, _c, _d;
return __generator(this, function (_e) {
switch (_e.label) {
case 0:
sender.send(A);
setTimeout(function () {
sender.send(B, C);
sender.send(D);
});
_a = expect;
return [4 /*yield*/, receiver.recv()];
case 1:
_a.apply(void 0, [_e.sent()]).toBe(A);
_b = expect;
return [4 /*yield*/, receiver.recv()];
case 2:
_b.apply(void 0, [_e.sent()]).toBe(B);
_c = expect;
return [4 /*yield*/, receiver.recv()];
case 3:
_c.apply(void 0, [_e.sent()]).toBe(C);
_d = expect;
return [4 /*yield*/, receiver.recv()];
case 4:
_d.apply(void 0, [_e.sent()]).toBe(D);
return [2 /*return*/];
}
});
}); });
test("close", function () { return __awaiter(void 0, void 0, void 0, function () {
var _a, _b, _c;
return __generator(this, function (_d) {
switch (_d.label) {
case 0:
expect(sender.send(A)).toBe(true);
expect(receiver.close()).toBe(true);
expect(sender.send(B)).toBe(false);
expect(receiver.close()).toBe(false);
_a = expect;
return [4 /*yield*/, receiver.recv()];
case 1:
_a.apply(void 0, [_d.sent()]).toBe(A);
_b = expect;
return [4 /*yield*/, receiver.recv()];
case 2:
_b.apply(void 0, [_d.sent()]).toBe(null);
_c = expect;
return [4 /*yield*/, receiver.recv()];
case 3:
_c.apply(void 0, [_d.sent()]).toBe(null);
return [2 /*return*/];
}
});
}); });
test("async iterator", function () { return __awaiter(void 0, void 0, void 0, function () {
var collected, receiver_1, receiver_1_1, msg, e_1_1;
var e_1, _a;
return __generator(this, function (_b) {
switch (_b.label) {
case 0:
sender.send(A, B, C);
receiver.close();
collected = [];
_b.label = 1;
case 1:
_b.trys.push([1, 6, 7, 12]);
receiver_1 = __asyncValues(receiver);
_b.label = 2;
case 2: return [4 /*yield*/, receiver_1.next()];
case 3:
if (!(receiver_1_1 = _b.sent(), !receiver_1_1.done)) return [3 /*break*/, 5];
msg = receiver_1_1.value;
collected.push(msg);
_b.label = 4;
case 4: return [3 /*break*/, 2];
case 5: return [3 /*break*/, 12];
case 6:
e_1_1 = _b.sent();
e_1 = { error: e_1_1 };
return [3 /*break*/, 12];
case 7:
_b.trys.push([7, , 10, 11]);
if (!(receiver_1_1 && !receiver_1_1.done && (_a = receiver_1["return"]))) return [3 /*break*/, 9];
return [4 /*yield*/, _a.call(receiver_1)];
case 8:
_b.sent();
_b.label = 9;
case 9: return [3 /*break*/, 11];
case 10:
if (e_1) throw e_1.error;
return [7 /*endfinally*/];
case 11: return [7 /*endfinally*/];
case 12:
expect(collected).toStrictEqual([A, B, C]);
return [2 /*return*/];
}
});
}); });
test("multi-consumer", function () { return __awaiter(void 0, void 0, void 0, function () {
var _a, _b, _c, _d;
return __generator(this, function (_e) {
switch (_e.label) {
case 0:
setTimeout(sender.send, 0, A, B, C);
setTimeout(receiver.close);
_a = expect;
return [4 /*yield*/, Promise.all([receiver.recv(), receiver.recv(), receiver.recv()])];
case 1:
_a.apply(void 0, [_e.sent()]).toStrictEqual([A, A, A]);
_b = expect;
return [4 /*yield*/, Promise.all([receiver.recv(), receiver.recv(), receiver.recv()])];
case 2:
_b.apply(void 0, [_e.sent()]).toStrictEqual([B, B, B]);
_c = expect;
return [4 /*yield*/, Promise.all([receiver.recv(), receiver.recv(), receiver.recv()])];
case 3:
_c.apply(void 0, [_e.sent()]).toStrictEqual([C, C, C]);
_d = expect;
return [4 /*yield*/, Promise.all([receiver.recv(), receiver.recv(), receiver.recv()])];
case 4:
_d.apply(void 0, [_e.sent()]).toStrictEqual([null, null, null]);
return [2 /*return*/];
}
});
}); });
test("collect", function () { return __awaiter(void 0, void 0, void 0, function () {
var _a;
return __generator(this, function (_b) {
switch (_b.label) {
case 0:
setTimeout(sender.send, 0, A, B);
setTimeout(receiver.close);
_a = expect;
return [4 /*yield*/, receiver.collect()];
case 1:
_a.apply(void 0, [_b.sent()]).toStrictEqual([A, B]);
return [2 /*return*/];
}
});
}); });
});