用 memcached 实现 Tornado 的 session 支持(三)
童鞋,我就知道你是个好学滴好孩子~来吧,让我们进行最后的探(zuo)索(si)!
上一次我们讲到哪里?哦。。。准备讲 SessionManager 是吧,来~一个一个函数看~
首先是初始化,设置密钥, memcache 地址,session 超时时间。
# 初始化需要一个用于 session 加密的 secret, memcache 地址, session 的过期时间 def __init__(self, secret, memcached_address, session_timeout): self.secret = secret self.memcached_address = memcached_address self.session_timeout = session_timeout接着是 _fetch 方法,以 session_id 为键从 memcached 中取出数据,并用 pickle 反序列化解析数据:
# 该方法用 session_id 从 memcache 中取出数据 def _fetch(self, session_id): try: # 连接 memcache 服务器 mc = memcache.Client(self.memcached_address, debug=0) # 获取数据 session_data = raw_data = mc.get(session_id) if raw_data != None: # 为了重新刷新 timeout mc.replace(session_id, raw_data, self.session_timeout, 0) # 反序列化 session_data = pickle.loads(raw_data) # 如果拿到的数据是字典形式,才进行返回 if type(session_data) == type({}): return session_data else: return {} except IOError: return {}get 经过安全检查后,以 SessionData 的形式返回 memcached 的数据(调用了 _fetch)方法。
def get(self, request_handler = None): # 获取对应的 session_id 和 hmac_key if (request_handler == None): session_id = None hmac_key = None else: # session 的基础还是靠 cookie session_id = request_handler.get_secure_cookie("session_id") hmac_key = request_handler.get_secure_cookie("verification") # session_id 不存在的时候则生成一个新的 session_id 和 hmac_key if session_id == None: session_exists = False session_id = self._generate_id() hmac_key = self._generate_hmac(session_id) else: session_exists = True # 检查 hmac_key check_hmac = self._generate_hmac(session_id) # 不通过则抛出异常 if hmac_key != check_hmac: raise InvalidSessionException() # 新建 SessionData 对象 session = SessionData(session_id, hmac_key) if session_exists: # 通过 _fetch 方法获取 memcache 中该 session 的所有数据 session_data = self._fetch(session_id) for key, data in session_data.iteritems(): session[key] = data return session至于 set 方法,是为了更新 memcached 的数据。
# 设置新的 session,需要设置 handler 的 cookie 和 memcache 客户端 def set(self, request_handler, session): # 设置浏览器的 cookie request_handler.set_secure_cookie("session_id", session.session_id) request_handler.set_secure_cookie("verification", session.hmac_key) # 用 pickle 进行序列化 session_data = pickle.dumps(dict(session.items()), pickle.HIGHEST_PROTOCOL) # 连接 memcache 服务器 mc = memcache.Client(self.memcached_address, debug=0) # 写入 memcache mc.set(session.session_id, session_data, self.session_timeout, 0)最后的两个函数,一个是生成 session_id,另一个用 session_id 与密钥加密后生成一个加密字符串,用于验证。
# 生成 session_id def _generate_id(self): new_id = hashlib.sha256(self.secret + str(uuid.uuid4())) return new_id.hexdigest() # 生成 hmac_key def _generate_hmac(self, session_id): return hmac.new(session_id, self.secret, hashlib.sha256).hexdigest()
我们在哪里初始化了 SessionManager 呢?还记得第一篇里面的 Application 类吗?噢...快回去翻翻。
好了,童鞋们我们放学了,大家回家吧~对本篇博客有任何意见,拒绝吐槽!
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。