測試 Postgres-DB 的連接

Question

如果軟件連接到特定的 Postgre-DB，我想在 GUI 上放一個按鈕。 我寫了一個小測試函數：如果它可以連接到數據庫，則返回 True，否則返回 False。

代碼有效，但有一個問題：如果沒有連接（我只是拔掉網線，沒有其他變化），它只是需要太多時間。 如果沒有連接，你能幫我加快代碼速度嗎？

這是我的簡單測試功能：

import psycopg2

def postgres_test():

    try:
        conn = psycopg2.connect("dbname='mydb' user='myuser' host='my_ip' password='mypassword'")
        conn.close()
        return True
    except:
        return False

Answer 1

感謝您的評論。 是的，這與超時有關。

這是我的更快的代碼：

import psycopg2

def postgres_test():

    try:
        conn = psycopg2.connect("dbname='mydb' user='myuser' host='my_ip' password='mypassword' connect_timeout=1 ")
        conn.close()
        return True
    except:
        return False