@xomlo/xomlo-sdk
Version:
121 lines • 6.21 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.main = void 0;
const AWS = __importStar(require("aws-sdk"));
const constants_1 = require("../constants");
const logger_1 = require("./log/logger");
const uploadToS3_1 = __importDefault(require("./storage/aws/uploadToS3"));
const containerStart_1 = require("./network/containerStart");
const containerEnd_1 = require("./network/containerEnd");
const envs_1 = require("../envs");
const loadLogs_1 = require("./storage/loadLogs");
function main(fn) {
return __awaiter(this, void 0, void 0, function* () {
logger_1.logger.debug(`Connector started in ${envs_1.isProduction ? 'production' : 'develop'} mode`);
envs_1.isProduction ? yield production(fn) : yield develop(fn);
logger_1.logger.debug(`Connector finished`);
});
}
exports.main = main;
function production(fn) {
return __awaiter(this, void 0, void 0, function* () {
logger_1.logger.debug(`Loading envs.`);
const envs = (0, envs_1.loadEnvsProduction)();
logger_1.logger.debug(`Setting AWS bucket secrets.`);
AWS.config.update({
accessKeyId: envs.aws.accessKey,
secretAccessKey: envs.aws.secretKey,
});
try {
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.RUNNING}`);
// get official payload on production otherwise getEnvs
const payload = yield (0, containerStart_1.containerStart)(envs.containerApiUrl, envs.taskId);
const fixedPayload = [];
Object.values(payload.payload).forEach((item) => {
fixedPayload[item.key] = item.value;
});
const startTime = process.hrtime();
try {
yield fn(fixedPayload);
const end = parseHrtimeToMilis(process.hrtime(startTime));
// upload result to bucket on production
logger_1.logger.debug(`Uploading result to ${envs.aws.bucket}/${payload.bucketName}/${constants_1.outputPath}`);
yield (0, uploadToS3_1.default)(envs.aws.bucket, payload.bucketName, constants_1.outputPath, 'public-read');
const logs = yield (0, loadLogs_1.loadLogs)();
logger_1.logger.debug('Uploading logs');
yield (0, containerEnd_1.containerEnd)(envs.containerApiUrl, payload.runId, constants_1.StatusValue.FINISHED, { duration: end }, logs);
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.FINISHED}, elapsedTime: ${end}ms`);
}
catch (err) {
logger_1.logger.error('Error in main function', err);
const end = parseHrtimeToMilis(process.hrtime(startTime));
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.FAILED}, elapsedTime: ${end}ms`);
const logs = yield (0, loadLogs_1.loadLogs)();
yield (0, containerEnd_1.containerEnd)(envs.containerApiUrl, payload.runId, constants_1.StatusValue.FAILED, { duration: end }, logs);
}
}
catch (err) {
logger_1.logger.error(`Could not get payload: ${err}`);
}
});
}
function develop(fn) {
return __awaiter(this, void 0, void 0, function* () {
logger_1.logger.debug(`Loading envs startsWith XOMLO_PAYLOAD_ from .env file`);
const envs = (0, envs_1.loadEnvsDevelop)();
try {
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.RUNNING}`);
// get official payload on production otherwise getEnvs
const payload = { runId: 'test-run', payload: envs };
const startTime = process.hrtime();
try {
yield fn(payload.payload);
const end = parseHrtimeToMilis(process.hrtime(startTime));
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.FINISHED}, elapsedTime: ${end}ms`);
}
catch (err) {
const end = parseHrtimeToMilis(process.hrtime(startTime));
logger_1.logger.error('Error in main function', err);
logger_1.logger.debug(`Connector status: ${constants_1.StatusValue.FAILED}, elapsedTime: ${end}ms`);
}
}
catch (err) {
logger_1.logger.error(err);
}
});
}
function parseHrtimeToMilis(hrtime) {
return Number(((hrtime[0] * 1e9 + hrtime[1]) / 1e6).toFixed());
}
//# sourceMappingURL=main.js.map