Skip to content

Instantly share code, notes, and snippets.

@asbubam

asbubam/check_lambda_provision.js

Last active March 7, 2021 05:47
Show Gist options
  • Save asbubam/c93fc504bccb1d74319862023a22e354 to your computer and use it in GitHub Desktop.
Save asbubam/c93fc504bccb1d74319862023a22e354 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. asbubam revised this gist Mar 7, 2021. No changes.
  2. asbubam created this gist Mar 6, 2021.
    67 changes: 67 additions & 0 deletions check_lambda_provision.js
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,67 @@
    const AWS = require('aws-sdk');
    const lambda = new AWS.Lambda({ region: 'ap-northeast-2' });

    const sleep = async (ms) => { await new Promise(resolve => setTimeout(resolve, ms)); };

    const getFuncList = async () => {
    let funcList = [];
    let isStop = false;
    let params = {}

    while (!isStop) {
    await sleep(500);
    let data = await lambda.listFunctions(params).promise();
    funcList.push(...data.Functions);

    params.Marker = data.NextMarker;
    if (!data.NextMarker) isStop = true;
    }

    return funcList;
    };


    const getProvisionConfig = async (funcName) => {
    let data = await lambda.listProvisionedConcurrencyConfigs({ FunctionName: funcName }).promise();

    let config;
    if (data.ProvisionedConcurrencyConfigs.length > 0) {
    config = data.ProvisionedConcurrencyConfigs[0];
    console.log(`${funcName}, [${config.AllocatedProvisionedConcurrentExecutions} / ${config.RequestedProvisionedConcurrentExecutions}]`);

    if (data.ProvisionedConcurrencyConfigs.length > 1) {
    console.log(data.ProvisionedConcurrencyConfigs);
    }
    }

    return config? config.AllocatedProvisionedConcurrentExecutions : 0;
    };

    const getReservedConfig = async (funcName) => {
    let data = await lambda.getFunctionConcurrency({ FunctionName: funcName }).promise();

    if (data.ReservedConcurrentExecutions) {
    console.log(`${funcName}, ${data.ReservedConcurrentExecutions}`);
    }
    };


    (async () => {
    let funcList= await getFuncList();
    console.log(`total lambda counts, ${funcList.length}`);

    console.log('>>> provisioned');
    let totalCount = 0;
    for (let i = 0; i < funcList.length; i++) {
    await sleep(100);
    func = funcList[i];
    totalCount += await getProvisionConfig(func.FunctionName);
    }
    console.log(totalCount);

    console.log('>>> reserved');
    for (let i = 0; i < funcList.length; i++) {
    func = funcList[i];
    await getReservedConfig(func.FunctionName);
    }
    })();