1 | var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
2 | function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
3 | return new (P || (P = Promise))(function (resolve, reject) {
|
4 | function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
5 | function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
6 | function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
7 | step((generator = generator.apply(thisArg, _arguments || [])).next());
|
8 | });
|
9 | };
|
10 | var __generator = (this && this.__generator) || function (thisArg, body) {
|
11 | var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
|
12 | return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
|
13 | function verb(n) { return function (v) { return step([n, v]); }; }
|
14 | function step(op) {
|
15 | if (f) throw new TypeError("Generator is already executing.");
|
16 | while (_) try {
|
17 | 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;
|
18 | if (y = 0, t) op = [op[0] & 2, t.value];
|
19 | switch (op[0]) {
|
20 | case 0: case 1: t = op; break;
|
21 | case 4: _.label++; return { value: op[1], done: false };
|
22 | case 5: _.label++; y = op[1]; op = [0]; continue;
|
23 | case 7: op = _.ops.pop(); _.trys.pop(); continue;
|
24 | default:
|
25 | if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
|
26 | if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
|
27 | if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
|
28 | if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
|
29 | if (t[2]) _.ops.pop();
|
30 | _.trys.pop(); continue;
|
31 | }
|
32 | op = body.call(thisArg, _);
|
33 | } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
|
34 | if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
|
35 | }
|
36 | };
|
37 | var __read = (this && this.__read) || function (o, n) {
|
38 | var m = typeof Symbol === "function" && o[Symbol.iterator];
|
39 | if (!m) return o;
|
40 | var i = m.call(o), r, ar = [], e;
|
41 | try {
|
42 | while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
|
43 | }
|
44 | catch (error) { e = { error: error }; }
|
45 | finally {
|
46 | try {
|
47 | if (r && !r.done && (m = i["return"])) m.call(i);
|
48 | }
|
49 | finally { if (e) throw e.error; }
|
50 | }
|
51 | return ar;
|
52 | };
|
53 | var __spread = (this && this.__spread) || function () {
|
54 | for (var ar = [], i = 0; i < arguments.length; i++) ar = ar.concat(__read(arguments[i]));
|
55 | return ar;
|
56 | };
|
57 | import { ConsoleLogger as Logger } from '@aws-amplify/core';
|
58 | import { PutEventsCommand, } from '@aws-sdk/client-pinpoint';
|
59 | import { isAppInForeground } from '../utils/AppUtils';
|
60 | var logger = new Logger('EventsBuffer');
|
61 | var RETRYABLE_CODES = [429, 500];
|
62 | var ACCEPTED_CODES = [202];
|
63 | var EventsBuffer = (function () {
|
64 | function EventsBuffer(client, config) {
|
65 | this._pause = false;
|
66 | this._flush = false;
|
67 | logger.debug('Instantiating buffer with config:', config);
|
68 | this._buffer = [];
|
69 | this._client = client;
|
70 | this._config = config;
|
71 | this._sendBatch = this._sendBatch.bind(this);
|
72 | this._startLoop();
|
73 | }
|
74 | EventsBuffer.prototype.push = function (event) {
|
75 | var _a;
|
76 | if (this._buffer > this._config.bufferSize) {
|
77 | logger.debug('Exceeded analytics events buffer size');
|
78 | return event.handlers.reject(new Error('Exceeded the size of analytics events buffer'));
|
79 | }
|
80 | var eventId = event.params.event.eventId;
|
81 | var bufferElement = (_a = {}, _a[eventId] = event, _a);
|
82 | this._buffer.push(bufferElement);
|
83 | };
|
84 | EventsBuffer.prototype.pause = function () {
|
85 | this._pause = true;
|
86 | };
|
87 | EventsBuffer.prototype.resume = function () {
|
88 | this._pause = false;
|
89 | };
|
90 | EventsBuffer.prototype.updateClient = function (client) {
|
91 | this._client = client;
|
92 | };
|
93 | EventsBuffer.prototype.flush = function () {
|
94 | this._flush = true;
|
95 | };
|
96 | EventsBuffer.prototype._startLoop = function () {
|
97 | if (this._interval) {
|
98 | clearInterval(this._interval);
|
99 | }
|
100 | var flushInterval = this._config.flushInterval;
|
101 | this._interval = setInterval(this._sendBatch, flushInterval);
|
102 | };
|
103 | EventsBuffer.prototype._sendBatch = function () {
|
104 | var bufferLength = this._buffer.length;
|
105 | if (this._flush && !bufferLength) {
|
106 | clearInterval(this._interval);
|
107 | }
|
108 |
|
109 |
|
110 |
|
111 |
|
112 | if (this._pause || !bufferLength || !isAppInForeground()) {
|
113 | return;
|
114 | }
|
115 | var flushSize = this._config.flushSize;
|
116 | var batchSize = Math.min(flushSize, bufferLength);
|
117 | var bufferSubset = this._buffer.splice(0, batchSize);
|
118 | this._putEvents(bufferSubset);
|
119 | };
|
120 | EventsBuffer.prototype._putEvents = function (buffer) {
|
121 | return __awaiter(this, void 0, void 0, function () {
|
122 | var eventMap, batchEventParams, command, data, err_1;
|
123 | return __generator(this, function (_a) {
|
124 | switch (_a.label) {
|
125 | case 0:
|
126 | eventMap = this._bufferToMap(buffer);
|
127 | batchEventParams = this._generateBatchEventParams(eventMap);
|
128 | _a.label = 1;
|
129 | case 1:
|
130 | _a.trys.push([1, 3, , 4]);
|
131 | command = new PutEventsCommand(batchEventParams);
|
132 | return [4 , this._client.send(command)];
|
133 | case 2:
|
134 | data = _a.sent();
|
135 | this._processPutEventsSuccessResponse(data, eventMap);
|
136 | return [3 , 4];
|
137 | case 3:
|
138 | err_1 = _a.sent();
|
139 | return [2 , this._handlePutEventsFailure(err_1, eventMap)];
|
140 | case 4: return [2 ];
|
141 | }
|
142 | });
|
143 | });
|
144 | };
|
145 | EventsBuffer.prototype._generateBatchEventParams = function (eventMap) {
|
146 | var batchEventParams = {
|
147 | ApplicationId: '',
|
148 | EventsRequest: {
|
149 | BatchItem: {},
|
150 | },
|
151 | };
|
152 | Object.values(eventMap).forEach(function (item) {
|
153 | var params = item.params;
|
154 | var event = params.event, timestamp = params.timestamp, config = params.config;
|
155 | var name = event.name, attributes = event.attributes, metrics = event.metrics, eventId = event.eventId, session = event.session;
|
156 | var appId = config.appId, endpointId = config.endpointId;
|
157 | var batchItem = batchEventParams.EventsRequest.BatchItem;
|
158 | batchEventParams.ApplicationId = batchEventParams.ApplicationId || appId;
|
159 | if (!batchItem[endpointId]) {
|
160 | batchItem[endpointId] = {
|
161 | Endpoint: {},
|
162 | Events: {},
|
163 | };
|
164 | }
|
165 | batchItem[endpointId].Events[eventId] = {
|
166 | EventType: name,
|
167 | Timestamp: new Date(timestamp).toISOString(),
|
168 | Attributes: attributes,
|
169 | Metrics: metrics,
|
170 | Session: session,
|
171 | };
|
172 | });
|
173 | return batchEventParams;
|
174 | };
|
175 | EventsBuffer.prototype._handlePutEventsFailure = function (err, eventMap) {
|
176 | logger.debug('_putEvents Failed: ', err);
|
177 | var statusCode = err.$metadata && err.$metadata.httpStatusCode;
|
178 | if (RETRYABLE_CODES.includes(statusCode)) {
|
179 | var retryableEvents = Object.values(eventMap);
|
180 | this._retry(retryableEvents);
|
181 | return;
|
182 | }
|
183 | };
|
184 | EventsBuffer.prototype._processPutEventsSuccessResponse = function (data, eventMap) {
|
185 | var Results = data.EventsResponse.Results;
|
186 | var retryableEvents = [];
|
187 | Object.entries(Results).forEach(function (_a) {
|
188 | var _b = __read(_a, 2), endpointId = _b[0], endpointValues = _b[1];
|
189 | var responses = endpointValues.EventsItemResponse;
|
190 | Object.entries(responses).forEach(function (_a) {
|
191 | var _b, _c;
|
192 | var _d = __read(_a, 2), eventId = _d[0], _e = _d[1], StatusCode = _e.StatusCode, Message = _e.Message;
|
193 | var eventObject = eventMap[eventId];
|
194 |
|
195 | var response = {
|
196 | EventsResponse: {
|
197 | Results: (_b = {},
|
198 | _b[endpointId] = {
|
199 | EventsItemResponse: (_c = {},
|
200 | _c[eventId] = { StatusCode: StatusCode, Message: Message },
|
201 | _c),
|
202 | },
|
203 | _b),
|
204 | },
|
205 | };
|
206 | if (ACCEPTED_CODES.includes(StatusCode)) {
|
207 | eventObject.handlers.resolve(response);
|
208 | return;
|
209 | }
|
210 | if (RETRYABLE_CODES.includes(StatusCode)) {
|
211 | retryableEvents.push(eventObject);
|
212 | return;
|
213 | }
|
214 | var name = eventObject.params.event.name;
|
215 | logger.error("event " + eventId + " : " + name + " failed with error: " + Message);
|
216 | return eventObject.handlers.reject(response);
|
217 | });
|
218 | });
|
219 | if (retryableEvents.length) {
|
220 | this._retry(retryableEvents);
|
221 | }
|
222 | };
|
223 | EventsBuffer.prototype._retry = function (retryableEvents) {
|
224 | var _a;
|
225 |
|
226 | var eligibleEvents = [];
|
227 | retryableEvents.forEach(function (event) {
|
228 | var _a;
|
229 | var params = event.params;
|
230 | var _b = params.event, eventId = _b.eventId, name = _b.name;
|
231 | if (params.resendLimit-- > 0) {
|
232 | logger.debug("resending event " + eventId + " : " + name + " with " + params.resendLimit + " retry attempts remaining");
|
233 | eligibleEvents.push((_a = {}, _a[eventId] = event, _a));
|
234 | return;
|
235 | }
|
236 | logger.debug("no retry attempts remaining for event " + eventId + " : " + name);
|
237 | });
|
238 |
|
239 | (_a = this._buffer).unshift.apply(_a, __spread(eligibleEvents));
|
240 | };
|
241 |
|
242 |
|
243 | EventsBuffer.prototype._bufferToMap = function (buffer) {
|
244 | return buffer.reduce(function (acc, curVal) {
|
245 | var _a = __read(Object.entries(curVal), 1), _b = __read(_a[0], 2), key = _b[0], value = _b[1];
|
246 | acc[key] = value;
|
247 | return acc;
|
248 | }, {});
|
249 | };
|
250 | return EventsBuffer;
|
251 | }());
|
252 | export default EventsBuffer;
|
253 |
|
\ | No newline at end of file |