mysql手动实现读写分离

import pymysql

# 数据库配置
master_config = {
    'host': 'master_host',
    'user': 'user',
    'password': 'password',
    'database': 'database'
}

slave_config = {
    'host': 'slave_host',
    'user': 'user',
    'password': 'password',
    'database': 'database'
}

def get_connection(sql_type):
    if sql_type.upper().startswith('SELECT'):
        return pymysql.connect(**slave_config)
    else:
        return pymysql.connect(**master_config)

def execute_query(sql, sql_type):
    conn = get_connection(sql_type)
    cursor = conn.cursor()
    cursor.execute(sql)
    result = cursor.fetchall()
    cursor.close()
    conn.close()
    return result

# 示例查询
select_query = "SELECT * FROM table_name"
insert_query = "INSERT INTO table_name (column1, column2) VALUES ('value1', 'value2')"

print(execute_query(select_query, 'SELECT'))
print(execute_query(insert_query, 'INSERT'))