UNPKG

6.32 kBJavaScriptView Raw
1"use strict";
2var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3 function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4 return new (P || (P = Promise))(function (resolve, reject) {
5 function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6 function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7 function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8 step((generator = generator.apply(thisArg, _arguments || [])).next());
9 });
10};
11var __generator = (this && this.__generator) || function (thisArg, body) {
12 var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
13 return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
14 function verb(n) { return function (v) { return step([n, v]); }; }
15 function step(op) {
16 if (f) throw new TypeError("Generator is already executing.");
17 while (_) try {
18 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;
19 if (y = 0, t) op = [op[0] & 2, t.value];
20 switch (op[0]) {
21 case 0: case 1: t = op; break;
22 case 4: _.label++; return { value: op[1], done: false };
23 case 5: _.label++; y = op[1]; op = [0]; continue;
24 case 7: op = _.ops.pop(); _.trys.pop(); continue;
25 default:
26 if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
27 if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
28 if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
29 if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
30 if (t[2]) _.ops.pop();
31 _.trys.pop(); continue;
32 }
33 op = body.call(thisArg, _);
34 } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
35 if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
36 }
37};
38Object.defineProperty(exports, "__esModule", { value: true });
39var SHUFFLE_RESPONSE_TIMEOUT_MS = 30000;
40var OutgoingShuffleState = /** @class */ (function () {
41 function OutgoingShuffleState(fromNode, destinationNodePointer, shuffleSet, asyncExecService, logger) {
42 this.fromNode = fromNode;
43 this.destinationNodePointer = destinationNodePointer;
44 this.shuffleSet = shuffleSet;
45 this.asyncExecService = asyncExecService;
46 this.logger = logger;
47 }
48 /**
49 * Store the channel for later use
50 */
51 OutgoingShuffleState.prototype.storeChannel = function (theChannel) {
52 this.channel = theChannel;
53 };
54 /**
55 * Send a shuffle request
56 *
57 * @returns {Promise}
58 */
59 OutgoingShuffleState.prototype.sendShuffleRequest = function () {
60 this.requireChannel().send("shuffleRequest", this.shuffleSet);
61 this.logger.debug("Sent shuffle request to " + this.destinationNodePointer.id + " : " + JSON.stringify(this.shuffleSet));
62 };
63 /**
64 * Receive and process a shuffle response
65 */
66 OutgoingShuffleState.prototype.processShuffleResponse = function () {
67 return __awaiter(this, void 0, void 0, function () {
68 var shuffleResponseMessage;
69 return __generator(this, function (_a) {
70 switch (_a.label) {
71 case 0: return [4 /*yield*/, this.requireChannel().receive("shuffleResponse", SHUFFLE_RESPONSE_TIMEOUT_MS)];
72 case 1:
73 shuffleResponseMessage = _a.sent();
74 this.logger.debug("Received shuffle response from " + this.destinationNodePointer.id + " : " + JSON.stringify(shuffleResponseMessage));
75 this.fromNode.handleShuffleResponse(this.destinationNodePointer, shuffleResponseMessage);
76 return [2 /*return*/];
77 }
78 });
79 });
80 };
81 /**
82 * Send an acknowledgement we received the response
83 */
84 OutgoingShuffleState.prototype.sendResponseAcknowledgement = function () {
85 return __awaiter(this, void 0, void 0, function () {
86 var _this = this;
87 return __generator(this, function (_a) {
88 switch (_a.label) {
89 case 0: return [4 /*yield*/, new Promise(function (resolve) {
90 _this.requireChannel().send("shuffleResponseAcknowledgement");
91 //
92 // Delay closing connection to allow acknowledgement to be sent (?)
93 //
94 _this.channelClosingTimeoutId = _this.asyncExecService.setTimeout(function () {
95 resolve();
96 }, 3000);
97 })];
98 case 1:
99 _a.sent();
100 return [2 /*return*/];
101 }
102 });
103 });
104 };
105 /**
106 * Cleanup any resources
107 */
108 OutgoingShuffleState.prototype.close = function () {
109 if (this.channel) {
110 this.channel.close();
111 }
112 this.clearChannelClosingTimeout();
113 };
114 OutgoingShuffleState.prototype.clearChannelClosingTimeout = function () {
115 if (this.channelClosingTimeoutId) {
116 this.asyncExecService.clearTimeout(this.channelClosingTimeoutId);
117 delete this.channelClosingTimeoutId;
118 }
119 };
120 OutgoingShuffleState.prototype.requireChannel = function () {
121 if (this.channel === undefined) {
122 throw new Error("Channel must have been stored first!");
123 }
124 return this.channel;
125 };
126 return OutgoingShuffleState;
127}());
128exports.OutgoingShuffleState = OutgoingShuffleState;
129//# sourceMappingURL=OutgoingShuffleState.js.map
\No newline at end of file