AWS Lambda 深度解析#

import time

# 这些代码在 Init 阶段执行
print("=== Init 阶段开始 ===")
init_start = time.time()

# 模拟依赖加载（约 100-300ms）
import boto3  # 约 50-100ms
import numpy as np  # 约 50-150ms
import pandas as pd  # 约 100-200ms

# 模拟业务初始化（约 50-100ms）
config = load_config()  # 从 SSM/Secrets Manager 加载
client = boto3.client('s3')

init_end = time.time()
print(f"=== Init 阶段结束，耗时: {(init_end - init_start) * 1000:.2f}ms ===")

def lambda_handler(event, context):
    # Handler 代码在 Invoke 阶段执行
    print(f"Handler 执行，当前环境 ID: {context.aws_request_id[:8]}")
    return {'statusCode': 200}

import os
import json

def lambda_handler(event, context):
    # AWS 提供的基础环境变量
    region = os.environ['AWS_REGION']
    function_name = os.environ['AWS_LAMBDA_FUNCTION_NAME']
    memory = os.environ['AWS_LAMBDA_FUNCTION_MEMORY_SIZE']

    # 自定义环境变量（在 Lambda 配置中设置）
    db_endpoint = os.environ['DB_ENDPOINT']
    cache_ttl = int(os.environ.get('CACHE_TTL', '300'))

    # Secrets Manager 密文（通过 KMS 加密）
    api_key = os.environ['API_KEY']

    return {
        'region': region,
        'function_name': function_name,
        'memory': memory
    }

import json

def handler(event):
    """自定义处理器"""
    action = event.get('action', 'unknown')

    if action == 'process':
        return {
            'statusCode': 200,
            'body': json.dumps({
                'message': f'Processed: {event.get("data")}'
            })
        }
    elif action == 'health':
        return {
            'statusCode': 200,
            'body': json.dumps({'status': 'healthy'})
        }
    else:
        return {
            'statusCode': 400,
            'body': json.dumps({'error': 'Unknown action'})
        }

import os

def lambda_handler(event, context):
    # Layer 中的包可以直接 import
    import boto3
    import pandas as pd

    # Layer 中的脚本可以直接执行
    # 假设 Layer 中有 /opt/utils/logging.py
    # 可以在代码中引用

    # 查看 Layer 路径
    opt_path = os.environ.get('LAMBDA_TASK_ROOT', '/opt')
    print(f"Layer 路径: {opt_path}")

    # 加载 Layer 中的自定义模块
    import sys
    sys.path.insert(0, '/opt/utils')
    from custom_utils import process_data

    return {'statusCode': 200}

import boto3

lambda_client = boto3.client('lambda')

def configure_vpc(function_name: str, subnet_ids: list, security_group_ids: list):
    """配置 Lambda 函数访问 VPC"""

    try:
        response = lambda_client.update_function_configuration(
            FunctionName=function_name,
            VpcConfig={
                'SubnetIds': subnet_ids,
                'SecurityGroupIds': security_group_ids
            }
        )

        print(f"VPC 配置成功:")
        print(f"  子网: {subnet_ids}")
        print(f"  安全组: {security_group_ids}")

        # VPC 配置需要几分钟生效
        print("注意: VPC 配置可能需要几分钟生效，期间函数可能无法访问 VPC 资源")

    except lambda_client.exceptions.InvalidParameterValueException as e:
        print(f"参数无效: {e}")

import os

def lambda_handler(event, context):
    # 检查当前配额
    remaining_time = context.get_remaining_time_in_millis()
    max_memory = os.environ.get('AWS_LAMBDA_FUNCTION_MEMORY_SIZE')

    # 处理大数据集
    if event.get('large_payload'):
        # 超出 Payload 限制时的处理
        return handle_large_payload_via_s3(event)

    # 处理长时间任务
    if event.get('long_running'):
        # 分片处理，避免超时
        return handle_long_task分期(event, context)

    # 内存敏感操作
    if event.get('memory_intensive'):
        # 流式处理，避免内存溢出
        return handle_memory_intensive(event)

    return {'statusCode': 200}

def handle_large_payload_via_s3(event):
    """通过 S3 处理大 Payload"""
    s3_key = event['s3_key']
    # 从 S3 读取数据，而不是从 event 中
    return {'processed': True}

def handle_long_task分期(event, context):
    """分片处理长时间任务"""
    # 检查剩余时间
    remaining = context.get_remaining_time_in_millis()

    if remaining < 60000:  # 少于 1 分钟
        # 保存检查点，下次继续
        save_checkpoint(event)
        return {'status': 'checkpoint', 'resume': True}

    # 正常处理
    result = process_batch(event)

    # 如果还没完成，保存检查点
    if not is_complete(result):
        save_checkpoint(event)
        return {'status': 'partial', 'resume': True}

    return {'status': 'complete'}