@csermet/multiprovider/dist/services/sageMakerExperiment/data.js

cloud-graph provider plugin for AWS used to fetch AWS cloud data.

"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 groupBy_1 = __importDefault(require("lodash/groupBy"));
const fetchUtils_1 = require("../../utils/fetchUtils");
const utils_1 = require("../../utils");
const errorLog_1 = __importDefault(require("../../utils/errorLog"));
const lodash_1 = require("lodash");
const serviceName = 'sageMakerExperiment';
const errorLog = new errorLog_1.default(serviceName);
const endpoint = utils_1.initTestEndpoint(serviceName);
/**
 * SageMakerExperiment
 */
// TODO: Create connections using sourceArn and sourceType to other services (other sage maker services it seems)
exports.default = async ({ regions, config, }) => {
    const result = [];
    const activeRegions = regions.split(',');
    for (const region of activeRegions) {
        let sageMakerExperimentData = [];
        try {
            sageMakerExperimentData = await fetchUtils_1.fetchAllPaginatedData({
                getResourcesFn: fetchUtils_1.convertToPromise({
                    sdkContext: new sagemaker_1.default({ ...config, region, endpoint }),
                    fnName: 'listExperiments',
                }),
                accessor: '',
            });
        }
        catch (err) {
            errorLog.generateAwsErrorLog({
                functionName: 'listExperiments',
                err,
            });
        }
        if (!lodash_1.isEmpty(sageMakerExperimentData))
            result.push(...sageMakerExperimentData.map(val => ({ ...val, region })));
    }
    return groupBy_1.default(result, 'region');
};