@csermet/multiprovider
Version:
cloud-graph provider plugin for AWS used to fetch AWS cloud data.
50 lines (49 loc) • 2.06 kB
JavaScript
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const sagemaker_1 = __importDefault(require("aws-sdk/clients/sagemaker"));
const lodash_1 = require("lodash");
const fetchUtils_1 = require("../../utils/fetchUtils");
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const serviceName = 'sageMakerNotebookInstance';
const endpoint = utils_1.initTestEndpoint(serviceName);
const errorLog = new errorLog_1.default(serviceName);
/**
* SageMakerNotebookInstance
*/
exports.default = async ({ regions, config, }) => {
const result = [];
const activeRegions = regions.split(',');
for (const region of activeRegions) {
const client = new sagemaker_1.default({ ...config, region, endpoint });
let sageMakerNotebookInstanceData;
try {
sageMakerNotebookInstanceData = await fetchUtils_1.fetchAllPaginatedData({
getResourcesFn: fetchUtils_1.convertToPromise({
sdkContext: client,
fnName: 'listNotebookInstances',
}),
accessor: '',
});
}
catch (err) {
errorLog.generateAwsErrorLog({ functionName: 'listNotebookInstances', err });
}
if (!lodash_1.isEmpty(sageMakerNotebookInstanceData)) {
for (const notebook of sageMakerNotebookInstanceData) {
try {
const notebookData = await client.describeNotebookInstance({ NotebookInstanceName: notebook.NotebookInstanceName }).promise();
result.push({ ...notebookData, region });
}
catch (err) {
errorLog.generateAwsErrorLog({ functionName: 'describeNotebookInstance', err });
}
}
}
}
errorLog.reset();
return lodash_1.groupBy(result, 'region');
};