尊龙凯时网址

python如何实现自定义读写分离 -尊龙凯时网址

2024-01-07

这篇文章将为大家详细讲解有关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)

关于“python如何实现自定义读写分离”这篇文章就分享到这里了,希望以上内容可以对大家有一定的帮助,使各位可以学到更多知识,如果觉得文章不错,请把它分享出去让更多的人看到。

网站地图