@aws-amplify/analytics/dist/esm/providers/kinesis-firehose/utils/getEventBuffer.mjs

Analytics category of aws-amplify

import { haveCredentialsChanged } from '@aws-amplify/core/internals/utils';
import { FirehoseClient, PutRecordBatchCommand } from '@aws-sdk/client-firehose';
import '@aws-amplify/core';
import '../../../errors/validation.mjs';
import { groupBy } from '../../../utils/groupBy.mjs';
import { EventBuffer } from '../../../utils/eventBuffer/EventBuffer.mjs';
import '../../../trackers/EventTracker.mjs';
import '../../../trackers/PageViewTracker.mjs';
import '../../../trackers/SessionTracker.mjs';

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
/**
 * These Records hold cached event buffers and AWS clients.
 * The hash key is determined by the region and session,
 * consisting of a combined value comprising [region, sessionToken, identityId].
 *
 * Only one active session should exist at any given moment.
 * When a new session is initiated, the previous ones should be released.
 * */
const eventBufferMap = {};
const cachedClients = {};
const createPutRecordsBatchCommand = (streamName, events) => new PutRecordBatchCommand({
    DeliveryStreamName: streamName,
    Records: events.map(event => ({
        Data: event.event,
    })),
});
const submitEvents = async (events, client, resendLimit) => {
    const groupedByStreamName = Object.entries(groupBy(event => event.streamName, events));
    const requests = groupedByStreamName
        .map(([streamName, groupedEvents]) => createPutRecordsBatchCommand(streamName, groupedEvents))
        .map(command => client.send(command));
    const responses = await Promise.allSettled(requests);
    const failedEvents = responses
        .map((response, i) => response.status === 'rejected' ? groupedByStreamName[i][1] : [])
        .flat();
    return resendLimit
        ? failedEvents
            .filter(event => event.retryCount < resendLimit)
            .map(event => ({ ...event, retryCount: event.retryCount + 1 }))
            .sort((a, b) => a.timestamp - b.timestamp)
        : [];
};
const getEventBuffer = ({ region, credentials, identityId, bufferSize, flushSize, flushInterval, resendLimit, userAgentValue, }) => {
    const sessionIdentityKey = [region, identityId].filter(id => !!id).join('-');
    const [cachedClient, cachedCredentials] = cachedClients[sessionIdentityKey] ?? [];
    let credentialsHaveChanged = false;
    // Check if credentials have changed for the cached client
    if (cachedClient) {
        credentialsHaveChanged = haveCredentialsChanged(cachedCredentials, credentials);
    }
    if (!eventBufferMap[sessionIdentityKey] || credentialsHaveChanged) {
        const getClient = () => {
            if (!cachedClient || credentialsHaveChanged) {
                cachedClients[sessionIdentityKey] = [
                    new FirehoseClient({
                        region,
                        credentials,
                        customUserAgent: userAgentValue,
                    }),
                    credentials,
                ];
            }
            const [firehoseClient] = cachedClients[sessionIdentityKey];
            return events => submitEvents(events, firehoseClient, resendLimit);
        };
        eventBufferMap[sessionIdentityKey] =
            new EventBuffer({
                bufferSize,
                flushSize,
                flushInterval,
            }, getClient);
        const releaseSessionKeys = Object.keys(eventBufferMap).filter(key => key !== sessionIdentityKey);
        for (const releaseSessionKey of releaseSessionKeys) {
            eventBufferMap[releaseSessionKey].flushAll().finally(() => {
                eventBufferMap[releaseSessionKey].release();
                delete eventBufferMap[releaseSessionKey];
                delete cachedClients[releaseSessionKey];
            });
        }
    }
    return eventBufferMap[sessionIdentityKey];
};

export { getEventBuffer };
//# sourceMappingURL=getEventBuffer.mjs.map