這篇文章將為大家詳細講解有關Python如何實現(xiàn)自定義讀寫分離,小編覺得挺實用的,因此分享給大家做個參考,希望大家閱讀完這篇文章后可以有所收獲。
思路
自定義Session類
重寫get_bind方法
根據(jù)self._flushing判斷讀寫操作, 選擇對應的數(shù)據(jù)庫
自定義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ù)庫 print("寫入數(shù)據(jù)") return state.db.get_engine(self.app, bind='master') else: # 讀操作, 使用從數(shù)據(jù)庫 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ù)據(jù)庫的連接地址 app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@192.168.105.134:3306/demo' # 設置數(shù)據(jù)庫的綁定地址 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ù)據(jù)庫變化 一般不會開啟, 影響性能 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False # 設置是否打印底層執(zhí)行的SQL語句 app.config['SQLALCHEMY_ECHO'] = False # 創(chuàng)建數(shù)據(jù)庫連接對象 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)
關于“Python如何實現(xiàn)自定義讀寫分離”這篇文章就分享到這里了,希望以上內容可以對大家有一定的幫助,使各位可以學到更多知識,如果覺得文章不錯,請把它分享出去讓更多的人看到。