@alwaysai/device-agent 1.4.0 → 2.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/lib/application-control/config.d.ts.map +1 -1
- package/lib/application-control/config.js +10 -5
- package/lib/application-control/config.js.map +1 -1
- package/lib/application-control/environment-variables.d.ts +1 -5
- package/lib/application-control/environment-variables.d.ts.map +1 -1
- package/lib/application-control/environment-variables.js +9 -26
- package/lib/application-control/environment-variables.js.map +1 -1
- package/lib/application-control/environment-variables.test.js +27 -7
- package/lib/application-control/environment-variables.test.js.map +1 -1
- package/lib/application-control/index.d.ts +4 -4
- package/lib/application-control/index.d.ts.map +1 -1
- package/lib/application-control/index.js +1 -4
- package/lib/application-control/index.js.map +1 -1
- package/lib/application-control/install.d.ts.map +1 -1
- package/lib/application-control/install.js +9 -7
- package/lib/application-control/install.js.map +1 -1
- package/lib/application-control/models.d.ts +5 -11
- package/lib/application-control/models.d.ts.map +1 -1
- package/lib/application-control/models.js +27 -64
- package/lib/application-control/models.js.map +1 -1
- package/lib/application-control/status.d.ts.map +1 -1
- package/lib/application-control/status.js +10 -12
- package/lib/application-control/status.js.map +1 -1
- package/lib/application-control/utils.d.ts +0 -4
- package/lib/application-control/utils.d.ts.map +1 -1
- package/lib/application-control/utils.js +3 -26
- package/lib/application-control/utils.js.map +1 -1
- package/lib/cloud-connection/bootstrap-provision.js +3 -2
- package/lib/cloud-connection/bootstrap-provision.js.map +1 -1
- package/lib/cloud-connection/device-agent-cloud-connection.d.ts +11 -16
- package/lib/cloud-connection/device-agent-cloud-connection.d.ts.map +1 -1
- package/lib/cloud-connection/device-agent-cloud-connection.js +295 -246
- package/lib/cloud-connection/device-agent-cloud-connection.js.map +1 -1
- package/lib/cloud-connection/device-agent.d.ts.map +1 -1
- package/lib/cloud-connection/device-agent.js +11 -9
- package/lib/cloud-connection/device-agent.js.map +1 -1
- package/lib/cloud-connection/live-updates-handler.d.ts +18 -27
- package/lib/cloud-connection/live-updates-handler.d.ts.map +1 -1
- package/lib/cloud-connection/live-updates-handler.js +58 -170
- package/lib/cloud-connection/live-updates-handler.js.map +1 -1
- package/lib/cloud-connection/live-updates-handler.test.js +76 -54
- package/lib/cloud-connection/live-updates-handler.test.js.map +1 -1
- package/lib/cloud-connection/passthrough-handler.d.ts +9 -4
- package/lib/cloud-connection/passthrough-handler.d.ts.map +1 -1
- package/lib/cloud-connection/passthrough-handler.js +95 -62
- package/lib/cloud-connection/passthrough-handler.js.map +1 -1
- package/lib/cloud-connection/shadow-handler.d.ts +5 -1
- package/lib/cloud-connection/shadow-handler.d.ts.map +1 -1
- package/lib/cloud-connection/shadow-handler.js +63 -31
- package/lib/cloud-connection/shadow-handler.js.map +1 -1
- package/lib/cloud-connection/shadow-handler.test.js +45 -57
- package/lib/cloud-connection/shadow-handler.test.js.map +1 -1
- package/lib/cloud-connection/shadow.d.ts.map +1 -1
- package/lib/cloud-connection/shadow.js +2 -1
- package/lib/cloud-connection/shadow.js.map +1 -1
- package/lib/cloud-connection/transaction-manager.d.ts +7 -2
- package/lib/cloud-connection/transaction-manager.d.ts.map +1 -1
- package/lib/cloud-connection/transaction-manager.js +29 -29
- package/lib/cloud-connection/transaction-manager.js.map +1 -1
- package/lib/cloud-connection/transaction-manager.test.js +105 -3
- package/lib/cloud-connection/transaction-manager.test.js.map +1 -1
- package/lib/device-control/device-control.d.ts +14 -6
- package/lib/device-control/device-control.d.ts.map +1 -1
- package/lib/device-control/device-control.js +172 -72
- package/lib/device-control/device-control.js.map +1 -1
- package/lib/docker/docker-compose.d.ts +14 -0
- package/lib/docker/docker-compose.d.ts.map +1 -0
- package/lib/docker/docker-compose.js +57 -0
- package/lib/docker/docker-compose.js.map +1 -0
- package/lib/index.js +2 -5
- package/lib/index.js.map +1 -1
- package/lib/infrastructure/agent-config.d.ts +46 -14
- package/lib/infrastructure/agent-config.d.ts.map +1 -1
- package/lib/infrastructure/agent-config.js +36 -21
- package/lib/infrastructure/agent-config.js.map +1 -1
- package/lib/infrastructure/agent-config.test.js +6 -1
- package/lib/infrastructure/agent-config.test.js.map +1 -1
- package/lib/infrastructure/config-check-utility.d.ts +6 -0
- package/lib/infrastructure/config-check-utility.d.ts.map +1 -0
- package/lib/infrastructure/config-check-utility.js +67 -0
- package/lib/infrastructure/config-check-utility.js.map +1 -0
- package/lib/infrastructure/config-check-utility.test.d.ts +2 -0
- package/lib/infrastructure/config-check-utility.test.d.ts.map +1 -0
- package/lib/infrastructure/config-check-utility.test.js +109 -0
- package/lib/infrastructure/config-check-utility.test.js.map +1 -0
- package/lib/infrastructure/device-certificate.d.ts +10 -0
- package/lib/infrastructure/device-certificate.d.ts.map +1 -0
- package/lib/infrastructure/device-certificate.js +47 -0
- package/lib/infrastructure/device-certificate.js.map +1 -0
- package/lib/infrastructure/device-certificate.test.d.ts +2 -0
- package/lib/infrastructure/device-certificate.test.d.ts.map +1 -0
- package/lib/infrastructure/device-certificate.test.js +24 -0
- package/lib/infrastructure/device-certificate.test.js.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-file.test.d.ts +2 -0
- package/lib/infrastructure/legacy-migration/legacy-file.test.d.ts.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-file.test.js +61 -0
- package/lib/infrastructure/legacy-migration/legacy-file.test.js.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-files.d.ts +75 -0
- package/lib/infrastructure/legacy-migration/legacy-files.d.ts.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-files.js +75 -0
- package/lib/infrastructure/legacy-migration/legacy-files.js.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.d.ts +6 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.d.ts.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.js +149 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.js.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.test.d.ts +2 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.test.d.ts.map +1 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.test.js +226 -0
- package/lib/infrastructure/legacy-migration/legacy-migration.test.js.map +1 -0
- package/lib/infrastructure/require-files-present-ready.test.d.ts +2 -0
- package/lib/infrastructure/require-files-present-ready.test.d.ts.map +1 -0
- package/lib/infrastructure/require-files-present-ready.test.js +44 -0
- package/lib/infrastructure/require-files-present-ready.test.js.map +1 -0
- package/lib/infrastructure/required-config-checks.d.ts +2 -0
- package/lib/infrastructure/required-config-checks.d.ts.map +1 -0
- package/lib/infrastructure/required-config-checks.js +30 -0
- package/lib/infrastructure/required-config-checks.js.map +1 -0
- package/lib/infrastructure/tokens-and-device-cfg.d.ts.map +1 -1
- package/lib/infrastructure/tokens-and-device-cfg.js +11 -8
- package/lib/infrastructure/tokens-and-device-cfg.js.map +1 -1
- package/lib/local-connection/rabbitmq-connection.d.ts.map +1 -1
- package/lib/local-connection/rabbitmq-connection.js +21 -21
- package/lib/local-connection/rabbitmq-connection.js.map +1 -1
- package/lib/secure-tunneling/secure-tunneling.d.ts +15 -23
- package/lib/secure-tunneling/secure-tunneling.d.ts.map +1 -1
- package/lib/secure-tunneling/secure-tunneling.js +52 -47
- package/lib/secure-tunneling/secure-tunneling.js.map +1 -1
- package/lib/secure-tunneling/secure-tunneling.test.js +29 -31
- package/lib/secure-tunneling/secure-tunneling.test.js.map +1 -1
- package/lib/subcommands/app/analytics.d.ts.map +1 -1
- package/lib/subcommands/app/analytics.js +1 -2
- package/lib/subcommands/app/analytics.js.map +1 -1
- package/lib/subcommands/app/env-vars.d.ts +4 -0
- package/lib/subcommands/app/env-vars.d.ts.map +1 -1
- package/lib/subcommands/app/env-vars.js +52 -6
- package/lib/subcommands/app/env-vars.js.map +1 -1
- package/lib/subcommands/app/index.d.ts.map +1 -1
- package/lib/subcommands/app/index.js +1 -3
- package/lib/subcommands/app/index.js.map +1 -1
- package/lib/subcommands/app/models.d.ts +0 -11
- package/lib/subcommands/app/models.d.ts.map +1 -1
- package/lib/subcommands/app/models.js +2 -58
- package/lib/subcommands/app/models.js.map +1 -1
- package/lib/subcommands/app/shadow.d.ts.map +1 -1
- package/lib/subcommands/app/shadow.js +6 -5
- package/lib/subcommands/app/shadow.js.map +1 -1
- package/lib/subcommands/app/version.d.ts.map +1 -1
- package/lib/subcommands/app/version.js +2 -4
- package/lib/subcommands/app/version.js.map +1 -1
- package/lib/subcommands/config.d.ts +2 -0
- package/lib/subcommands/config.d.ts.map +1 -0
- package/lib/subcommands/config.js +39 -0
- package/lib/subcommands/config.js.map +1 -0
- package/lib/subcommands/device/clean.d.ts +1 -1
- package/lib/subcommands/device/clean.d.ts.map +1 -1
- package/lib/subcommands/device/clean.js +25 -15
- package/lib/subcommands/device/clean.js.map +1 -1
- package/lib/subcommands/device/get-info.d.ts +2 -0
- package/lib/subcommands/device/get-info.d.ts.map +1 -0
- package/lib/subcommands/device/get-info.js +36 -0
- package/lib/subcommands/device/get-info.js.map +1 -0
- package/lib/subcommands/device/index.d.ts.map +1 -1
- package/lib/subcommands/device/index.js +13 -2
- package/lib/subcommands/device/index.js.map +1 -1
- package/lib/subcommands/device/init.d.ts +5 -0
- package/lib/subcommands/device/init.d.ts.map +1 -0
- package/lib/subcommands/device/{device.js → init.js} +10 -49
- package/lib/subcommands/device/init.js.map +1 -0
- package/lib/subcommands/device/migrate.d.ts +2 -0
- package/lib/subcommands/device/migrate.d.ts.map +1 -0
- package/lib/subcommands/device/migrate.js +24 -0
- package/lib/subcommands/device/migrate.js.map +1 -0
- package/lib/subcommands/device/refresh.d.ts +2 -0
- package/lib/subcommands/device/refresh.d.ts.map +1 -0
- package/lib/subcommands/device/refresh.js +25 -0
- package/lib/subcommands/device/refresh.js.map +1 -0
- package/lib/subcommands/device/restart.d.ts +2 -0
- package/lib/subcommands/device/restart.d.ts.map +1 -0
- package/lib/subcommands/device/restart.js +14 -0
- package/lib/subcommands/device/restart.js.map +1 -0
- package/lib/subcommands/index.d.ts +1 -1
- package/lib/subcommands/index.d.ts.map +1 -1
- package/lib/subcommands/index.js +3 -1
- package/lib/subcommands/index.js.map +1 -1
- package/lib/subcommands/rabbitmq-connection.d.ts +1 -1
- package/lib/subcommands/rabbitmq-connection.d.ts.map +1 -1
- package/lib/util/aai-error.d.ts +12 -0
- package/lib/util/aai-error.d.ts.map +1 -0
- package/lib/util/aai-error.js +11 -0
- package/lib/util/aai-error.js.map +1 -0
- package/lib/util/aws-regions.d.ts +2 -0
- package/lib/util/aws-regions.d.ts.map +1 -0
- package/lib/util/{cloud-mode-ready.js → aws-regions.js} +2 -20
- package/lib/util/aws-regions.js.map +1 -0
- package/lib/util/check-for-updates.d.ts +3 -0
- package/lib/util/check-for-updates.d.ts.map +1 -0
- package/lib/util/check-for-updates.js +46 -0
- package/lib/util/check-for-updates.js.map +1 -0
- package/lib/util/clean-certs.d.ts.map +1 -1
- package/lib/util/clean-certs.js +5 -4
- package/lib/util/clean-certs.js.map +1 -1
- package/lib/util/directories.d.ts +4 -18
- package/lib/util/directories.d.ts.map +1 -1
- package/lib/util/directories.js +18 -32
- package/lib/util/directories.js.map +1 -1
- package/lib/util/file.d.ts +11 -0
- package/lib/util/file.d.ts.map +1 -0
- package/lib/util/file.js +127 -0
- package/lib/util/file.js.map +1 -0
- package/lib/util/file.test.d.ts +2 -0
- package/lib/util/file.test.d.ts.map +1 -0
- package/lib/util/file.test.js +87 -0
- package/lib/util/file.test.js.map +1 -0
- package/lib/util/get-device-id.d.ts.map +1 -1
- package/lib/util/get-device-id.js +7 -1
- package/lib/util/get-device-id.js.map +1 -1
- package/lib/util/http-client.js +3 -3
- package/lib/util/http-client.js.map +1 -1
- package/package.json +22 -19
- package/readme.md +15 -35
- package/src/application-control/config.ts +10 -13
- package/src/application-control/environment-variables.test.ts +28 -7
- package/src/application-control/environment-variables.ts +13 -40
- package/src/application-control/index.ts +3 -16
- package/src/application-control/install.ts +16 -10
- package/src/application-control/models.ts +40 -98
- package/src/application-control/status.ts +9 -7
- package/src/application-control/utils.ts +1 -29
- package/src/cloud-connection/bootstrap-provision.ts +7 -7
- package/src/cloud-connection/device-agent-cloud-connection.ts +647 -509
- package/src/cloud-connection/device-agent.ts +16 -7
- package/src/cloud-connection/live-updates-handler.test.ts +137 -64
- package/src/cloud-connection/live-updates-handler.ts +103 -234
- package/src/cloud-connection/passthrough-handler.ts +134 -75
- package/src/cloud-connection/shadow-handler.test.ts +45 -57
- package/src/cloud-connection/shadow-handler.ts +114 -56
- package/src/cloud-connection/shadow.ts +4 -1
- package/src/cloud-connection/transaction-manager.test.ts +127 -3
- package/src/cloud-connection/transaction-manager.ts +68 -39
- package/src/device-control/device-control.ts +179 -72
- package/src/docker/docker-compose.ts +61 -0
- package/src/index.ts +2 -6
- package/src/infrastructure/agent-config.test.ts +9 -2
- package/src/infrastructure/agent-config.ts +45 -46
- package/src/infrastructure/config-check-utility.test.ts +154 -0
- package/src/infrastructure/config-check-utility.ts +77 -0
- package/src/infrastructure/device-certificate.test.ts +40 -0
- package/src/infrastructure/device-certificate.ts +58 -0
- package/src/infrastructure/legacy-migration/legacy-file.test.ts +88 -0
- package/src/infrastructure/legacy-migration/legacy-files.ts +101 -0
- package/src/infrastructure/legacy-migration/legacy-migration.test.ts +396 -0
- package/src/infrastructure/legacy-migration/legacy-migration.ts +229 -0
- package/src/infrastructure/require-files-present-ready.test.ts +53 -0
- package/src/infrastructure/required-config-checks.ts +33 -0
- package/src/infrastructure/tokens-and-device-cfg.ts +12 -10
- package/src/local-connection/rabbitmq-connection.ts +28 -23
- package/src/secure-tunneling/secure-tunneling.test.ts +37 -39
- package/src/secure-tunneling/secure-tunneling.ts +74 -69
- package/src/subcommands/app/analytics.ts +2 -4
- package/src/subcommands/app/env-vars.ts +72 -9
- package/src/subcommands/app/index.ts +3 -11
- package/src/subcommands/app/models.ts +5 -81
- package/src/subcommands/app/shadow.ts +6 -5
- package/src/subcommands/app/version.ts +3 -4
- package/src/subcommands/config.ts +42 -0
- package/src/subcommands/device/clean.ts +32 -18
- package/src/subcommands/device/get-info.ts +49 -0
- package/src/subcommands/device/index.ts +13 -2
- package/src/subcommands/device/{device.ts → init.ts} +11 -69
- package/src/subcommands/device/migrate.ts +20 -0
- package/src/subcommands/device/refresh.ts +23 -0
- package/src/subcommands/device/restart.ts +11 -0
- package/src/subcommands/index.ts +3 -1
- package/src/util/aai-error.ts +20 -0
- package/src/util/{cloud-mode-ready.ts → aws-regions.ts} +0 -24
- package/src/util/check-for-updates.ts +53 -0
- package/src/util/clean-certs.ts +8 -4
- package/src/util/directories.ts +23 -67
- package/src/util/file.test.ts +90 -0
- package/src/util/file.ts +156 -0
- package/src/util/get-device-id.ts +7 -7
- package/src/util/http-client.ts +2 -2
- package/lib/docker/docker-compose-cmd.d.ts +0 -5
- package/lib/docker/docker-compose-cmd.d.ts.map +0 -1
- package/lib/docker/docker-compose-cmd.js +0 -16
- package/lib/docker/docker-compose-cmd.js.map +0 -1
- package/lib/subcommands/device/device.d.ts +0 -7
- package/lib/subcommands/device/device.d.ts.map +0 -1
- package/lib/subcommands/device/device.js.map +0 -1
- package/lib/util/cloud-mode-ready.d.ts +0 -3
- package/lib/util/cloud-mode-ready.d.ts.map +0 -1
- package/lib/util/cloud-mode-ready.js.map +0 -1
- package/lib/util/download-file.d.ts +0 -6
- package/lib/util/download-file.d.ts.map +0 -1
- package/lib/util/download-file.js +0 -25
- package/lib/util/download-file.js.map +0 -1
- package/lib/util/fetch-with-timeout.d.ts +0 -4
- package/lib/util/fetch-with-timeout.d.ts.map +0 -1
- package/lib/util/fetch-with-timeout.js +0 -30
- package/lib/util/fetch-with-timeout.js.map +0 -1
- package/lib/util/parsing.d.ts +0 -2
- package/lib/util/parsing.d.ts.map +0 -1
- package/lib/util/parsing.js +0 -17
- package/lib/util/parsing.js.map +0 -1
- package/lib/util/safe-rimraf.d.ts +0 -2
- package/lib/util/safe-rimraf.d.ts.map +0 -1
- package/lib/util/safe-rimraf.js +0 -16
- package/lib/util/safe-rimraf.js.map +0 -1
- package/src/docker/docker-compose-cmd.ts +0 -15
- package/src/util/download-file.ts +0 -25
- package/src/util/fetch-with-timeout.ts +0 -35
- package/src/util/parsing.ts +0 -11
- package/src/util/safe-rimraf.ts +0 -14
|
@@ -1,73 +1,95 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
const live_updates_handler_1 = require("./live-updates-handler");
|
|
4
|
-
const
|
|
5
|
-
|
|
4
|
+
const device_agent_schemas_1 = require("@alwaysai/device-agent-schemas");
|
|
5
|
+
const crypto_1 = require("crypto");
|
|
6
|
+
const stream_1 = require("stream");
|
|
7
|
+
jest.mock('../util/sleep');
|
|
8
|
+
jest.mock('./messages');
|
|
9
|
+
jest.useFakeTimers();
|
|
6
10
|
// https://github.com/facebook/react-native/issues/35701
|
|
7
11
|
Object.defineProperty(global, 'performance', {
|
|
8
12
|
writable: true
|
|
9
13
|
});
|
|
10
|
-
const
|
|
11
|
-
|
|
12
|
-
appState: true
|
|
13
|
-
};
|
|
14
|
-
const testFalseToggles = {
|
|
15
|
-
deviceStats: false,
|
|
16
|
-
appState: false
|
|
17
|
-
};
|
|
18
|
-
const mockClient = jest.fn();
|
|
19
|
-
const clientId = 'test-client';
|
|
20
|
-
const emptyTxId = '';
|
|
21
|
-
// NOTE: this was the way I found to mock private class functions
|
|
22
|
-
const mockStartPublishingLiveUpdates = jest.spyOn(live_updates_handler_1.LiveUpdatesHandler.prototype, 'startPublishingLiveUpdates');
|
|
23
|
-
mockStartPublishingLiveUpdates.mockResolvedValue(null);
|
|
14
|
+
const mockTransactionId = '9b1deb4d-3b7d-4bad-9bdd-2b0d7b3dcb6d';
|
|
15
|
+
const mockProjectId = '12345678-abcd-abcd-abcd-0123456789ab';
|
|
24
16
|
describe('Test Live Updates Handler', () => {
|
|
25
17
|
let liveUpdatesHandler;
|
|
26
|
-
let publisher;
|
|
27
18
|
beforeEach(() => {
|
|
28
|
-
|
|
29
|
-
liveUpdatesHandler = new live_updates_handler_1.LiveUpdatesHandler(publisher, clientId);
|
|
19
|
+
liveUpdatesHandler = new live_updates_handler_1.LiveUpdatesHandler();
|
|
30
20
|
jest.clearAllMocks();
|
|
31
21
|
});
|
|
32
|
-
test('
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
22
|
+
test('enable interval', async () => {
|
|
23
|
+
const mockPublishingFn = jest.fn();
|
|
24
|
+
const mockClearInterval = jest.spyOn(global, 'clearInterval');
|
|
25
|
+
const mockSetInterval = jest.spyOn(global, 'setInterval');
|
|
26
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, mockPublishingFn);
|
|
27
|
+
jest.advanceTimersByTime(live_updates_handler_1.DEFAULT_INTERVALS_MS);
|
|
28
|
+
expect(mockPublishingFn).toBeCalledTimes(2);
|
|
29
|
+
expect(mockClearInterval).toBeCalledWith(undefined);
|
|
30
|
+
expect(mockSetInterval).toBeCalledTimes(1);
|
|
31
|
+
});
|
|
32
|
+
test('subsequent enable resets interval with no duplicates', async () => {
|
|
33
|
+
const mockPublishingFn = jest.fn();
|
|
34
|
+
const mockSetInterval = jest.spyOn(global, 'setInterval');
|
|
35
|
+
const mockClearInterval = jest.spyOn(global, 'clearInterval');
|
|
36
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, mockPublishingFn, mockTransactionId);
|
|
37
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, mockPublishingFn, mockTransactionId);
|
|
38
|
+
jest.advanceTimersByTime(live_updates_handler_1.DEFAULT_INTERVALS_MS);
|
|
39
|
+
expect(mockClearInterval.mock.calls[0][0]).toEqual(undefined);
|
|
40
|
+
expect(mockClearInterval.mock.calls[1][0]).toEqual(mockSetInterval.mock.results[0].value);
|
|
41
|
+
expect(mockSetInterval).toBeCalledTimes(2);
|
|
42
|
+
expect(mockPublishingFn).toBeCalledTimes(3);
|
|
43
43
|
});
|
|
44
44
|
test('test disable live updates', async () => {
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
expect(
|
|
50
|
-
expect(liveUpdatesHandler.getAppStateLiveUpdates()).toBe(true);
|
|
51
|
-
expect(liveUpdatesHandler.getAppLogsLiveUpdates()).toBe(false);
|
|
52
|
-
// Check to see that attributes are properly set to false when disabled
|
|
53
|
-
jest.clearAllMocks();
|
|
54
|
-
void liveUpdatesHandler.handleToggles(testFalseToggles, emptyTxId);
|
|
55
|
-
expect(mockStartPublishingLiveUpdates).toBeCalledTimes(0);
|
|
56
|
-
expect(jest.mocked(setTimeout)).toHaveBeenCalledTimes(1);
|
|
57
|
-
expect(liveUpdatesHandler.getDeviceStatsLiveUpdates()).toBe(false);
|
|
58
|
-
expect(liveUpdatesHandler.getAppStateLiveUpdates()).toBe(false);
|
|
59
|
-
expect(liveUpdatesHandler.getAppLogsLiveUpdates()).toBe(false);
|
|
45
|
+
const mockSetInterval = jest.spyOn(global, 'setInterval');
|
|
46
|
+
const mockClearInterval = jest.spyOn(global, 'clearInterval');
|
|
47
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, async () => undefined, mockTransactionId);
|
|
48
|
+
liveUpdatesHandler.disable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, mockTransactionId);
|
|
49
|
+
expect(mockClearInterval).toHaveBeenLastCalledWith(mockSetInterval.mock.results[0].value);
|
|
60
50
|
});
|
|
61
|
-
test('timeout turns off live updates', async () => {
|
|
62
|
-
jest.
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
51
|
+
test('kill all timeout turns off all live updates', async () => {
|
|
52
|
+
const mockPublishingFn = jest.fn();
|
|
53
|
+
const mockSetInterval = jest.spyOn(global, 'setInterval');
|
|
54
|
+
const mockClearInterval = jest.spyOn(global, 'clearInterval');
|
|
55
|
+
jest.spyOn(global, 'setTimeout');
|
|
56
|
+
const numLivingIntervals = 5;
|
|
57
|
+
for (let i = 0; i < numLivingIntervals; i++) {
|
|
58
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.status_response, mockPublishingFn, (0, crypto_1.randomUUID)());
|
|
59
|
+
}
|
|
60
|
+
mockSetInterval.mock.results.forEach((result) => clearInterval(result.value));
|
|
67
61
|
jest.runAllTimers();
|
|
68
|
-
expect(
|
|
69
|
-
expect(
|
|
70
|
-
|
|
62
|
+
expect(mockPublishingFn).toBeCalledTimes(numLivingIntervals);
|
|
63
|
+
expect(mockSetInterval).toBeCalledTimes(numLivingIntervals);
|
|
64
|
+
for (let i = 0; i < numLivingIntervals; i++) {
|
|
65
|
+
expect(mockClearInterval).toHaveBeenCalledWith(mockSetInterval.mock.results[i].value);
|
|
66
|
+
}
|
|
67
|
+
});
|
|
68
|
+
test('startStream streams data to publish', async () => {
|
|
69
|
+
const mockStream = new stream_1.PassThrough();
|
|
70
|
+
const mockPublishingFn = jest
|
|
71
|
+
.fn()
|
|
72
|
+
.mockImplementation((logChunk) => undefined);
|
|
73
|
+
await liveUpdatesHandler.startStream(mockProjectId, async () => mockStream, mockPublishingFn);
|
|
74
|
+
const data = 'Data for stream';
|
|
75
|
+
mockStream.emit('data', data);
|
|
76
|
+
expect(mockPublishingFn).toBeCalledWith(data);
|
|
77
|
+
mockStream.end();
|
|
78
|
+
mockStream.destroy();
|
|
79
|
+
});
|
|
80
|
+
test('failure does not disable interval', async () => {
|
|
81
|
+
const mockPublishingFn = jest
|
|
82
|
+
.fn()
|
|
83
|
+
.mockImplementationOnce(() => undefined)
|
|
84
|
+
.mockImplementation(() => {
|
|
85
|
+
throw new Error('Manual throw for unit test');
|
|
86
|
+
});
|
|
87
|
+
jest.spyOn(global, 'setInterval');
|
|
88
|
+
const mockClearInterval = jest.spyOn(global, 'clearInterval');
|
|
89
|
+
await liveUpdatesHandler.enable(device_agent_schemas_1.keyMirrors.toClientMessageType.device_stats, mockPublishingFn, mockTransactionId);
|
|
90
|
+
jest.advanceTimersByTime(live_updates_handler_1.DEFAULT_INTERVALS_MS);
|
|
91
|
+
expect(mockClearInterval).toBeCalledTimes(1);
|
|
92
|
+
expect(mockPublishingFn).toBeCalledTimes(2);
|
|
71
93
|
});
|
|
72
94
|
});
|
|
73
95
|
//# sourceMappingURL=live-updates-handler.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"live-updates-handler.test.js","sourceRoot":"","sources":["../../src/cloud-connection/live-updates-handler.test.ts"],"names":[],"mappings":";;AAAA,
|
|
1
|
+
{"version":3,"file":"live-updates-handler.test.js","sourceRoot":"","sources":["../../src/cloud-connection/live-updates-handler.test.ts"],"names":[],"mappings":";;AAAA,iEAGgC;AAChC,yEAA4D;AAC5D,mCAAoC;AACpC,mCAAqC;AAErC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAC3B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACxB,IAAI,CAAC,aAAa,EAAE,CAAC;AAErB,wDAAwD;AACxD,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,aAAa,EAAE;IAC3C,QAAQ,EAAE,IAAI;CACf,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,sCAAsC,CAAC;AACjE,MAAM,aAAa,GAAG,sCAAsC,CAAC;AAE7D,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,IAAI,kBAAsC,CAAC;IAE3C,UAAU,CAAC,GAAG,EAAE;QACd,kBAAkB,GAAG,IAAI,yCAAkB,EAAE,CAAC;QAC9C,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAC9D,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAE1D,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,gBAAgB,CACjB,CAAC;QAEF,IAAI,CAAC,mBAAmB,CAAC,2CAAoB,CAAC,CAAC;QAE/C,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,iBAAiB,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAC1D,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAE9D,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,gBAAgB,EAChB,iBAAiB,CAClB,CAAC;QACF,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,gBAAgB,EAChB,iBAAiB,CAClB,CAAC;QAEF,IAAI,CAAC,mBAAmB,CAAC,2CAAoB,CAAC,CAAC;QAE/C,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC9D,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAChD,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CACtC,CAAC;QACF,MAAM,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAC1D,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAE9D,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,KAAK,IAAI,EAAE,CAAC,SAAS,EACrB,iBAAiB,CAClB,CAAC;QACF,kBAAkB,CAAC,OAAO,CACxB,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,iBAAiB,CAClB,CAAC;QAEF,MAAM,CAAC,iBAAiB,CAAC,CAAC,wBAAwB,CAChD,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CACtC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAC1D,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAC9D,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAEjC,MAAM,kBAAkB,GAAG,CAAC,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,kBAAkB,EAAE,CAAC,EAAE,EAAE;YAC3C,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,eAAe,EAC9C,gBAAgB,EAChB,IAAA,mBAAU,GAAE,CACb,CAAC;SACH;QAED,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAC9C,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAC5B,CAAC;QACF,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAC7D,MAAM,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAC5D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,kBAAkB,EAAE,CAAC,EAAE,EAAE;YAC3C,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CACtC,CAAC;SACH;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,UAAU,GAAG,IAAI,oBAAW,EAAE,CAAC;QACrC,MAAM,gBAAgB,GAAG,IAAI;aAC1B,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,QAAgB,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,kBAAkB,CAAC,WAAW,CAClC,aAAa,EACb,KAAK,IAAI,EAAE,CAAC,UAAU,EACtB,gBAAgB,CACjB,CAAC;QAEF,MAAM,IAAI,GAAG,iBAAiB,CAAC;QAC/B,UAAU,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE9B,MAAM,CAAC,gBAAgB,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAE9C,UAAU,CAAC,GAAG,EAAE,CAAC;QACjB,UAAU,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,gBAAgB,GAAG,IAAI;aAC1B,EAAE,EAAE;aACJ,sBAAsB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACvC,kBAAkB,CAAC,GAAG,EAAE;YACvB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QACL,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAClC,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAE9D,MAAM,kBAAkB,CAAC,MAAM,CAC7B,iCAAU,CAAC,mBAAmB,CAAC,YAAY,EAC3C,gBAAgB,EAChB,iBAAiB,CAClB,CAAC;QAEF,IAAI,CAAC,mBAAmB,CAAC,2CAAoB,CAAC,CAAC;QAE/C,MAAM,CAAC,iBAAiB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,11 +1,16 @@
|
|
|
1
|
+
import * as amqp from 'amqplib';
|
|
1
2
|
import { Publisher } from './publisher';
|
|
3
|
+
import { ShadowHandler } from './shadow-handler';
|
|
2
4
|
export declare class PassthroughHandler {
|
|
3
5
|
publisher: Publisher;
|
|
4
|
-
|
|
5
|
-
|
|
6
|
+
shadowHandler: ShadowHandler;
|
|
7
|
+
connection: amqp.Connection | undefined;
|
|
8
|
+
channel: amqp.Channel;
|
|
6
9
|
packetQueue: any;
|
|
7
|
-
constructor(publisher: Publisher);
|
|
10
|
+
constructor(publisher: Publisher, shadowHandler: ShadowHandler);
|
|
11
|
+
runChannel: () => Promise<void>;
|
|
12
|
+
establishLocalConnection(): Promise<void>;
|
|
13
|
+
private publishPassthroughStatusUpdate;
|
|
8
14
|
setup(): Promise<void>;
|
|
9
15
|
}
|
|
10
|
-
export declare function runChannel(passthroughHandler: PassthroughHandler): Promise<void>;
|
|
11
16
|
//# sourceMappingURL=passthrough-handler.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"passthrough-handler.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"passthrough-handler.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAahC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAMjD,qBAAa,kBAAkB;IACtB,SAAS,EAAE,SAAS,CAAC;IACrB,aAAa,EAAE,aAAa,CAAC;IAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IACxC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC;IACtB,WAAW,MAAC;gBAEP,SAAS,EAAE,SAAS,EAAE,aAAa,EAAE,aAAa;IAK9D,UAAU,sBAoER;IAEI,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC;YA8CjC,8BAA8B;IAetC,KAAK;CA0BZ"}
|
|
@@ -1,8 +1,9 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.
|
|
4
|
-
|
|
5
|
-
const amqp = require(
|
|
3
|
+
exports.PassthroughHandler = void 0;
|
|
4
|
+
const util_1 = require("alwaysai/lib/util");
|
|
5
|
+
const amqp = require("amqplib");
|
|
6
|
+
const environment_1 = require("../environment");
|
|
6
7
|
const constants_1 = require("../local-connection/constants");
|
|
7
8
|
const rabbitmq_connection_1 = require("../local-connection/rabbitmq-connection");
|
|
8
9
|
const logger_1 = require("../util/logger");
|
|
@@ -11,19 +12,80 @@ const messageQueue = [];
|
|
|
11
12
|
const ackQueue = [];
|
|
12
13
|
const MAX_LOCAL_CONNECTION_ATTEMPTS = 10;
|
|
13
14
|
class PassthroughHandler {
|
|
14
|
-
constructor(publisher) {
|
|
15
|
+
constructor(publisher, shadowHandler) {
|
|
16
|
+
this.runChannel = async () => {
|
|
17
|
+
logger_1.logger.debug('Beginning to consume packets');
|
|
18
|
+
await this.channel.consume(this.packetQueue, (msg) => {
|
|
19
|
+
// NOTE: this needs to be an arrow function and then the whole contents of processPublish are below
|
|
20
|
+
if ((msg === null || msg === void 0 ? void 0 : msg.content) !== undefined) {
|
|
21
|
+
const packet = JSON.parse(msg.content.toString());
|
|
22
|
+
messageQueue.push({ packet, msg });
|
|
23
|
+
while (messageQueue.length > 0) {
|
|
24
|
+
const entry = messageQueue.shift();
|
|
25
|
+
const { packet, msg } = entry;
|
|
26
|
+
try {
|
|
27
|
+
const parsedPacket = JSON.parse(packet);
|
|
28
|
+
if (parsedPacket === null || parsedPacket === void 0 ? void 0 : parsedPacket['action']) {
|
|
29
|
+
switch (parsedPacket['action']) {
|
|
30
|
+
case 'analytics':
|
|
31
|
+
ackQueue.push(msg);
|
|
32
|
+
// FIXME: put real topic here
|
|
33
|
+
this.publisher.publishToCloudWithAck(packet, (errOrResp) => {
|
|
34
|
+
while (ackQueue.length > 0) {
|
|
35
|
+
const msg = ackQueue.shift();
|
|
36
|
+
if (errOrResp === true) {
|
|
37
|
+
this.channel.ack(msg); // acknowledge, allow queue to discard
|
|
38
|
+
}
|
|
39
|
+
else if (errOrResp === false) {
|
|
40
|
+
this.channel.reject(msg, true); // reject and requeue
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
});
|
|
44
|
+
break;
|
|
45
|
+
case 'heartbeat':
|
|
46
|
+
this.channel.ack(msg);
|
|
47
|
+
logger_1.logger.silly(`Heartbeat package received & acknowledged: ${packet}`);
|
|
48
|
+
break;
|
|
49
|
+
default:
|
|
50
|
+
this.channel.ack(msg);
|
|
51
|
+
logger_1.logger.debug(`Unknown 'action' package received & acknowledged: ${packet}`);
|
|
52
|
+
break;
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
else {
|
|
56
|
+
this.channel.ack(msg);
|
|
57
|
+
logger_1.logger.debug(`Received & acknowledged a RabbitMQ Package of unknown structure: ${parsedPacket}`);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
catch (e) {
|
|
61
|
+
logger_1.logger.error(`There was a problem parsing RabbitMQ packet!\n${(0, util_1.stringifyError)(e)}`);
|
|
62
|
+
this.channel.ack(msg);
|
|
63
|
+
logger_1.logger.debug('Problematic packet was acknowledged');
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
}, {
|
|
68
|
+
noAck: false // When true, RabbitMQ deletes message as soon as it is consumed
|
|
69
|
+
});
|
|
70
|
+
};
|
|
15
71
|
this.publisher = publisher;
|
|
72
|
+
this.shadowHandler = shadowHandler;
|
|
16
73
|
}
|
|
17
|
-
async
|
|
18
|
-
await (0, rabbitmq_connection_1.setupRabbitMQContainer)();
|
|
74
|
+
async establishLocalConnection() {
|
|
19
75
|
let connectAttempts = 0;
|
|
20
76
|
let connected = false;
|
|
21
|
-
logger_1.logger.debug(`
|
|
77
|
+
logger_1.logger.debug(`Establishing local connection...`);
|
|
22
78
|
while (connectAttempts <= MAX_LOCAL_CONNECTION_ATTEMPTS &&
|
|
23
79
|
this.connection === undefined) {
|
|
24
80
|
try {
|
|
25
81
|
this.connection = await amqp.connect(`amqp://${constants_1.LOCAL_CONNECTION_HOST}:${constants_1.LOCAL_CONNECTION_PORT}`);
|
|
26
82
|
this.channel = await this.connection.createChannel();
|
|
83
|
+
this.connection.on('error', async () => {
|
|
84
|
+
logger_1.logger.error(`Local connection failed. Attempting to reconnect...`);
|
|
85
|
+
await (0, rabbitmq_connection_1.stopRabbitMQContainer)();
|
|
86
|
+
this.connection = undefined;
|
|
87
|
+
await this.setup();
|
|
88
|
+
});
|
|
27
89
|
connected = true;
|
|
28
90
|
}
|
|
29
91
|
catch (e) {
|
|
@@ -34,74 +96,45 @@ class PassthroughHandler {
|
|
|
34
96
|
}
|
|
35
97
|
}
|
|
36
98
|
if (connected === true) {
|
|
37
|
-
this.channel.prefetch(1); // This ensures we only get one packet at a time! This appears to have prevented throttling
|
|
99
|
+
await this.channel.prefetch(1); // This ensures we only get one packet at a time! This appears to have prevented throttling
|
|
38
100
|
this.packetQueue = `${constants_1.LOCAL_CONNECTION_ROUTING_KEY}`;
|
|
39
101
|
await this.channel.assertQueue(this.packetQueue, {
|
|
40
102
|
durable: true
|
|
41
103
|
});
|
|
104
|
+
logger_1.logger.info(`Local connection established.`);
|
|
42
105
|
}
|
|
43
106
|
else {
|
|
44
107
|
throw new Error('Unable to establish connection to alwaysAI Local Connection, please try restarting Device Agent.');
|
|
45
108
|
}
|
|
46
109
|
}
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
}
|
|
66
|
-
else if (errOrResp === false) {
|
|
67
|
-
passthroughHandler.channel.reject(msg, true); // reject and requeue
|
|
68
|
-
}
|
|
69
|
-
}
|
|
70
|
-
});
|
|
71
|
-
break;
|
|
72
|
-
case 'heartbeat':
|
|
73
|
-
passthroughHandler.channel.ack(msg);
|
|
74
|
-
logger_1.logger.debug(`Heartbeat package received & acknowledged: ${packet}`);
|
|
75
|
-
break;
|
|
76
|
-
default:
|
|
77
|
-
passthroughHandler.channel.ack(msg);
|
|
78
|
-
logger_1.logger.debug(`Unknown 'action' package received & acknowledged: ${packet}`);
|
|
79
|
-
break;
|
|
80
|
-
}
|
|
110
|
+
async publishPassthroughStatusUpdate(status, message) {
|
|
111
|
+
const deviceAgentPassthroughStatus = {
|
|
112
|
+
passthrough: {
|
|
113
|
+
status: status,
|
|
114
|
+
message: message !== null && message !== void 0 ? message : ''
|
|
115
|
+
}
|
|
116
|
+
};
|
|
117
|
+
await this.shadowHandler.updateDeviceAgentStatusShadow(deviceAgentPassthroughStatus);
|
|
118
|
+
}
|
|
119
|
+
async setup() {
|
|
120
|
+
if (environment_1.ALWAYSAI_ANALYTICS_PASSTHROUGH === true) {
|
|
121
|
+
logger_1.logger.debug(`Setting up alwaysAI Local Connection on host: ${constants_1.LOCAL_CONNECTION_HOST} and channel key: ${constants_1.LOCAL_CONNECTION_ROUTING_KEY}`);
|
|
122
|
+
await this.publishPassthroughStatusUpdate('starting');
|
|
123
|
+
await (0, rabbitmq_connection_1.setupRabbitMQContainer)();
|
|
124
|
+
try {
|
|
125
|
+
await this.establishLocalConnection();
|
|
126
|
+
await this.runChannel();
|
|
127
|
+
await this.publishPassthroughStatusUpdate('running', `Passthrough running on host: ${constants_1.LOCAL_CONNECTION_HOST} and channel key: ${constants_1.LOCAL_CONNECTION_ROUTING_KEY}`);
|
|
81
128
|
}
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
129
|
+
catch (e) {
|
|
130
|
+
logger_1.logger.error(`There was a problem maintaining RabbitMQ connection!\n${(0, util_1.stringifyError)(e)}`);
|
|
131
|
+
await this.publishPassthroughStatusUpdate('error', (0, util_1.stringifyError)(e));
|
|
85
132
|
}
|
|
86
133
|
}
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
passthroughHandler.channel.ack(msg);
|
|
90
|
-
logger_1.logger.debug(`Problematic packet was acknowledged`);
|
|
134
|
+
else {
|
|
135
|
+
await this.publishPassthroughStatusUpdate('disabled');
|
|
91
136
|
}
|
|
92
137
|
}
|
|
93
138
|
}
|
|
94
|
-
|
|
95
|
-
logger_1.logger.debug('Beginning to consume packets');
|
|
96
|
-
passthroughHandler.channel.consume(passthroughHandler.packetQueue, function (msg) {
|
|
97
|
-
if (msg.content !== undefined) {
|
|
98
|
-
const packet = JSON.parse(msg.content.toString());
|
|
99
|
-
messageQueue.push({ packet, msg });
|
|
100
|
-
processPublish(passthroughHandler);
|
|
101
|
-
}
|
|
102
|
-
}, {
|
|
103
|
-
noAck: false // When true, RabbitMQ deletes message as soon as it is consumed
|
|
104
|
-
});
|
|
105
|
-
}
|
|
106
|
-
exports.runChannel = runChannel;
|
|
139
|
+
exports.PassthroughHandler = PassthroughHandler;
|
|
107
140
|
//# sourceMappingURL=passthrough-handler.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"passthrough-handler.js","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":";;;
|
|
1
|
+
{"version":3,"file":"passthrough-handler.js","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":";;;AAIA,4CAAmD;AACnD,gCAAgC;AAChC,gDAAgE;AAChE,6DAIuC;AACvC,iFAGiD;AACjD,2CAAwC;AACxC,yCAAkC;AAIlC,MAAM,YAAY,GAAU,EAAE,CAAC;AAC/B,MAAM,QAAQ,GAAU,EAAE,CAAC;AAC3B,MAAM,6BAA6B,GAAG,EAAE,CAAC;AAEzC,MAAa,kBAAkB;IAO7B,YAAY,SAAoB,EAAE,aAA4B;QAK9D,eAAU,GAAG,KAAK,IAAI,EAAE;YACtB,eAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;YAC7C,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CACxB,IAAI,CAAC,WAAW,EAChB,CAAC,GAAG,EAAE,EAAE;gBACN,mGAAmG;gBACnG,IAAI,CAAA,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,OAAO,MAAK,SAAS,EAAE;oBAC9B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAClD,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;oBACnC,OAAO,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;wBAC9B,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC;wBACnC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;wBAC9B,IAAI;4BACF,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;4BACxC,IAAI,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAG,QAAQ,CAAC,EAAE;gCAC5B,QAAQ,YAAY,CAAC,QAAQ,CAAC,EAAE;oCAC9B,KAAK,WAAW;wCACd,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wCACnB,6BAA6B;wCAC7B,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAClC,MAAM,EACN,CAAC,SAAS,EAAE,EAAE;4CACZ,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;gDAC1B,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gDAC7B,IAAI,SAAS,KAAK,IAAI,EAAE;oDACtB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAsC;iDAC9D;qDAAM,IAAI,SAAS,KAAK,KAAK,EAAE;oDAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,qBAAqB;iDACtD;6CACF;wCACH,CAAC,CACF,CAAC;wCACF,MAAM;oCACR,KAAK,WAAW;wCACd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wCACtB,eAAM,CAAC,KAAK,CACV,8CAA8C,MAAM,EAAE,CACvD,CAAC;wCACF,MAAM;oCACR;wCACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wCACtB,eAAM,CAAC,KAAK,CACV,qDAAqD,MAAM,EAAE,CAC9D,CAAC;wCACF,MAAM;iCACT;6BACF;iCAAM;gCACL,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gCACtB,eAAM,CAAC,KAAK,CACV,oEAAoE,YAAY,EAAE,CACnF,CAAC;6BACH;yBACF;wBAAC,OAAO,CAAC,EAAE;4BACV,eAAM,CAAC,KAAK,CACV,iDAAiD,IAAA,qBAAc,EAC7D,CAAC,CACF,EAAE,CACJ,CAAC;4BACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;4BACtB,eAAM,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC;yBACrD;qBACF;iBACF;YACH,CAAC,EACD;gBACE,KAAK,EAAE,KAAK,CAAC,gEAAgE;aAC9E,CACF,CAAC;QACJ,CAAC,CAAC;QAxEA,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAwED,KAAK,CAAC,wBAAwB;QAC5B,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,eAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACjD,OACE,eAAe,IAAI,6BAA6B;YAChD,IAAI,CAAC,UAAU,KAAK,SAAS,EAC7B;YACA,IAAI;gBACF,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAClC,UAAU,iCAAqB,IAAI,iCAAqB,EAAE,CAC3D,CAAC;gBACF,IAAI,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;gBACrD,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE;oBACrC,eAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;oBACpE,MAAM,IAAA,2CAAqB,GAAE,CAAC;oBAC9B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;oBAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC,CAAC,CAAC;gBAEH,SAAS,GAAG,IAAI,CAAC;aAClB;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,qBAAqB,GAAG,IAAI,GAAG,IAAI,GAAG,eAAe,CAAC;gBAC5D,eAAM,CAAC,KAAK,CACV,8DAA8D,eAAe,OAAO,6BAA6B,KAC/G,qBAAqB,GAAG,IAC1B,6BAA6B,CAC9B,CAAC;gBACF,MAAM,IAAA,eAAK,EAAC,qBAAqB,CAAC,CAAC;gBACnC,eAAe,IAAI,CAAC,CAAC;aACtB;SACF;QACD,IAAI,SAAS,KAAK,IAAI,EAAE;YACtB,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,2FAA2F;YAC3H,IAAI,CAAC,WAAW,GAAG,GAAG,wCAA4B,EAAE,CAAC;YACrD,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC/C,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,eAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;SAC9C;aAAM;YACL,MAAM,IAAI,KAAK,CACb,kGAAkG,CACnG,CAAC;SACH;IACH,CAAC;IAEO,KAAK,CAAC,8BAA8B,CAC1C,MAA8B,EAC9B,OAAgB;QAEhB,MAAM,4BAA4B,GAAkC;YAClE,WAAW,EAAE;gBACX,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE;aACvB;SACF,CAAC;QACF,MAAM,IAAI,CAAC,aAAa,CAAC,6BAA6B,CACpD,4BAA4B,CAC7B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,4CAA8B,KAAK,IAAI,EAAE;YAC3C,eAAM,CAAC,KAAK,CACV,iDAAiD,iCAAqB,qBAAqB,wCAA4B,EAAE,CAC1H,CAAC;YACF,MAAM,IAAI,CAAC,8BAA8B,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,IAAA,4CAAsB,GAAE,CAAC;YAC/B,IAAI;gBACF,MAAM,IAAI,CAAC,wBAAwB,EAAE,CAAC;gBACtC,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;gBACxB,MAAM,IAAI,CAAC,8BAA8B,CACvC,SAAS,EACT,gCAAgC,iCAAqB,qBAAqB,wCAA4B,EAAE,CACzG,CAAC;aACH;YAAC,OAAO,CAAC,EAAE;gBACV,eAAM,CAAC,KAAK,CACV,yDAAyD,IAAA,qBAAc,EACrE,CAAC,CACF,EAAE,CACJ,CAAC;gBACF,MAAM,IAAI,CAAC,8BAA8B,CAAC,OAAO,EAAE,IAAA,qBAAc,EAAC,CAAC,CAAC,CAAC,CAAC;aACvE;SACF;aAAM;YACL,MAAM,IAAI,CAAC,8BAA8B,CAAC,UAAU,CAAC,CAAC;SACvD;IACH,CAAC;CACF;AAzKD,gDAyKC"}
|
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
import { AppConfig } from '@alwaysai/app-configuration-schemas';
|
|
2
|
-
import { EnvVars } from '
|
|
2
|
+
import { EnvVars, SecureTunnelShadowUpdate } from '@alwaysai/device-agent-schemas';
|
|
3
|
+
import { DeviceAgentStatusShadowUpdate } from '@alwaysai/device-agent-schemas/lib/shadow-schema';
|
|
3
4
|
import { Publisher } from './publisher';
|
|
4
5
|
import { AppConfigModels } from './shadow';
|
|
5
6
|
export declare type AppConfigUpdate = {
|
|
@@ -24,6 +25,7 @@ export declare class ShadowHandler {
|
|
|
24
25
|
};
|
|
25
26
|
constructor(clientId: string, publisher: Publisher);
|
|
26
27
|
private generateAppConfigUpdate;
|
|
28
|
+
private generateEnvVarsUpdate;
|
|
27
29
|
private processProjectShadowUpdates;
|
|
28
30
|
private processProjectShadowUpdate;
|
|
29
31
|
handleProjectShadow({ topic, payload, clientToken }: {
|
|
@@ -37,6 +39,8 @@ export declare class ShadowHandler {
|
|
|
37
39
|
projectId: string;
|
|
38
40
|
envVars: EnvVars;
|
|
39
41
|
}): Promise<void>;
|
|
42
|
+
updateSecureTunnelShadow(secureTunnelShadowUpdate: SecureTunnelShadowUpdate): Promise<void>;
|
|
43
|
+
updateDeviceAgentStatusShadow(deviceAgentStatusShadowUpdate: DeviceAgentStatusShadowUpdate): Promise<void>;
|
|
40
44
|
getProjectShadowUpdates(): void;
|
|
41
45
|
clearProjectShadow(projectId: string): void;
|
|
42
46
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"shadow-handler.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/shadow-handler.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EAEV,MAAM,qCAAqC,CAAC;AAC7C,OAAO,
|
|
1
|
+
{"version":3,"file":"shadow-handler.d.ts","sourceRoot":"","sources":["../../src/cloud-connection/shadow-handler.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,SAAS,EAEV,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAEL,OAAO,EAKP,wBAAwB,EAGzB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAEL,6BAA6B,EAE9B,MAAM,kDAAkD,CAAC;AAK1D,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,eAAe,EAAuB,MAAM,UAAU,CAAC;AAEhE,oBAAY,eAAe,GAAG;IAC5B,SAAS,EAAE,SAAS,CAAC;IACrB,aAAa,CAAC,EAAE,eAAe,CAAC;CACjC,CAAC;AAEF,oBAAY,YAAY,GAAG;IACzB,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAEF,oBAAY,YAAY,GAAG;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,eAAe,CAAC;IAC/B,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B,CAAC;AAEF,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,SAAS,CAAY;IACtB,mBAAmB,EAAE,MAAM,EAAE,CAAM;IAC1C,SAAgB,YAAY,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;gBACzC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS;YAoDpC,uBAAuB;YAgDvB,qBAAqB;YAkCrB,2BAA2B;YAkC3B,0BAA0B;IAwD3B,mBAAmB,CAAC,EAC/B,KAAK,EACL,OAAO,EACP,WAAW,EACZ,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,GAAG,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;KACrB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAwDd,sBAAsB;IAatB,mBAAmB,CAAC,SAAS,EAAE,MAAM;IAqBrC,oBAAoB,CAAC,EAChC,SAAS,EACT,OAAO,EACR,EAAE;QACD,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,OAAO,CAAC;KAClB;IAkBY,wBAAwB,CACnC,wBAAwB,EAAE,wBAAwB;IAavC,6BAA6B,CACxC,6BAA6B,EAAE,6BAA6B;IAavD,uBAAuB;IAOvB,kBAAkB,CAAC,SAAS,EAAE,MAAM;CAY5C"}
|