@appsemble/node-utils/container/scale.js

NodeJS utilities used by Appsemble internally.

import { logger } from '@appsemble/node-utils';
import { getContainerNamespace, getKubeConfig, handleKubernetesError } from './helpers.js';
export async function scaleDeployment(namespace, deploymentName, replicas = 0) {
    const { appsApi } = getKubeConfig();
    const patch = [];
    patch.push({
        op: 'replace',
        path: '/spec/replicas',
        value: replicas,
    });
    try {
        logger.verbose(`Scaling deployment ${deploymentName} to ${replicas} replicas ... `);
        await appsApi.patchNamespacedDeployment({
            name: deploymentName,
            namespace,
            body: patch,
        });
        logger.verbose(`Deployment ${deploymentName} scaled to ${replicas} replicas successfully `);
    }
    catch (error) {
        handleKubernetesError(error);
        throw error;
    }
}
export async function stopIdleContainers(interval = 10) {
    const { appsApi } = getKubeConfig();
    let deployments;
    logger.verbose('Scaling containers');
    try {
        deployments = await appsApi.listDeploymentForAllNamespaces();
    }
    catch (error) {
        handleKubernetesError(error);
        return;
    }
    for (const deployment of deployments.items) {
        // @ts-expect-error 2339 Property does not exist on type undefined
        const { annotations, name, namespace } = deployment.metadata;
        if (deployment.metadata?.namespace !== getContainerNamespace() ||
            deployment.spec?.replicas === 0) {
            continue;
        }
        const milliseconds = interval * 60 * 1000;
        const now = new Date();
        const lastCall = annotations.lastRequestTimestamp
            ? new Date(annotations.lastRequestTimestamp)
            : null;
        // Set timestamp metadata if missing, and continue
        if (!lastCall) {
            const patch = [
                {
                    op: 'add',
                    path: '/metadata/annotations',
                    value: { lastRequestTimestamp: now.toISOString() },
                },
            ];
            try {
                logger.silly(`Updating metadata of deployment ${name}`);
                await appsApi.patchNamespacedDeployment({ name, namespace, body: patch });
                logger.verbose(`Updated metadata of deployment ${name} successfully `);
            }
            catch (error) {
                handleKubernetesError(error);
            }
            continue;
        }
        // If timestamp exists and is old enough, scale down
        const diff = now.getTime() - lastCall.getTime();
        if (Math.abs(diff) > milliseconds) {
            await scaleDeployment(namespace, name, 0);
        }
    }
}
export async function waitForPodReadiness(namespace, appSelector, maxWait = 10000) {
    let isReady = false;
    const interval = 500;
    let elapsed = 0;
    const { coreApi } = getKubeConfig();
    while (!isReady) {
        const pods = await coreApi.listNamespacedPod({
            namespace,
            labelSelector: `app=${appSelector}`,
        });
        if (pods.items.length > 0) {
            isReady = pods.items.every((pod) => pod.status?.conditions?.some((condition) => condition.type === 'Ready' && condition.status === 'True'));
        }
        if (!isReady) {
            logger.silly(`Waiting for ${appSelector} to be ready ...`);
            await new Promise((resolve) => {
                setTimeout(resolve, interval);
            });
            elapsed += interval;
            if (elapsed >= maxWait) {
                logger.warn(`Deployment ${appSelector} does not respond`);
                return;
            }
        }
    }
}
export async function setLastRequestAnnotation(namespace, deploymentName) {
    const { appsApi } = getKubeConfig();
    const now = new Date();
    const patch = [
        {
            op: 'add',
            path: '/metadata/annotations',
            value: { lastRequestTimestamp: now.toISOString() },
        },
    ];
    try {
        logger.silly(`Setting 'lastRequestTimestamp' annotation of '${deploymentName}' to ${now.toISOString()}`);
        await appsApi.patchNamespacedDeployment({
            name: deploymentName,
            namespace,
            body: patch,
        });
        logger.silly(`'LastRequestTimestamp' annotation of '${deploymentName}' updated successfully`);
    }
    catch (error) {
        logger.warn('Could not set last request annotation:');
        handleKubernetesError(error);
    }
}
//# sourceMappingURL=scale.js.map