Python 实现 Redis ORM
Python 实现 Redis ORM
议题
我们来为 Redis 写一个简单而优雅的 ORM。这篇文章的灵感来自于 Django ORM。
这篇文章假定你对 Redis 以及 Python 中的 redis 库 redis-py 有了基本的了解。
实体
假设我们正在开发一个轮询应用程序,这个应用包括 Question 和 Choice。每一个问题都有多个选项。
我们希望在我们的应用程序中具有以下能力:
- 存储问题列表并检索
- 根据 id 检索问题
- 存储选项
- 关联问题与选项列表
- 将问题与选项取消关联
- 检索一个问题的所有选项
- 跟踪一个选项的投票数
我们希望将 Redis 作为我们的数据库使用。
模型
我们的模型主要就是Question 和 Choice。由于这两个模型与其他模型都有一些相似的功能,因此我们可以创建一个基类 Model,Question 与 Choice 都继承自这个基类 Model。
基础模型
Model 应该是这样的:
类 Question 如下:
我们来为 Model 的不同方法添加实现。
latest_instance_id_key:
list_key:
add_to_list:
add_to_list 使用了 Redis 中的列表将所有实例的 key 存入 redis。我么这里使用 redis-py 中的 lpush 方法,它会映射到 redis 中的 LPUSH 操作。
稍后我们会添加生成 self.id 的代码。
确保 connection 定义在 models.py 模块内是可访问的。
latest_instance_id:
increment_latest_instance_id:
cache_key
save:
注意我们这里在 save() 方法中用的是 self.repr() 方法。它可以调用子类中定义的 repr() 方法,返回一个我们要保存的属性的字典。这个稍后解释。
你需要关注的是 save() 中是怎样调用 increment_latest_instance_id() 和 add_to_list() 的。
save() 方法还用到了 redis-py 中的 hmset 方法,其对应 redis 中的 HMSET 命令。通过这种方式保存实例中我们需要的属性。
子类模型
现在我们来实现类 Question。
由于子类的 save 方法使用到了 repr() 方法,所以其每一个子类都要实现 repr() 方法。repr() 应该返回的是我们希望在 redis 中持久化的字典形式的数据。比如我们希望一个问题的 id 以及 question_text 保存在 redis 中,那么 repr() 返回的字典中应该包含这些属性。
当我们试着使用 save() 保存第一个实例的时候,其内部会调用 latest_instance_id_key() 方法,期望能从 redis 取出已经存在的键 question-latest-id。
我们来创建一个文件 migration 并向 redis 中保存一个键。
运行这个文件:
使用 ORM
通过 ipython 运行我们的代码。
首先确保你的 redis 服务器 redis-server 已经启动了。
现在启动一个 redis 客户端 redis-cli,然后校验一下 question-1 是否已经插入到 redis 中了。
这里同样还校验了 questions 的存在。再来看看 id 有没有保存到 questions 中。
现在再来校验一下 id 和 question_text 是否也被保存在 questions 中。
我们再来用 ipython 创建一个 question 并将其保存在 redis 中。
现在看看我们期望的键值是否都保存到 redis 中了。
提取问题
我们添加一个 Question 方法来根据给出的 id 提取一个问题。
注意这里我们传入了 id 作为 cache_key。相应地我们就需要修改 cache_key 的代码:
现在来使用之前添加的 questions 的 ids 来提取它们。
不过需要确保 save 功能可用。
非常棒!
提取所有问题
我们再添加一个方法来提取所有问题。
现在使用这个方法提取问题。
get_question 和 get_questions 拥有相似的功能并且可以使用我们稍后添加的模型。所以,现在将它们移到基类中去。将 get_question 重命名为 get,将 get_questions 重命名为 list。
现在就类用一下 get() 和 list() 方法,并校验一下其功能是否符合预期。
其他子类模型
现在来添加一个 Choice 模型。
在保存 choice 模型之前,我们需要添加一个键 choice-latest-id。还记得我们之前的 migrations.py 文件吗?
在 ipython 中执行。
现在我们通过 Choice 来演示 ORM 操作。
这应该归功于我们的基类。通过 Choice 我们可以自由地使用 save(),get(),list() 等方法。
英文原文:https://www.agiliq.com/blog/2019/11/writing-an-orm-for-redis/
译者:居老师的龙尾巴