分布式mysql使用

import pymysql

# 连接到分片节点
def connect_to_shard(shard_id):
    return pymysql.connect(host=f'shard{shard_id}.example.com', user='user', password='password', db='mydb')

# 分片逻辑
def get_shard_id(key):
    return hash(key) % 3

# 插入数据
def insert_data(key, value):
    shard_id = get_shard_id(key)
    conn = connect_to_shard(shard_id)
    cursor = conn.cursor()
    cursor.execute(f"INSERT INTO mytable (key, value) VALUES ('{key}', '{value}')")
    conn.commit()
    cursor.close()
    conn.close()

# 查询数据
def query_data(key):
    shard_id = get_shard_id(key)
    conn = connect_to_shard(shard_id)
    cursor = conn.cursor()
    cursor.execute(f"SELECT value FROM mytable WHERE key = '{key}'")
    result = cursor.fetchone()
    cursor.close()
    conn.close()
    return result

# 示例调用
insert_data('user1', 'data1')
print(query_data('user1'))