PowerShellでAWS Lambdaの並列実行数を制御しリソース枯渇を防ぐ方法

Install-Module -Name AWSPowerShell -Scope CurrentUser

Set-AWSCredential -AccessKey "YourAccessKey" -SecretKey "YourSecretKey" -Region "YourRegion"

Get-LMFunction

New-LMFunction -FunctionName "MyFunction" -Runtime "nodejs18.x" -Role "YourRoleARN" -Handler "index.handler" -Code S3Bucket="YourBucket",S3Key="YourCode.zip"

Update-LMFunctionConfiguration -FunctionName "MyFunction" -ReservedConcurrentExecutions 10

$Error[0] | Format-List * -Force

# AWS認証情報の設定
$AccessKey = "YourAccessKey"
$SecretKey = "YourSecretKey"
$Region = "YourRegion"
Set-AWSCredential -AccessKey $AccessKey -SecretKey $SecretKey -Region $Region

# 関数名と並列実行数の設定
$FunctionName = "MyLambdaFunction"
$MaxConcurrentExecutions = 5

# 並列実行数の設定
try {
    Write-Host "Setting reserved concurrent executions for $FunctionName..."
    Update-LMFunctionConfiguration -FunctionName $FunctionName -ReservedConcurrentExecutions $MaxConcurrentExecutions
    Write-Host "Successfully set $MaxConcurrentExecutions concurrent executions for $FunctionName."
} catch {
    Write-Error "Failed to set concurrent executions: $_"
}

# 実行状態の確認
try {
    Write-Host "Fetching function configuration for $FunctionName..."
    $FunctionConfig = Get-LMFunction -FunctionName $FunctionName
    Write-Host "Function Name: $($FunctionConfig.FunctionName)"
    Write-Host "Reserved Concurrent Executions: $($FunctionConfig.ReservedConcurrentExecutions)"
} catch {
    Write-Error "Failed to retrieve function configuration: $_"
}

# 実行中の関数のモニタリング（オプション）
try {
    Write-Host "Monitoring invocation metrics..."
    $Metrics = Get-CWMetricStatistics -Namespace "AWS/Lambda" -MetricName "ConcurrentExecutions" `
        -Dimensions @(@{Name="FunctionName";Value=$FunctionName}) `
        -StartTime (Get-Date).AddMinutes(-10) -EndTime (Get-Date) `
        -Period 60 -Statistics "Average"
    Write-Host "Recent invocation metrics:"
    $Metrics.Datapoints | ForEach-Object {
        Write-Host "Timestamp: $($_.Timestamp), Average Concurrent Executions: $($_.Average)"
    }
} catch {
    Write-Error "Failed to fetch invocation metrics: $_"
}

# CloudWatchメトリクスの取得
$FunctionName = "MyLambdaFunction"

try {
    Write-Host "Fetching invocation metrics for $FunctionName..."
    $Metrics = Get-CWMetricStatistics -Namespace "AWS/Lambda" -MetricName "ConcurrentExecutions" `
        -Dimensions @(@{Name="FunctionName";Value=$FunctionName}) `
        -StartTime (Get-Date).AddMinutes(-10) -EndTime (Get-Date) `
        -Period 60 -Statistics "Average"
    Write-Host "Recent invocation metrics:"
    $Metrics.Datapoints | ForEach-Object {
        Write-Host "Timestamp: $($_.Timestamp), Average Concurrent Executions: $($_.Average)"
    }
} catch {
    Write-Error "Failed to fetch invocation metrics: $_"
}

# スロットリングメトリクスの取得
try {
    Write-Host "Fetching throttles metrics for $FunctionName..."
    $ThrottlesMetrics = Get-CWMetricStatistics -Namespace "AWS/Lambda" -MetricName "Throttles" `
        -Dimensions @(@{Name="FunctionName";Value=$FunctionName}) `
        -StartTime (Get-Date).AddMinutes(-10) -EndTime (Get-Date) `
        -Period 60 -Statistics "Sum"
    Write-Host "Recent throttle metrics:"
    $ThrottlesMetrics.Datapoints | ForEach-Object {
        Write-Host "Timestamp: $($_.Timestamp), Throttles: $($_.Sum)"
    }
} catch {
    Write-Error "Failed to fetch throttle metrics: $_"
}

# Lambda関数を呼び出す関数
function Invoke-LambdaFunctionWithRetry {
    param (
        [string]$FunctionName,
        [int]$MaxRetries = 3,
        [int]$RetryDelaySeconds = 5
    )

    $RetryCount = 0
    while ($RetryCount -lt $MaxRetries) {
        try {
            # Lambda関数の呼び出し
            Write-Host "Invoking Lambda function: $FunctionName (Attempt: $($RetryCount + 1))"
            $Response = Invoke-LMFunction -FunctionName $FunctionName -Payload '{"key":"value"}'
            Write-Host "Invocation successful: $($Response.LogResult)"
            return $Response
        } catch {
            Write-Error "Error invoking Lambda function: $_"
            $RetryCount++
            if ($RetryCount -ge $MaxRetries) {
                throw "Max retry attempts reached for function $FunctionName."
            }
            Write-Host "Retrying in $RetryDelaySeconds seconds..."
            Start-Sleep -Seconds $RetryDelaySeconds
        }
    }
}

$RetryDelaySeconds = 5 * [math]::Pow(2, $RetryCount)

# 関数名と並列実行数の設定リスト
$FunctionsConfig = @(
    @{Name = "FunctionA"; ReservedConcurrentExecutions = 10},
    @{Name = "FunctionB"; ReservedConcurrentExecutions = 5},
    @{Name = "FunctionC"; ReservedConcurrentExecutions = 8}
)

foreach ($Function in $FunctionsConfig) {
    try {
        $FunctionName = $Function.Name
        $ReservedConcurrentExecutions = $Function.ReservedConcurrentExecutions

        # 並列実行数の設定
        Write-Host "Setting reserved concurrent executions for $FunctionName..."
        Update-LMFunctionConfiguration -FunctionName $FunctionName -ReservedConcurrentExecutions $ReservedConcurrentExecutions
        Write-Host "Successfully set $ReservedConcurrentExecutions concurrent executions for $FunctionName."
    } catch {
        Write-Error "Failed to set concurrent executions for $FunctionName: $_"
    }
}

# 各関数の設定を確認
foreach ($Function in $FunctionsConfig) {
    try {
        $FunctionName = $Function.Name

        # 関数の状態確認
        Write-Host "Fetching configuration for $FunctionName..."
        $FunctionConfig = Get-LMFunction -FunctionName $FunctionName
        Write-Host "Function Name: $($FunctionConfig.FunctionName)"
        Write-Host "Reserved Concurrent Executions: $($FunctionConfig.ReservedConcurrentExecutions)"
    } catch {
        Write-Error "Failed to fetch configuration for $FunctionName: $_"
    }
}