Python Flask-SQLAlchemy 实现自定义读写分离方案

清欢

发布时间：2026-10-12| 3人浏览过

原创

本文介绍如何在 Flask-SQLAlchemy 中通过重写 get_bind 和 create_session 方法实现自定义读写分离。核心思路是根据 self._flushing 状态区分读写操作，分别路由至主库和从库。通过自定义 RoutingSession 和 RoutingSQLAlchemy 类，开发者可灵活控制数据库连接策略，但需注意自动选择的局限性。

本文详细讲解了在 Python 中实现自定义读写分离的具体代码示例。文中提供的示例代码详尽且具备较高的参考价值，适合希望优化数据库性能或进行架构调整的开发者参考学习。

实现思路

具体实现中，需要定义自定义 Session 类。
- 核心在于重写 get_bind 方法。
- 通过检查 self._flushing 属性来判定当前是读操作还是写操作，进而动态选择对应的数据库连接。
同时需定义自定义 SQLAlchemy 类以集成上述逻辑。
- 同时需要重写 create_session 方法，在其中实例化并使用自定义的 Session 类，以接管会话创建过程。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy, SignallingSession, get_state
from sqlalchemy import orm


class RoutingSession(SignallingSession):
  def get_bind(self, mapper=None, clause=None):
    state = get_state(self.app)

    # 判断读写操作
    if self._flushing: # 写操作 ,使用主数据库
      print("写入数据")
      return state.db.get_engine(self.app, bind='master')
    else: # 读操作, 使用从数据库
      print('读取数据')
      return state.db.get_engine(self.app, bind='slave')


class RoutingSQLAlchemy(SQLAlchemy):
  def create_session(self, options):
    return orm.sessionmaker(class_=RoutingSession, db=self, **options)


app = Flask(__name__)
# 设置数据库的连接地址
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@192.168.105.134:3306/demo'
# 设置数据库的绑定地址
app.config['SQLALCHEMY_BINDS'] = {
  'master': "mysql://root:mysql@192.168.105.134:3306/demo",
  'slave': "mysql://root:mysql@192.168.105.134:8306/demo"
}
# 设置是否追踪数据库变化  一般不会开启, 影响性能
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 设置是否打印底层执行的SQL语句
app.config['SQLALCHEMY_ECHO'] = False

# 创建数据库连接对象
db = RoutingSQLAlchemy(app)

# 用户表 一
class User(db.Model):
  __tablename__ = 't_user'
  id = db.Column(db.Integer, primary_key=True)
  name = db.Column(db.String(20), unique=True)


@app.route('/')
def index():
  # 增加数据
  user1 = User(name='zs')
  db.session.add(user1)
  db.session.commit()

  # 查询数据
  users = User.query.all()
  print(users)
  return "index"

if __name__ == '__main__':
  # 删除所有继承自db.Model的表
  db.drop_all()
  # 创建所有继承自db.Model的表
  db.create_all()
  app.run(debug=True)

这种基于内部状态自动路由的方式存在一定局限，即开发者无法在应用层直接控制具体的数据库选择，灵活性稍显不足。

以上内容即为本文的全部介绍，希望能对读者的学习和工作有所助益，同时也欢迎大家继续支持本站获取更多技术资源。