搭建mysql分布式数据库

import pymysql

# 配置分片规则
shard_config = {
    'user_shard_1': {'host': 'shard1_host', 'port': 3306, 'user': 'user', 'password': 'password'},
    'user_shard_2': {'host': 'shard2_host', 'port': 3306, 'user': 'user', 'password': 'password'}
}

def get_shard_connection(user_id):
    shard_key = 'user_shard_' + str(user_id % 2 + 1)
    return pymysql.connect(**shard_config[shard_key])

# 插入数据示例
def insert_user(user_id, name):
    conn = get_shard_connection(user_id)
    cursor = conn.cursor()
    sql = "INSERT INTO users (user_id, name) VALUES (%s, %s)"
    cursor.execute(sql, (user_id, name))
    conn.commit()
    cursor.close()
    conn.close()

# 查询数据示例
def get_user(user_id):
    conn = get_shard_connection(user_id)
    cursor = conn.cursor()
    sql = "SELECT * FROM users WHERE user_id = %s"
    cursor.execute(sql, (user_id,))
    result = cursor.fetchone()
    cursor.close()
    conn.close()
    return result

# 示例调用
insert_user(1, 'Alice')
print(get_user(1))