Cách kiểm tra xem ứng dụng Python có đang chạy trong hàm lambda AWS không?

Question

Tôi đã có một ứng dụng Python kết nối với cơ sở dữ liệu và tôi muốn thông tin đăng nhập db khác nhau khi nó chạy trong env địa phương (để thử nghiệm) hoặc trong một hàm lambda (để sản xuất).

Có cách nào, từ ứng dụng Python, để phát hiện ra rằng nó đang chạy bên trong hàm lambda?

Answer 1

Làm cách nào để kiểm tra sự tồn tại của đối tượng context trong hàm xử lý? http://docs.aws.amazon.com/lambda/latest/dg/python-programming-model-handler-types.html

Answer 2

Đối với kiểm tra đơn vị tôi sử dụng cấu trúc:

+ my_function/ 
+- __init__.py - empty files 
+- code/ 
    +- __init__.py 
    +- lambda_function.py 
+- unittest/ 
    +- __init__.py 
    +- tests.py - from ..code.lambda_function import * 

Khi chạy thử nghiệm đơn vị với python -m my_function.unittest.tests, trong lambda_function.py các __name__ == 'my_function.code.lambda_function'.

Khi chạy trong chạy Lambda, __name__ == 'lambda_function'. Lưu ý rằng bạn sẽ nhận được cùng một giá trị nếu bạn chạy với python -m my_function.code.lambda_function vì vậy bạn sẽ luôn cần một trình bao bọc.

Answer 3

Vì this bug, bạn có thể biết liệu bạn có đang chạy bên trong Hàm AWS Lambda hay không.

import multiprocessing 

def on_lambda(): 
    try: 
     multiprocessing.Pool() 
     on_lambda = False 
    except: 
     on_lambda = True 
    return on_lambda 

Tôi đã sử dụng điều này để triển khai thành công báo cáo chỉ số hợp lý. Cho phép hy vọng họ không sửa lỗi này sớm!