1 | "use strict";
|
2 | var __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 | };
|
11 | var __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 | };
|
38 | Object.defineProperty(exports, "__esModule", { value: true });
|
39 | var SHUFFLE_RESPONSE_TIMEOUT_MS = 30000;
|
40 | var OutgoingShuffleState = (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 |
|
50 |
|
51 | OutgoingShuffleState.prototype.storeChannel = function (theChannel) {
|
52 | this.channel = theChannel;
|
53 | };
|
54 | |
55 |
|
56 |
|
57 |
|
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 |
|
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 , 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 ];
|
77 | }
|
78 | });
|
79 | });
|
80 | };
|
81 | |
82 |
|
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 , new Promise(function (resolve) {
|
90 | _this.requireChannel().send("shuffleResponseAcknowledgement");
|
91 |
|
92 |
|
93 |
|
94 | _this.channelClosingTimeoutId = _this.asyncExecService.setTimeout(function () {
|
95 | resolve();
|
96 | }, 3000);
|
97 | })];
|
98 | case 1:
|
99 | _a.sent();
|
100 | return [2 ];
|
101 | }
|
102 | });
|
103 | });
|
104 | };
|
105 | |
106 |
|
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 | }());
|
128 | exports.OutgoingShuffleState = OutgoingShuffleState;
|
129 |
|
\ | No newline at end of file |