@alwaysai/device-agent 0.0.18 → 0.0.20
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/cloud-connection/passthrough-handler.d.ts.map +1 -1
- package/lib/cloud-connection/passthrough-handler.js +25 -7
- package/lib/cloud-connection/passthrough-handler.js.map +1 -1
- package/lib/local-connection/rabbitmq-connection.d.ts +4 -1
- package/lib/local-connection/rabbitmq-connection.d.ts.map +1 -1
- package/lib/local-connection/rabbitmq-connection.js +61 -39
- package/lib/local-connection/rabbitmq-connection.js.map +1 -1
- package/lib/subcommands/index.d.ts +1 -4
- 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 +2 -0
- package/lib/subcommands/rabbitmq-connection.d.ts.map +1 -0
- package/lib/subcommands/rabbitmq-connection.js +14 -0
- package/lib/subcommands/rabbitmq-connection.js.map +1 -0
- package/lib/util/directories.d.ts +3 -0
- package/lib/util/directories.d.ts.map +1 -1
- package/lib/util/directories.js +5 -3
- package/lib/util/directories.js.map +1 -1
- package/package.json +1 -1
- package/readme.md +35 -2
- package/src/cloud-connection/passthrough-handler.ts +27 -7
- package/src/local-connection/rabbitmq-connection.ts +82 -36
- package/src/subcommands/index.ts +3 -1
- package/src/subcommands/rabbitmq-connection.ts +11 -0
- package/src/util/directories.ts +12 -2
|
@@ -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,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAKxC,qBAAa,kBAAkB;IACtB,SAAS,EAAE,SAAS,CAAC;IACrB,UAAU,MAAC;IACX,OAAO,MAAC;IACR,WAAW,MAAC;gBAEP,SAAS,EAAE,SAAS;IAInB,KAAK;CA6BnB;AAqBD,wBAAsB,UAAU,CAAC,kBAAkB,EAAE,kBAAkB,iBAetE"}
|
|
@@ -5,6 +5,7 @@ exports.runChannel = exports.PassthroughHandler = void 0;
|
|
|
5
5
|
const amqp = require('amqplib');
|
|
6
6
|
const rabbitmq_connection_1 = require("../local-connection/rabbitmq-connection");
|
|
7
7
|
const logger_1 = require("../util/logger");
|
|
8
|
+
const sleep_1 = require("../util/sleep");
|
|
8
9
|
const messageQueue = [];
|
|
9
10
|
const ackQueue = [];
|
|
10
11
|
class PassthroughHandler {
|
|
@@ -13,13 +14,30 @@ class PassthroughHandler {
|
|
|
13
14
|
}
|
|
14
15
|
async setup() {
|
|
15
16
|
await (0, rabbitmq_connection_1.setupRabbitMQContainer)();
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
17
|
+
let connectAttempts = 0;
|
|
18
|
+
let connected = false;
|
|
19
|
+
while (connectAttempts <= 10 && this.connection === undefined) {
|
|
20
|
+
try {
|
|
21
|
+
this.connection = await amqp.connect('amqp://localhost');
|
|
22
|
+
this.channel = await this.connection.createChannel();
|
|
23
|
+
connected = true;
|
|
24
|
+
}
|
|
25
|
+
catch (e) {
|
|
26
|
+
logger_1.logger.debug(`Attempting to connect to alwaysAI Local Connection ${connectAttempts}`);
|
|
27
|
+
await (0, sleep_1.default)(1000 + 1000 * connectAttempts);
|
|
28
|
+
connectAttempts += 1;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
if (connected === true) {
|
|
32
|
+
this.channel.prefetch(1); // This ensures we only get one packet at a time! This appears to have prevented throttling
|
|
33
|
+
this.packetQueue = 'edgeiq-analytics-publish';
|
|
34
|
+
await this.channel.assertQueue(this.packetQueue, {
|
|
35
|
+
durable: true
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
else {
|
|
39
|
+
throw new Error('Unable to establish connection to alwaysAI Local Connection, please try restarting Device Agent.');
|
|
40
|
+
}
|
|
23
41
|
}
|
|
24
42
|
}
|
|
25
43
|
exports.PassthroughHandler = PassthroughHandler;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"passthrough-handler.js","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":";;;AAAA,2BAA2B;AAC3B,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;AAChC,iFAAiF;AACjF,2CAAwC;
|
|
1
|
+
{"version":3,"file":"passthrough-handler.js","sourceRoot":"","sources":["../../src/cloud-connection/passthrough-handler.ts"],"names":[],"mappings":";;;AAAA,2BAA2B;AAC3B,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;AAChC,iFAAiF;AACjF,2CAAwC;AACxC,yCAAkC;AAGlC,MAAM,YAAY,GAAU,EAAE,CAAC;AAC/B,MAAM,QAAQ,GAAU,EAAE,CAAC;AAE3B,MAAa,kBAAkB;IAM7B,YAAY,SAAoB;QAC9B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,MAAM,IAAA,4CAAsB,GAAE,CAAC;QAC/B,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,OAAO,eAAe,IAAI,EAAE,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE;YAC7D,IAAI;gBACF,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACzD,IAAI,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC;gBACrD,SAAS,GAAG,IAAI,CAAC;aAClB;YAAC,OAAO,CAAC,EAAE;gBACV,eAAM,CAAC,KAAK,CACV,sDAAsD,eAAe,EAAE,CACxE,CAAC;gBACF,MAAM,IAAA,eAAK,EAAC,IAAI,GAAG,IAAI,GAAG,eAAe,CAAC,CAAC;gBAC3C,eAAe,IAAI,CAAC,CAAC;aACtB;SACF;QACD,IAAI,SAAS,KAAK,IAAI,EAAE;YACtB,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,2FAA2F;YACrH,IAAI,CAAC,WAAW,GAAG,0BAA0B,CAAC;YAC9C,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE;gBAC/C,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;SACJ;aAAM;YACL,MAAM,IAAI,KAAK,CACb,kGAAkG,CACnG,CAAC;SACH;IACH,CAAC;CACF;AAvCD,gDAuCC;AAED,SAAS,cAAc,CAAC,kBAAsC;IAC5D,OAAO,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;QAC9B,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,EAAE,CAAC;QACnC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC;QAC9B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,6BAA6B;QAC7B,kBAAkB,CAAC,SAAS,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAAC,SAAS,EAAE,EAAE;YACvE,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;gBAC1B,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC7B,IAAI,SAAS,KAAK,IAAI,EAAE;oBACtB,kBAAkB,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,sCAAsC;iBAC5E;qBAAM,IAAI,SAAS,KAAK,KAAK,EAAE;oBAC9B,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,qBAAqB;iBACpE;aACF;QACH,CAAC,CAAC,CAAC;KACJ;AACH,CAAC;AAEM,KAAK,UAAU,UAAU,CAAC,kBAAsC;IACrE,eAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAC7C,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAChC,kBAAkB,CAAC,WAAW,EAC9B,UAAU,GAAG;QACX,IAAI,GAAG,CAAC,OAAO,KAAK,SAAS,EAAE;YAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;YAClD,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;YACnC,cAAc,CAAC,kBAAkB,CAAC,CAAC;SACpC;IACH,CAAC,EACD;QACE,KAAK,EAAE,KAAK,CAAC,gEAAgE;KAC9E,CACF,CAAC;AACJ,CAAC;AAfD,gCAeC"}
|
|
@@ -1,4 +1,7 @@
|
|
|
1
|
-
export declare
|
|
1
|
+
export declare const rabbitMQServiceName = "alwaysAIRabbitMQ";
|
|
2
|
+
export declare const rabbitMQContainerName = "alwaysAIRabbitMQContainer";
|
|
3
|
+
export declare function checkRabbitMQContainerRunning(): Promise<boolean | undefined>;
|
|
4
|
+
export declare function writeRabbitMQDockerComposeFile(): Promise<void>;
|
|
2
5
|
export declare function setupRabbitMQContainer(): Promise<void>;
|
|
3
6
|
export declare function stopRabbitMQContainer(): Promise<void>;
|
|
4
7
|
//# sourceMappingURL=rabbitmq-connection.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rabbitmq-connection.d.ts","sourceRoot":"","sources":["../../src/local-connection/rabbitmq-connection.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"rabbitmq-connection.d.ts","sourceRoot":"","sources":["../../src/local-connection/rabbitmq-connection.ts"],"names":[],"mappings":"AAWA,eAAO,MAAM,mBAAmB,qBAAqB,CAAC;AACtD,eAAO,MAAM,qBAAqB,8BAA8B,CAAC;AAEjE,wBAAsB,6BAA6B,iCAelD;AAED,wBAAsB,8BAA8B,kBAuBnD;AAED,wBAAsB,sBAAsB,kBA2B3C;AAED,wBAAsB,qBAAqB,kBAa1C"}
|
|
@@ -1,57 +1,79 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.stopRabbitMQContainer = exports.setupRabbitMQContainer = exports.checkRabbitMQContainerRunning = void 0;
|
|
4
|
-
const util_1 = require("alwaysai/lib/util");
|
|
3
|
+
exports.stopRabbitMQContainer = exports.setupRabbitMQContainer = exports.writeRabbitMQDockerComposeFile = exports.checkRabbitMQContainerRunning = exports.rabbitMQContainerName = exports.rabbitMQServiceName = void 0;
|
|
5
4
|
const logger_1 = require("../util/logger");
|
|
6
5
|
const sleep_1 = require("../util/sleep");
|
|
6
|
+
const docker_compose_1 = require("docker-compose");
|
|
7
|
+
const YAML = require("yaml");
|
|
8
|
+
const directories_1 = require("../util/directories");
|
|
9
|
+
const util_1 = require("alwaysai/lib/util");
|
|
10
|
+
exports.rabbitMQServiceName = 'alwaysAIRabbitMQ';
|
|
11
|
+
exports.rabbitMQContainerName = 'alwaysAIRabbitMQContainer';
|
|
7
12
|
async function checkRabbitMQContainerRunning() {
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
+
logger_1.logger.debug('Checking alwaysAI Local Connection Container status');
|
|
14
|
+
const containerData = await docker_compose_1.default.ps({ cwd: directories_1.DEVICE_AGENT_CFG_PATH });
|
|
15
|
+
if (!(containerData === undefined)) {
|
|
16
|
+
const rabbitmqService = containerData.data.services[0];
|
|
17
|
+
if (rabbitmqService.name === exports.rabbitMQContainerName &&
|
|
18
|
+
rabbitmqService.state === 'Up') {
|
|
19
|
+
logger_1.logger.debug('alwaysAI Local Connection is running');
|
|
20
|
+
return true;
|
|
21
|
+
}
|
|
22
|
+
logger_1.logger.debug('alwaysAI Local Connection is not running');
|
|
23
|
+
return false;
|
|
24
|
+
}
|
|
13
25
|
}
|
|
14
26
|
exports.checkRabbitMQContainerRunning = checkRabbitMQContainerRunning;
|
|
15
|
-
async function
|
|
27
|
+
async function writeRabbitMQDockerComposeFile() {
|
|
16
28
|
const spawner = (0, util_1.JsSpawner)();
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
'
|
|
25
|
-
'
|
|
26
|
-
'-
|
|
27
|
-
|
|
28
|
-
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
29
|
+
if (!(await spawner.exists(directories_1.DEVICE_AGENT_CFG_PATH))) {
|
|
30
|
+
await (0, util_1.JsSpawner)().mkdirp(directories_1.DEVICE_AGENT_CFG_PATH);
|
|
31
|
+
}
|
|
32
|
+
const rabbitmqDockerComposeCmd = {
|
|
33
|
+
services: {
|
|
34
|
+
[exports.rabbitMQServiceName]: {
|
|
35
|
+
container_name: exports.rabbitMQContainerName,
|
|
36
|
+
image: 'rabbitmq:3.11',
|
|
37
|
+
ports: ['5672:5672'],
|
|
38
|
+
hostname: 'my-rabbit',
|
|
39
|
+
restart: 'on-failure'
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
};
|
|
43
|
+
const RabbitMQDockerComposeYaml = YAML.stringify(rabbitmqDockerComposeCmd);
|
|
44
|
+
await spawner.writeFile(directories_1.DEVICE_AGENT_DOCKER_COMPOSE_PATH, RabbitMQDockerComposeYaml);
|
|
45
|
+
}
|
|
46
|
+
exports.writeRabbitMQDockerComposeFile = writeRabbitMQDockerComposeFile;
|
|
47
|
+
async function setupRabbitMQContainer() {
|
|
48
|
+
logger_1.logger.debug('Attempting to start the alwaysAI Local Connection Container');
|
|
49
|
+
try {
|
|
50
|
+
await writeRabbitMQDockerComposeFile();
|
|
51
|
+
}
|
|
52
|
+
catch (e) {
|
|
53
|
+
logger_1.logger.error(`An error occurred setting up docker-compose.yaml for Device Agent pass through:\n{e.message}`);
|
|
54
|
+
}
|
|
55
|
+
try {
|
|
56
|
+
const upOut = await docker_compose_1.default.upAll({
|
|
57
|
+
cwd: directories_1.DEVICE_AGENT_CFG_PATH
|
|
33
58
|
});
|
|
34
|
-
|
|
59
|
+
logger_1.logger.debug(`Docker compose up for alwaysAI Local Connection:\n${JSON.stringify(upOut)}`);
|
|
60
|
+
// check if the container is up
|
|
35
61
|
while (!(await checkRabbitMQContainerRunning())) {
|
|
36
|
-
await (0, sleep_1.default)(
|
|
62
|
+
await (0, sleep_1.default)(10);
|
|
37
63
|
}
|
|
38
64
|
}
|
|
65
|
+
catch (e) {
|
|
66
|
+
logger_1.logger.error(`Unable to start alwaysAI Device Agent Local Connection Container:\n${e.messaage}`);
|
|
67
|
+
}
|
|
39
68
|
}
|
|
40
69
|
exports.setupRabbitMQContainer = setupRabbitMQContainer;
|
|
41
70
|
async function stopRabbitMQContainer() {
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
logger_1.logger.debug('No RabbitMQ container running');
|
|
49
|
-
}
|
|
50
|
-
else {
|
|
51
|
-
await spawner.run({
|
|
52
|
-
exe: 'docker',
|
|
53
|
-
args: ['stop', `${rabbitMqContainer}`]
|
|
54
|
-
});
|
|
71
|
+
try {
|
|
72
|
+
const downOut = await docker_compose_1.default.down({ cwd: directories_1.DEVICE_AGENT_CFG_PATH });
|
|
73
|
+
logger_1.logger.debug(`Docker compose down for alwaysAI Local Connection:\n${JSON.stringify(downOut)}`);
|
|
74
|
+
}
|
|
75
|
+
catch (e) {
|
|
76
|
+
logger_1.logger.error(`Unable to stop alwaysAI Device Agent Local Connection Container:\n${e.message}`);
|
|
55
77
|
}
|
|
56
78
|
}
|
|
57
79
|
exports.stopRabbitMQContainer = stopRabbitMQContainer;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rabbitmq-connection.js","sourceRoot":"","sources":["../../src/local-connection/rabbitmq-connection.ts"],"names":[],"mappings":";;;AAAA,
|
|
1
|
+
{"version":3,"file":"rabbitmq-connection.js","sourceRoot":"","sources":["../../src/local-connection/rabbitmq-connection.ts"],"names":[],"mappings":";;;AAAA,2CAAwC;AACxC,yCAAkC;AAClC,mDAAqC;AACrC,6BAA6B;AAE7B,qDAG6B;AAC7B,4CAA8C;AAEjC,QAAA,mBAAmB,GAAG,kBAAkB,CAAC;AACzC,QAAA,qBAAqB,GAAG,2BAA2B,CAAC;AAE1D,KAAK,UAAU,6BAA6B;IACjD,eAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACpE,MAAM,aAAa,GAAG,MAAM,wBAAO,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,mCAAqB,EAAE,CAAC,CAAC;IACvE,IAAI,CAAC,CAAC,aAAa,KAAK,SAAS,CAAC,EAAE;QAClC,MAAM,eAAe,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACvD,IACE,eAAe,CAAC,IAAI,KAAK,6BAAqB;YAC9C,eAAe,CAAC,KAAK,KAAK,IAAI,EAC9B;YACA,eAAM,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;YACrD,OAAO,IAAI,CAAC;SACb;QACD,eAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QACzD,OAAO,KAAK,CAAC;KACd;AACH,CAAC;AAfD,sEAeC;AAEM,KAAK,UAAU,8BAA8B;IAClD,MAAM,OAAO,GAAG,IAAA,gBAAS,GAAE,CAAC;IAC5B,IAAI,CAAC,CAAC,MAAM,OAAO,CAAC,MAAM,CAAC,mCAAqB,CAAC,CAAC,EAAE;QAClD,MAAM,IAAA,gBAAS,GAAE,CAAC,MAAM,CAAC,mCAAqB,CAAC,CAAC;KACjD;IAED,MAAM,wBAAwB,GAAG;QAC/B,QAAQ,EAAE;YACR,CAAC,2BAAmB,CAAC,EAAE;gBACrB,cAAc,EAAE,6BAAqB;gBACrC,KAAK,EAAE,eAAe;gBACtB,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,QAAQ,EAAE,WAAW;gBACrB,OAAO,EAAE,YAAY;aACtB;SACF;KACF,CAAC;IAEF,MAAM,yBAAyB,GAAG,IAAI,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;IAC3E,MAAM,OAAO,CAAC,SAAS,CACrB,8CAAgC,EAChC,yBAAyB,CAC1B,CAAC;AACJ,CAAC;AAvBD,wEAuBC;AAEM,KAAK,UAAU,sBAAsB;IAC1C,eAAM,CAAC,KAAK,CAAC,6DAA6D,CAAC,CAAC;IAC5E,IAAI;QACF,MAAM,8BAA8B,EAAE,CAAC;KACxC;IAAC,OAAO,CAAC,EAAE;QACV,eAAM,CAAC,KAAK,CACV,8FAA8F,CAC/F,CAAC;KACH;IACD,IAAI;QACF,MAAM,KAAK,GAAG,MAAM,wBAAO,CAAC,KAAK,CAAC;YAChC,GAAG,EAAE,mCAAqB;SAC3B,CAAC,CAAC;QACH,eAAM,CAAC,KAAK,CACV,qDAAqD,IAAI,CAAC,SAAS,CACjE,KAAK,CACN,EAAE,CACJ,CAAC;QACF,+BAA+B;QAC/B,OAAO,CAAC,CAAC,MAAM,6BAA6B,EAAE,CAAC,EAAE;YAC/C,MAAM,IAAA,eAAK,EAAC,EAAE,CAAC,CAAC;SACjB;KACF;IAAC,OAAO,CAAC,EAAE;QACV,eAAM,CAAC,KAAK,CACV,sEAAsE,CAAC,CAAC,QAAQ,EAAE,CACnF,CAAC;KACH;AACH,CAAC;AA3BD,wDA2BC;AAEM,KAAK,UAAU,qBAAqB;IACzC,IAAI;QACF,MAAM,OAAO,GAAG,MAAM,wBAAO,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mCAAqB,EAAE,CAAC,CAAC;QACnE,eAAM,CAAC,KAAK,CACV,uDAAuD,IAAI,CAAC,SAAS,CACnE,OAAO,CACR,EAAE,CACJ,CAAC;KACH;IAAC,OAAO,CAAC,EAAE;QACV,eAAM,CAAC,KAAK,CACV,qEAAqE,CAAC,CAAC,OAAO,EAAE,CACjF,CAAC;KACH;AACH,CAAC;AAbD,sDAaC"}
|
|
@@ -1,7 +1,4 @@
|
|
|
1
|
-
export declare const subcommands: (import("@alwaysai/alwayscli/lib/types").CliBranch | import("@alwaysai/alwayscli/lib/types").CliLeaf<import("@alwaysai/alwayscli").CliInput<undefined, false>, {
|
|
2
|
-
email: import("@alwaysai/alwayscli").CliInput<string, true>;
|
|
3
|
-
password: import("@alwaysai/alwayscli").CliInput<string, true>;
|
|
4
|
-
}, import("@alwaysai/alwayscli").CliInput<undefined, false>> | import("@alwaysai/alwayscli/lib/types").CliLeaf<import("@alwaysai/alwayscli").CliInput<string, true>, {
|
|
1
|
+
export declare const subcommands: (import("@alwaysai/alwayscli/lib/types").CliBranch | import("@alwaysai/alwayscli/lib/types").CliLeaf<import("@alwaysai/alwayscli").CliInput<undefined, false>, {}, import("@alwaysai/alwayscli").CliInput<undefined, false>> | import("@alwaysai/alwayscli/lib/types").CliLeaf<import("@alwaysai/alwayscli").CliInput<string, true>, {
|
|
5
2
|
version: import("@alwaysai/alwayscli").CliInput<number | undefined, boolean>;
|
|
6
3
|
path: import("@alwaysai/alwayscli").CliInput<string | undefined, false>;
|
|
7
4
|
}, import("@alwaysai/alwayscli").CliInput<undefined, false>>)[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/subcommands/index.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/subcommands/index.ts"],"names":[],"mappings":"AAMA,eAAO,MAAM,WAAW;;;+DAMvB,CAAC"}
|
package/lib/subcommands/index.js
CHANGED
|
@@ -5,10 +5,12 @@ const app_1 = require("./app");
|
|
|
5
5
|
const device_1 = require("./device");
|
|
6
6
|
const login_1 = require("./login");
|
|
7
7
|
const get_model_package_1 = require("./get-model-package");
|
|
8
|
+
const rabbitmq_connection_1 = require("./rabbitmq-connection");
|
|
8
9
|
exports.subcommands = [
|
|
9
10
|
login_1.loginCliLeaf,
|
|
10
11
|
app_1.appCliBranch,
|
|
11
12
|
device_1.deviceCliBranch,
|
|
12
|
-
get_model_package_1.getModelPackageCliLeaf
|
|
13
|
+
get_model_package_1.getModelPackageCliLeaf,
|
|
14
|
+
rabbitmq_connection_1.stopRabbitMQContainerCliLeaf
|
|
13
15
|
];
|
|
14
16
|
//# sourceMappingURL=index.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/subcommands/index.ts"],"names":[],"mappings":";;;AAAA,+BAAqC;AACrC,qCAA2C;AAC3C,mCAAuC;AACvC,2DAA6D;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/subcommands/index.ts"],"names":[],"mappings":";;;AAAA,+BAAqC;AACrC,qCAA2C;AAC3C,mCAAuC;AACvC,2DAA6D;AAC7D,+DAAqE;AAExD,QAAA,WAAW,GAAG;IACzB,oBAAY;IACZ,kBAAY;IACZ,wBAAe;IACf,0CAAsB;IACtB,kDAA4B;CAC7B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rabbitmq-connection.d.ts","sourceRoot":"","sources":["../../src/subcommands/rabbitmq-connection.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,4BAA4B,yKAOvC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.stopRabbitMQContainerCliLeaf = void 0;
|
|
4
|
+
const alwayscli_1 = require("@alwaysai/alwayscli");
|
|
5
|
+
const rabbitmq_connection_1 = require("../local-connection/rabbitmq-connection");
|
|
6
|
+
exports.stopRabbitMQContainerCliLeaf = (0, alwayscli_1.CliLeaf)({
|
|
7
|
+
name: 'stop-analytics-pass-through',
|
|
8
|
+
description: 'Stop the Device Agent connection to edgeIQ',
|
|
9
|
+
hidden: true,
|
|
10
|
+
async action(_, opts) {
|
|
11
|
+
await (0, rabbitmq_connection_1.stopRabbitMQContainer)();
|
|
12
|
+
}
|
|
13
|
+
});
|
|
14
|
+
//# sourceMappingURL=rabbitmq-connection.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"rabbitmq-connection.js","sourceRoot":"","sources":["../../src/subcommands/rabbitmq-connection.ts"],"names":[],"mappings":";;;AAAA,mDAA8C;AAC9C,iFAAgF;AAEnE,QAAA,4BAA4B,GAAG,IAAA,mBAAO,EAAC;IAClD,IAAI,EAAE,6BAA6B;IACnC,WAAW,EAAE,4CAA4C;IACzD,MAAM,EAAE,IAAI;IACZ,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,IAAI;QAClB,MAAM,IAAA,2CAAqB,GAAE,CAAC;IAChC,CAAC;CACF,CAAC,CAAC"}
|
|
@@ -1,4 +1,7 @@
|
|
|
1
1
|
export declare const APP_ROOT: string;
|
|
2
|
+
export declare const DEVICE_AGENT_CFG_DIR = "device-agent";
|
|
3
|
+
export declare const DEVICE_AGENT_CFG_PATH: string;
|
|
4
|
+
export declare const DEVICE_AGENT_DOCKER_COMPOSE_PATH: string;
|
|
2
5
|
export declare const CREDENTIALS_FILE_PATH: string;
|
|
3
6
|
export declare const AWS_ROOT_CERTIFICATE_FILE_NAME = "AmazonRootCA1.pem";
|
|
4
7
|
export declare const AWS_ROOT_CERTIFICATE_FILE_PATH: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"directories.d.ts","sourceRoot":"","sources":["../../src/util/directories.ts"],"names":[],"mappings":"AAUA,eAAO,MAAM,QAAQ,
|
|
1
|
+
{"version":3,"file":"directories.d.ts","sourceRoot":"","sources":["../../src/util/directories.ts"],"names":[],"mappings":"AAUA,eAAO,MAAM,QAAQ,QAAgC,CAAC;AAEtD,eAAO,MAAM,oBAAoB,iBAAiB,CAAC;AACnD,eAAO,MAAM,qBAAqB,QAGjC,CAAC;AACF,eAAO,MAAM,gCAAgC,QAG5C,CAAC;AAEF,eAAO,MAAM,qBAAqB,QAGjC,CAAC;AAEF,eAAO,MAAM,8BAA8B,sBAAsB,CAAC;AAClE,eAAO,MAAM,8BAA8B,QAG1C,CAAC;AAEF,eAAO,MAAM,eAAe,yCAU3B,CAAC;AAMF,eAAO,MAAM,kBAAkB,2BAA2B,CAAC;AAC3D,eAAO,MAAM,+BAA+B,cACU,CAAC;AACvD,eAAO,MAAM,+BAA+B,cACG,CAAC;AAChD,eAAO,MAAM,sBAAsB,cACW,CAAC;AAC/C,eAAO,MAAM,qCAAqC,oCACf,CAAC;AAEpC,eAAO,MAAM,+BAA+B,cACM,CAAC;AAEnD,eAAO,MAAM,+BAA+B,cAKzC,CAAC;AAEJ,eAAO,MAAM,+BAA+B,cAKzC,CAAC;AAEJ,eAAO,MAAM,sBAAsB,cACyC,CAAC;AAM7E,eAAO,MAAM,4BAA4B,mCAAmC,CAAC;AAC7E,eAAO,MAAM,4BAA4B,4BAA4B,CAAC;AACtE,eAAO,MAAM,mBAAmB,2BAA2B,CAAC;AAE5D,eAAO,MAAM,4BAA4B,QAGxC,CAAC;AAEF,eAAO,MAAM,4BAA4B,QAGxC,CAAC"}
|
package/lib/util/directories.js
CHANGED
|
@@ -1,11 +1,13 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.DEVICE_CERTIFICATE_FILE_PATH = exports.DEVICE_PRIVATE_KEY_FILE_PATH = exports.DEVICE_ID_FILE_NAME = exports.DEVICE_CERTIFICATE_FILE_NAME = exports.DEVICE_PRIVATE_KEY_FILE_NAME = exports.BOOTSTRAP_ID_FILE_PATH = exports.BOOTSTRAP_CERTIFICATE_FILE_PATH = exports.BOOTSTRAP_PRIVATE_KEY_FILE_PATH = exports.BOOTSTRAP_CERTIFICATES_DIR_PATH = exports.CERTIFICATE_OWNERSHIP_TOKEN_FILE_NAME = exports.BOOTSTRAP_ID_FILE_NAME = exports.BOOTSTRAP_CERTIFICATE_FILE_NAME = exports.BOOTSTRAP_PRIVATE_KEY_FILE_NAME = exports.BOOTSTRAP_DIR_NAME = exports.shortenSystemId = exports.AWS_ROOT_CERTIFICATE_FILE_PATH = exports.AWS_ROOT_CERTIFICATE_FILE_NAME = exports.CREDENTIALS_FILE_PATH = exports.APP_ROOT = void 0;
|
|
3
|
+
exports.DEVICE_CERTIFICATE_FILE_PATH = exports.DEVICE_PRIVATE_KEY_FILE_PATH = exports.DEVICE_ID_FILE_NAME = exports.DEVICE_CERTIFICATE_FILE_NAME = exports.DEVICE_PRIVATE_KEY_FILE_NAME = exports.BOOTSTRAP_ID_FILE_PATH = exports.BOOTSTRAP_CERTIFICATE_FILE_PATH = exports.BOOTSTRAP_PRIVATE_KEY_FILE_PATH = exports.BOOTSTRAP_CERTIFICATES_DIR_PATH = exports.CERTIFICATE_OWNERSHIP_TOKEN_FILE_NAME = exports.BOOTSTRAP_ID_FILE_NAME = exports.BOOTSTRAP_CERTIFICATE_FILE_NAME = exports.BOOTSTRAP_PRIVATE_KEY_FILE_NAME = exports.BOOTSTRAP_DIR_NAME = exports.shortenSystemId = exports.AWS_ROOT_CERTIFICATE_FILE_PATH = exports.AWS_ROOT_CERTIFICATE_FILE_NAME = exports.CREDENTIALS_FILE_PATH = exports.DEVICE_AGENT_DOCKER_COMPOSE_PATH = exports.DEVICE_AGENT_CFG_PATH = exports.DEVICE_AGENT_CFG_DIR = exports.APP_ROOT = void 0;
|
|
4
4
|
const constants_1 = require("alwaysai/lib/constants");
|
|
5
5
|
const path_1 = require("path");
|
|
6
6
|
const system_id_1 = require("../infrastructure/system-id");
|
|
7
|
-
|
|
8
|
-
exports.
|
|
7
|
+
exports.APP_ROOT = (0, path_1.join)(constants_1.AAI_DIR, 'applications');
|
|
8
|
+
exports.DEVICE_AGENT_CFG_DIR = 'device-agent';
|
|
9
|
+
exports.DEVICE_AGENT_CFG_PATH = (0, path_1.join)(constants_1.LOCAL_AAI_CFG_DIR, exports.DEVICE_AGENT_CFG_DIR);
|
|
10
|
+
exports.DEVICE_AGENT_DOCKER_COMPOSE_PATH = (0, path_1.join)(exports.DEVICE_AGENT_CFG_PATH, constants_1.DOCKER_COMPOSE_FILE);
|
|
9
11
|
exports.CREDENTIALS_FILE_PATH = (0, path_1.join)(constants_1.LOCAL_AAI_CFG_DIR, constants_1.DEVICE_TOKEN_FILE_NAME);
|
|
10
12
|
exports.AWS_ROOT_CERTIFICATE_FILE_NAME = 'AmazonRootCA1.pem';
|
|
11
13
|
exports.AWS_ROOT_CERTIFICATE_FILE_PATH = (0, path_1.join)(constants_1.LOCAL_CERT_AND_KEY_DIR, exports.AWS_ROOT_CERTIFICATE_FILE_NAME);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"directories.js","sourceRoot":"","sources":["../../src/util/directories.ts"],"names":[],"mappings":";;;AAAA,
|
|
1
|
+
{"version":3,"file":"directories.js","sourceRoot":"","sources":["../../src/util/directories.ts"],"names":[],"mappings":";;;AAAA,sDAMgC;AAChC,+BAA4B;AAC5B,2DAA0D;AAE7C,QAAA,QAAQ,GAAG,IAAA,WAAI,EAAC,mBAAO,EAAE,cAAc,CAAC,CAAC;AAEzC,QAAA,oBAAoB,GAAG,cAAc,CAAC;AACtC,QAAA,qBAAqB,GAAG,IAAA,WAAI,EACvC,6BAAiB,EACjB,4BAAoB,CACrB,CAAC;AACW,QAAA,gCAAgC,GAAG,IAAA,WAAI,EAClD,6BAAqB,EACrB,+BAAmB,CACpB,CAAC;AAEW,QAAA,qBAAqB,GAAG,IAAA,WAAI,EACvC,6BAAiB,EACjB,kCAAsB,CACvB,CAAC;AAEW,QAAA,8BAA8B,GAAG,mBAAmB,CAAC;AACrD,QAAA,8BAA8B,GAAG,IAAA,WAAI,EAChD,kCAAsB,EACtB,sCAA8B,CAC/B,CAAC;AAEK,MAAM,eAAe,GAAG,GAAG,EAAE;IAClC,MAAM,EAAE,GAAG,IAAA,uBAAW,GAAE,CAAC;IACzB,QAAQ,EAAE,EAAE;QACV,KAAK,aAAa;YAChB,OAAO,KAAK,CAAC;QACf,KAAK,IAAI;YACP,OAAO,IAAI,CAAC;QACd,KAAK,YAAY;YACf,OAAO,MAAM,CAAC;KACjB;AACH,CAAC,CAAC;AAVW,QAAA,eAAe,mBAU1B;AAEF;;qEAEqE;AAExD,QAAA,kBAAkB,GAAG,wBAAwB,CAAC;AACpD,MAAM,+BAA+B,GAAG,GAAG,EAAE,CAClD,yBAAyB,IAAA,uBAAe,GAAE,UAAU,CAAC;AAD1C,QAAA,+BAA+B,mCACW;AAChD,MAAM,+BAA+B,GAAG,GAAG,EAAE,CAClD,kBAAkB,IAAA,uBAAe,GAAE,UAAU,CAAC;AADnC,QAAA,+BAA+B,mCACI;AACzC,MAAM,sBAAsB,GAAG,GAAG,EAAE,CACzC,qBAAqB,IAAA,uBAAe,GAAE,MAAM,CAAC;AADlC,QAAA,sBAAsB,0BACY;AAClC,QAAA,qCAAqC,GAChD,iCAAiC,CAAC;AAE7B,MAAM,+BAA+B,GAAG,GAAG,EAAE,CAClD,IAAA,WAAI,EAAC,kCAAsB,EAAE,0BAAkB,CAAC,CAAC;AADtC,QAAA,+BAA+B,mCACO;AAE5C,MAAM,+BAA+B,GAAG,GAAG,EAAE,CAClD,IAAA,WAAI,EACF,kCAAsB,EACtB,0BAAkB,EAClB,IAAA,uCAA+B,GAAE,CAClC,CAAC;AALS,QAAA,+BAA+B,mCAKxC;AAEG,MAAM,+BAA+B,GAAG,GAAG,EAAE,CAClD,IAAA,WAAI,EACF,kCAAsB,EACtB,0BAAkB,EAClB,IAAA,uCAA+B,GAAE,CAClC,CAAC;AALS,QAAA,+BAA+B,mCAKxC;AAEG,MAAM,sBAAsB,GAAG,GAAG,EAAE,CACzC,IAAA,WAAI,EAAC,kCAAsB,EAAE,0BAAkB,EAAE,IAAA,8BAAsB,GAAE,CAAC,CAAC;AADhE,QAAA,sBAAsB,0BAC0C;AAE7E;;qEAEqE;AAExD,QAAA,4BAA4B,GAAG,gCAAgC,CAAC;AAChE,QAAA,4BAA4B,GAAG,yBAAyB,CAAC;AACzD,QAAA,mBAAmB,GAAG,wBAAwB,CAAC;AAE/C,QAAA,4BAA4B,GAAG,IAAA,WAAI,EAC9C,kCAAsB,EACtB,oCAA4B,CAC7B,CAAC;AAEW,QAAA,4BAA4B,GAAG,IAAA,WAAI,EAC9C,kCAAsB,EACtB,oCAA4B,CAC7B,CAAC"}
|
package/package.json
CHANGED
package/readme.md
CHANGED
|
@@ -9,7 +9,7 @@ Note that the Device Agent is still in an experimental phase and these commands
|
|
|
9
9
|
are likely to change. This guide will be updated with the latest usage as things
|
|
10
10
|
change.
|
|
11
11
|
|
|
12
|
-
## System Requirements
|
|
12
|
+
## System Requirements & Prerequisites
|
|
13
13
|
|
|
14
14
|
* Supported OS:
|
|
15
15
|
* Debian Bullseye, Buster
|
|
@@ -21,6 +21,14 @@ change.
|
|
|
21
21
|
* armv7hf
|
|
22
22
|
* `docker` version >= 19.03
|
|
23
23
|
* `curl` installed (required to download provisioning scripts)
|
|
24
|
+
* Passwordless `sudo` for `npm` if using `pm2`
|
|
25
|
+
|
|
26
|
+
To enable passwordless `sudo` for `npm` for the current user, run `sudo visudo`
|
|
27
|
+
and add the following line to the end of the file:
|
|
28
|
+
|
|
29
|
+
```bash
|
|
30
|
+
<username> <hostname> = (root) NOPASSWD: /usr/bin/npm
|
|
31
|
+
```
|
|
24
32
|
|
|
25
33
|
## Provision Device
|
|
26
34
|
|
|
@@ -55,11 +63,36 @@ Where:
|
|
|
55
63
|
devices page of the alwaysAI Dashboard. If a device name is not provided, one
|
|
56
64
|
will be generated for you and logged to the console for reference.
|
|
57
65
|
|
|
66
|
+
Confirm the Device Agent is running with the following command:
|
|
67
|
+
|
|
68
|
+
```bash
|
|
69
|
+
$ pm2 list
|
|
70
|
+
┌────┬────────────────────┬──────────┬──────┬───────────┬──────────┬──────────┐
|
|
71
|
+
│ id │ name │ mode │ ↺ │ status │ cpu │ memory │
|
|
72
|
+
├────┼────────────────────┼──────────┼──────┼───────────┼──────────┼──────────┤
|
|
73
|
+
│ 0 │ aai-agent │ fork │ 15 │ online │ 0% │ 2.8mb │
|
|
74
|
+
└────┴────────────────────┴──────────┴──────┴───────────┴──────────┴──────────┘
|
|
75
|
+
```
|
|
76
|
+
|
|
77
|
+
To restart the Device Agent run the following command:
|
|
78
|
+
|
|
79
|
+
```bash
|
|
80
|
+
$ pm2 restart aai-agent
|
|
81
|
+
Use --update-env to update environment variables
|
|
82
|
+
[PM2] Applying action restartProcessId on app [aai-agent](ids: [ 0 ])
|
|
83
|
+
[PM2] [aai-agent](0) ✓
|
|
84
|
+
┌────┬────────────────────┬──────────┬──────┬───────────┬──────────┬──────────┐
|
|
85
|
+
│ id │ name │ mode │ ↺ │ status │ cpu │ memory │
|
|
86
|
+
├────┼────────────────────┼──────────┼──────┼───────────┼──────────┼──────────┤
|
|
87
|
+
│ 0 │ aai-agent │ fork │ 16 │ online │ 0% │ 2.6mb │
|
|
88
|
+
└────┴────────────────────┴──────────┴──────┴───────────┴──────────┴──────────┘
|
|
89
|
+
```
|
|
90
|
+
|
|
58
91
|
If you'd like to only provision the device, but not start the Device Agent in
|
|
59
92
|
the background (skip step 3), run with the `--provision-only` flag:
|
|
60
93
|
|
|
61
94
|
```
|
|
62
|
-
$ curl -fsSL https://artifacts.alwaysai.co/device-agent/provision
|
|
95
|
+
$ curl -fsSL https://artifacts.alwaysai.co/device-agent/provision.sh | bash -s -- --email <email> --password <password> [--device-name <device_name>] --provision-only
|
|
63
96
|
```
|
|
64
97
|
|
|
65
98
|
When the script completes, the device will be provisioned but the Device Agent will not be running in the background.
|
|
@@ -2,6 +2,7 @@
|
|
|
2
2
|
const amqp = require('amqplib');
|
|
3
3
|
import { setupRabbitMQContainer } from '../local-connection/rabbitmq-connection';
|
|
4
4
|
import { logger } from '../util/logger';
|
|
5
|
+
import sleep from '../util/sleep';
|
|
5
6
|
import { Publisher } from './publisher';
|
|
6
7
|
|
|
7
8
|
const messageQueue: any[] = [];
|
|
@@ -19,13 +20,32 @@ export class PassthroughHandler {
|
|
|
19
20
|
|
|
20
21
|
public async setup() {
|
|
21
22
|
await setupRabbitMQContainer();
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
|
|
23
|
+
let connectAttempts = 0;
|
|
24
|
+
let connected = false;
|
|
25
|
+
while (connectAttempts <= 10 && this.connection === undefined) {
|
|
26
|
+
try {
|
|
27
|
+
this.connection = await amqp.connect('amqp://localhost');
|
|
28
|
+
this.channel = await this.connection.createChannel();
|
|
29
|
+
connected = true;
|
|
30
|
+
} catch (e) {
|
|
31
|
+
logger.debug(
|
|
32
|
+
`Attempting to connect to alwaysAI Local Connection ${connectAttempts}`
|
|
33
|
+
);
|
|
34
|
+
await sleep(1000 + 1000 * connectAttempts);
|
|
35
|
+
connectAttempts += 1;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
if (connected === true) {
|
|
39
|
+
this.channel.prefetch(1); // This ensures we only get one packet at a time! This appears to have prevented throttling
|
|
40
|
+
this.packetQueue = 'edgeiq-analytics-publish';
|
|
41
|
+
await this.channel.assertQueue(this.packetQueue, {
|
|
42
|
+
durable: true
|
|
43
|
+
});
|
|
44
|
+
} else {
|
|
45
|
+
throw new Error(
|
|
46
|
+
'Unable to establish connection to alwaysAI Local Connection, please try restarting Device Agent.'
|
|
47
|
+
);
|
|
48
|
+
}
|
|
29
49
|
}
|
|
30
50
|
}
|
|
31
51
|
|
|
@@ -1,53 +1,99 @@
|
|
|
1
|
-
import { JsSpawner } from 'alwaysai/lib/util';
|
|
2
1
|
import { logger } from '../util/logger';
|
|
3
2
|
import sleep from '../util/sleep';
|
|
3
|
+
import compose from 'docker-compose';
|
|
4
|
+
import * as YAML from 'yaml';
|
|
5
|
+
|
|
6
|
+
import {
|
|
7
|
+
DEVICE_AGENT_CFG_PATH,
|
|
8
|
+
DEVICE_AGENT_DOCKER_COMPOSE_PATH
|
|
9
|
+
} from '../util/directories';
|
|
10
|
+
import { JsSpawner } from 'alwaysai/lib/util';
|
|
11
|
+
|
|
12
|
+
export const rabbitMQServiceName = 'alwaysAIRabbitMQ';
|
|
13
|
+
export const rabbitMQContainerName = 'alwaysAIRabbitMQContainer';
|
|
4
14
|
|
|
5
15
|
export async function checkRabbitMQContainerRunning() {
|
|
16
|
+
logger.debug('Checking alwaysAI Local Connection Container status');
|
|
17
|
+
const containerData = await compose.ps({ cwd: DEVICE_AGENT_CFG_PATH });
|
|
18
|
+
if (!(containerData === undefined)) {
|
|
19
|
+
const rabbitmqService = containerData.data.services[0];
|
|
20
|
+
if (
|
|
21
|
+
rabbitmqService.name === rabbitMQContainerName &&
|
|
22
|
+
rabbitmqService.state === 'Up'
|
|
23
|
+
) {
|
|
24
|
+
logger.debug('alwaysAI Local Connection is running');
|
|
25
|
+
return true;
|
|
26
|
+
}
|
|
27
|
+
logger.debug('alwaysAI Local Connection is not running');
|
|
28
|
+
return false;
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
|
|
32
|
+
export async function writeRabbitMQDockerComposeFile() {
|
|
6
33
|
const spawner = JsSpawner();
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
34
|
+
if (!(await spawner.exists(DEVICE_AGENT_CFG_PATH))) {
|
|
35
|
+
await JsSpawner().mkdirp(DEVICE_AGENT_CFG_PATH);
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
const rabbitmqDockerComposeCmd = {
|
|
39
|
+
services: {
|
|
40
|
+
[rabbitMQServiceName]: {
|
|
41
|
+
container_name: rabbitMQContainerName,
|
|
42
|
+
image: 'rabbitmq:3.11',
|
|
43
|
+
ports: ['5672:5672'],
|
|
44
|
+
hostname: 'my-rabbit',
|
|
45
|
+
restart: 'on-failure'
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
};
|
|
49
|
+
|
|
50
|
+
const RabbitMQDockerComposeYaml = YAML.stringify(rabbitmqDockerComposeCmd);
|
|
51
|
+
await spawner.writeFile(
|
|
52
|
+
DEVICE_AGENT_DOCKER_COMPOSE_PATH,
|
|
53
|
+
RabbitMQDockerComposeYaml
|
|
54
|
+
);
|
|
11
55
|
}
|
|
12
56
|
|
|
13
57
|
export async function setupRabbitMQContainer() {
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
logger.
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
'5672:5672',
|
|
26
|
-
'-d',
|
|
27
|
-
'--hostname',
|
|
28
|
-
'my-rabbit',
|
|
29
|
-
'rabbitmq'
|
|
30
|
-
]
|
|
58
|
+
logger.debug('Attempting to start the alwaysAI Local Connection Container');
|
|
59
|
+
try {
|
|
60
|
+
await writeRabbitMQDockerComposeFile();
|
|
61
|
+
} catch (e) {
|
|
62
|
+
logger.error(
|
|
63
|
+
`An error occurred setting up docker-compose.yaml for Device Agent pass through:\n{e.message}`
|
|
64
|
+
);
|
|
65
|
+
}
|
|
66
|
+
try {
|
|
67
|
+
const upOut = await compose.upAll({
|
|
68
|
+
cwd: DEVICE_AGENT_CFG_PATH
|
|
31
69
|
});
|
|
32
|
-
|
|
70
|
+
logger.debug(
|
|
71
|
+
`Docker compose up for alwaysAI Local Connection:\n${JSON.stringify(
|
|
72
|
+
upOut
|
|
73
|
+
)}`
|
|
74
|
+
);
|
|
75
|
+
// check if the container is up
|
|
33
76
|
while (!(await checkRabbitMQContainerRunning())) {
|
|
34
|
-
await sleep(
|
|
77
|
+
await sleep(10);
|
|
35
78
|
}
|
|
79
|
+
} catch (e) {
|
|
80
|
+
logger.error(
|
|
81
|
+
`Unable to start alwaysAI Device Agent Local Connection Container:\n${e.messaage}`
|
|
82
|
+
);
|
|
36
83
|
}
|
|
37
84
|
}
|
|
38
85
|
|
|
39
86
|
export async function stopRabbitMQContainer() {
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
}
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
});
|
|
87
|
+
try {
|
|
88
|
+
const downOut = await compose.down({ cwd: DEVICE_AGENT_CFG_PATH });
|
|
89
|
+
logger.debug(
|
|
90
|
+
`Docker compose down for alwaysAI Local Connection:\n${JSON.stringify(
|
|
91
|
+
downOut
|
|
92
|
+
)}`
|
|
93
|
+
);
|
|
94
|
+
} catch (e) {
|
|
95
|
+
logger.error(
|
|
96
|
+
`Unable to stop alwaysAI Device Agent Local Connection Container:\n${e.message}`
|
|
97
|
+
);
|
|
52
98
|
}
|
|
53
99
|
}
|
package/src/subcommands/index.ts
CHANGED
|
@@ -2,10 +2,12 @@ import { appCliBranch } from './app';
|
|
|
2
2
|
import { deviceCliBranch } from './device';
|
|
3
3
|
import { loginCliLeaf } from './login';
|
|
4
4
|
import { getModelPackageCliLeaf } from './get-model-package';
|
|
5
|
+
import { stopRabbitMQContainerCliLeaf } from './rabbitmq-connection';
|
|
5
6
|
|
|
6
7
|
export const subcommands = [
|
|
7
8
|
loginCliLeaf,
|
|
8
9
|
appCliBranch,
|
|
9
10
|
deviceCliBranch,
|
|
10
|
-
getModelPackageCliLeaf
|
|
11
|
+
getModelPackageCliLeaf,
|
|
12
|
+
stopRabbitMQContainerCliLeaf
|
|
11
13
|
];
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { CliLeaf } from '@alwaysai/alwayscli';
|
|
2
|
+
import { stopRabbitMQContainer } from '../local-connection/rabbitmq-connection';
|
|
3
|
+
|
|
4
|
+
export const stopRabbitMQContainerCliLeaf = CliLeaf({
|
|
5
|
+
name: 'stop-analytics-pass-through',
|
|
6
|
+
description: 'Stop the Device Agent connection to edgeIQ',
|
|
7
|
+
hidden: true,
|
|
8
|
+
async action(_, opts) {
|
|
9
|
+
await stopRabbitMQContainer();
|
|
10
|
+
}
|
|
11
|
+
});
|
package/src/util/directories.ts
CHANGED
|
@@ -1,14 +1,24 @@
|
|
|
1
1
|
import {
|
|
2
2
|
AAI_DIR,
|
|
3
3
|
DEVICE_TOKEN_FILE_NAME,
|
|
4
|
+
DOCKER_COMPOSE_FILE,
|
|
4
5
|
LOCAL_AAI_CFG_DIR,
|
|
5
6
|
LOCAL_CERT_AND_KEY_DIR
|
|
6
7
|
} from 'alwaysai/lib/constants';
|
|
7
8
|
import { join } from 'path';
|
|
8
9
|
import { getSystemId } from '../infrastructure/system-id';
|
|
9
|
-
import * as path from 'path';
|
|
10
10
|
|
|
11
|
-
export const APP_ROOT =
|
|
11
|
+
export const APP_ROOT = join(AAI_DIR, 'applications');
|
|
12
|
+
|
|
13
|
+
export const DEVICE_AGENT_CFG_DIR = 'device-agent';
|
|
14
|
+
export const DEVICE_AGENT_CFG_PATH = join(
|
|
15
|
+
LOCAL_AAI_CFG_DIR,
|
|
16
|
+
DEVICE_AGENT_CFG_DIR
|
|
17
|
+
);
|
|
18
|
+
export const DEVICE_AGENT_DOCKER_COMPOSE_PATH = join(
|
|
19
|
+
DEVICE_AGENT_CFG_PATH,
|
|
20
|
+
DOCKER_COMPOSE_FILE
|
|
21
|
+
);
|
|
12
22
|
|
|
13
23
|
export const CREDENTIALS_FILE_PATH = join(
|
|
14
24
|
LOCAL_AAI_CFG_DIR,
|