Python如何實(shí)現(xiàn)自定義讀寫分離

這篇文章將為大家詳細(xì)講解有關(guān)Python如何實(shí)現(xiàn)自定義讀寫分離，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

思路

• 自定義Session類

• 重寫get_bind方法

• 根據(jù)self._flushing判斷讀寫操作, 選擇對(duì)應(yīng)的數(shù)據(jù)庫(kù)

• 自定義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: # 寫操作 ,使用主數(shù)據(jù)庫(kù)
      print("寫入數(shù)據(jù)")
      return state.db.get_engine(self.app, bind='master')
    else: # 讀操作, 使用從數(shù)據(jù)庫(kù)
      print('讀取數(shù)據(jù)')
      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__)
# 設(shè)置數(shù)據(jù)庫(kù)的連接地址
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@192.168.105.134:3306/demo'
# 設(shè)置數(shù)據(jù)庫(kù)的綁定地址
app.config['SQLALCHEMY_BINDS'] = {
  'master': "mysql://root:mysql@192.168.105.134:3306/demo",
  'slave': "mysql://root:mysql@192.168.105.134:8306/demo"
}
# 設(shè)置是否追蹤數(shù)據(jù)庫(kù)變化  一般不會(huì)開啟, 影響性能
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 設(shè)置是否打印底層執(zhí)行的SQL語(yǔ)句
app.config['SQLALCHEMY_ECHO'] = False

# 創(chuàng)建數(shù)據(jù)庫(kù)連接對(duì)象
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():
  # 增加數(shù)據(jù)
  user1 = User(name='zs')
  db.session.add(user1)
  db.session.commit()

  # 查詢數(shù)據(jù)
  users = User.query.all()
  print(users)
  return "index"

if __name__ == '__main__':
  # 刪除所有繼承自db.Model的表
  db.drop_all()
  # 創(chuàng)建所有繼承自db.Model的表
  db.create_all()
  app.run(debug=True)

關(guān)于“Python如何實(shí)現(xiàn)自定義讀寫分離”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。