SqlAlchemy 中操做数据库时session和scoped_session的区别(源码分析)

原生session:

from sqlalchemy.orm import sessionmaker from sqlalchemy import create_engine from sqlalchemy应用.models import Users engine = create_engine( "mysql+pymysql://root:root@127.0.0.1:3306/pro6?charset=utf8", max_overflow=0,  # 超过链接池大小外最多建立的链接
    pool_size=5,  # 链接池大小
) #from sqlalchemy.orm.session import Session
 SessionF = sessionmaker(bind=engine) session = SessionF() print(session) obj1 = Users(name='ctz', email='49274573@qq.com', extra='aaaa') session.add(obj1) session.commit() session.close()

问题：因为没法提供线程共享功能，因此在开发时要注意,要给每一个线程都建立本身的session

打印sesion可知他是sqlalchemy.orm.session.Session的对象

查看Session的源码 可获得:

class Session(_SessionClassMethods): """Manages persistence operations for ORM-mapped objects. The Session's usage paradigm is described at :doc:`/orm/session`. """ public_methods = ( '__contains__', '__iter__', 'add', 'add_all', 'begin', 'begin_nested', 'close', 'commit', 'connection', 'delete', 'execute', 'expire', 'expire_all', 'expunge', 'expunge_all', 'flush', 'get_bind', 'is_modified', 'bulk_save_objects', 'bulk_insert_mappings', 'bulk_update_mappings', 'merge', 'query', 'refresh', 'rollback', 'scalar')

2.scoped_session

from sqlalchemy.orm import sessionmaker from sqlalchemy.ext.declarative import declarative_base from sqlalchemy import create_engine from sqlalchemy应用.models import Users from sqlalchemy.orm import scoped_session engine=create_engine( "mysql+pymysql://root:root@127.0.0.1:3306/pro6?charset=utf8", max_overflow=0,  # 超过链接池大小外最多建立的链接
    pool_size=5,  # 链接池大小
)  SessionF=sessionmaker(bind=engine) #scoped_session封装了两个值 Session 和 registry,registry加括号就执行了ThreadLocalRegistry的__call__方法,若是

# 当前本地线程中有session就返回session,没有就将session添加到了本地线程

#self.registry()=session
session=scoped_session(SessionF) print(session) obj1=Users(name='ctz',email='49274573@qq.com',extra='aaaa') session.remove() session.query_property()

优势:支持线程安全,为每一个线程都建立一个session:

 两种方式:经过本地线程Threading.Local()和建立惟一标识的方法(参考flask请求源码)

源码分析：

首先咱们在scoped_session中放入了Sesion对象

SessionF=sessionmaker(bind=engine)

session=scoped_session(Session)

1、scoped_session类中

class scoped_session(object): session_factory = None def __init__(self, session_factory, scopefunc=None): #传递过来的那个Session对象
        self.session_factory = session_factory #scopefunc惟一标示函数
        if scopefunc: self.registry = ScopedRegistry(session_factory, scopefunc) else: self.registry = ThreadLocalRegistry(session_factory)

第一次进来scopefunc惟一标识为None，咱们将Session做为参数传递到ThreadLocalRegistry中，

ThreadLocalRegistry类中

class ThreadLocalRegistry(ScopedRegistry): """A :class:`.ScopedRegistry` that uses a ``threading.local()`` variable for storage. """

    def __init__(self, createfunc): #传递过来的那个Session对象
        self.createfunc = createfunc self.registry = threading.local() #scoped_session.registry()后执行 def __call__(self): try:
 #若是本地线程中有值的话直接将值返回， return self.registry.value except AttributeError:
 #没有值的话就示例话Session(),并将他存到本地线程中，并把实例的对象返回 #至关于Session()后的对象加到了本地线程中
            val = self.registry.value = self.createfunc() return val

其中__call__()只有当scoped_session.registry加括号执行

那咱们怎么调用那些方法呢？

def instrument(name): def do(self, *args, **kwargs): return getattr(self.registry(), name)(*args, **kwargs) return do for meth in Session.public_methods: setattr(scoped_session, meth, instrument(meth))

其中 Session就是sqlalchemy.orm.session.Session

public_methods = ( '__contains__', '__iter__', 'add', 'add_all', 'begin', 'begin_nested', 'close', 'commit', 'connection', 'delete', 'execute', 'expire', 'expire_all', 'expunge', 'expunge_all', 'flush', 'get_bind', 'is_modified', 'bulk_save_objects', 'bulk_insert_mappings', 'bulk_update_mappings', 'merge', 'query', 'refresh', 'rollback', 'scalar')

在instrument函数中 

self.registry()帮咱们执行了ThreadLocalRegistry中的___call__方法，拿到了sesion对象


方法源码示例:

def has(self): return hasattr(self.registry, "value")

def remove(self): if self.registry.has(): self.registry().close() self.registry.clear()

实际两种方式原理都是同样的都是第一种，只是第二种将session放到了本地线程中，为每个进程都设置了一个session，实现了线程安全