Python如何實現自定義讀寫分離

這篇文章將為大家詳細講解有關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如何實現自定義讀寫分離”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。