堡垒机数据库实时连接

import psycopg2

# 堡垒机连接配置
bastion_host = 'bastion.example.com'
bastion_port = 22
bastion_user = 'bastion_user'
bastion_password = 'bastion_password'

# 数据库连接配置
db_host = 'database.example.com'
db_port = 5432
db_name = 'mydatabase'
db_user = 'db_user'
db_password = 'db_password'

# 通过堡垒机连接到数据库
def connect_to_database():
    # 连接到堡垒机
    bastion_conn = paramiko.SSHClient()
    bastion_conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    bastion_conn.connect(bastion_host, port=bastion_port, username=bastion_user, password=bastion_password)

    # 使用堡垒机作为跳板连接到数据库
    transport = bastion_conn.get_transport()
    dest_addr = (db_host, db_port)
    local_addr = ('localhost', 0)
    channel = transport.open_channel("direct-tcpip", dest_addr, local_addr)

    # 连接到数据库
    conn = psycopg2.connect(host='localhost', port=channel.get_local_port(), dbname=db_name, user=db_user, password=db_password)
    return conn

# 示例查询
def query_database():
    conn = connect_to_database()
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM mytable")
    results = cursor.fetchall()
    for row in results:
        print(row)
    cursor.close()
    conn.close()

if __name__ == "__main__":
    query_database()