Có một thư viện ưa thích để viết điều kiện kiểm tra điểm cuối của dịch vụ - kiểm tra sức khỏe .
Bạn có thể sử dụng nó thay vì tạo điểm cuối theo cách thủ công vì có một số tính năng không phù hợp (ví dụ: EnvironmentDump ).
Trong ứng dụng của mình, tôi cũng có nhu cầu như vậy nên tôi đã triển khai kiểm tra xem cơ sở dữ liệu có đáp ứng không
app = Flask(__name__)
# wrap the flask app and give a heathcheck url
health = HealthCheck(app, "/healthcheck")
def health_database_status():
is_database_working = True
output = 'database is ok'
try:
# to check database we will execute raw query
session = DatabaseSession.get_database_session()
session.execute('SELECT 1')
except Exception as e:
output = str(e)
is_database_working = False
return is_database_working, output
health.add_check(health_database_status)
Như tôi thấy, trong ứng dụng của bạn, bạn có thể thực thi truy vấn với db.engine.execute('SELECT 1') .
