基本用法
TIP
ctx.database
并非内置服务,因此如果你的插件需要使用数据库功能,需要声明依赖。
对于几乎所有大型机器人项目,数据库的使用都是不可或缺的。但如果每个插件都独立处理与数据库的交互,这将导致插件之间的兼容性非常差——用户要么选择同时安装多个数据库,要么只能放弃一些功能。为此,Koishi 设计了一整套对象关系映射 (ORM) 接口,它易于扩展并广泛地运用于各种插件中,足以应对绝大部分使用场景。
本页中涉及的完整 API:
get
:查询数据
使用 database.get()
方法以获取特定表中的数据。下面是一个最基本的形式:
// 获取 schedule 表中 id 为 1234 的数据行,返回一个数组
await ctx.database.get('schedule', 1234)
// 获取 schedule 表中 id 为 1234 或 5678 的数据行,返回一个数组
await ctx.database.get('schedule', [1234, 5678])
对于复杂的数据表,如果你只需要获取少数字段,你可以通过第三个参数手动指定要获取的字段:
// 返回的数组中每个元素只会包含 command, time 属性
await ctx.database.get('schedule', [1234], ['command', 'time'])
你还可以向第二个参数传入一个对象,用来查询非主键上的数据或者同时指定多列的值:
// 获取名为 schedule 的表中 assignee 为 123456 或 456789 的数据行
await ctx.database.get('schedule', {
assignee: ['123456', '456789'],
})
对于需要进行复杂的数据库搜索的,你可以使用查询表达式:
// 获取名为 schedule 的表中 id 大于 2 但是小于等于 5 的数据行
await ctx.database.get('schedule', {
id: { $gt: 2, $lte: 5 },
})
为了处理更加复杂的需求,我们还引入了求值表达式。它表现为一个接受 row
参数的回调函数,并通过一系列 $
上的运算返回一个 Boolean
类型的值。这种写法的表达能力更强,例如你可以让多个字段共同参与运算:
import { $ } from 'koishi'
// 上述两个搜索条件的或运算
await ctx.database.get('schedule', (row) =>
$.or(
$.in(row.assignee, [
'123456',
'456789',
]),
$.and(
$.gt(row.id, 2),
$.lte(row.id, 5),
),
),
)
TIP
虽然求值表达式在形式上是一个回调函数,但是 Koishi 并不会将数据全部拉取到内存中,而是会将这个函数的行为编译成相对应的查询语句,提交给数据库运行。因此你可以放心地使用这种写法,它并不会带来额外的性能问题 (如果你遇到查询性能瓶颈,这更有可能是数据模型本身导致的,例如缺少必要的索引)。
create
:插入数据
使用 database.create()
方法以插入数据。
// 向 schedule 表中添加一行数据,data 是要添加的数据行
// 返回值是添加的行的完整数据 (包括自动填充的 id 和默认属性等)
await ctx.database.create('schedule', data)
如果你想要批量插入数据,可以使用下面介绍的 database.upsert()
方法。
set
:修改数据
database.set()
方法需要传入三个参数:表名、查询条件和要修改的数据。
// 第二个参数也可以使用上面介绍的查询表达式
await ctx.database.set('schedule', 1234, {
assignee: 'telegram:123456',
time: new Date(),
})
如果要修改的数据与已有数据相关,可以使用求值表达式:
// 让所有日期为今天的数据行的 count 字段在原有基础上增加 1
await ctx.database.set('foo', { date: new Date() }, (row) => ({
count: $.add(row.count, 1),
}))
upsert
:修改或插入数据
database.upsert()
的逻辑稍微有些不同,需要你传入一个数组:
// 用一个数组来对数据进行更新,你需要确保每一个元素都拥有这个数据表的主键
// 修改时只会用每一行中出现的键进行覆盖,不会影响未定义的字段
await ctx.database.upsert('foo', (row) => [
{ id: 1, foo: 'hello' },
{ id: 2, foo: 'world' },
// 如果此列存在,则会按照 $.concat() 的行为进行修改
// 如果此列不存在,row.bar 则会使用默认值
{ id: 3, bar: $.concat(row.bar, ['koishi']) },
])
如果初始的数据库是这样的:
id | foo | bar |
---|---|---|
(默认值) | null | bar |
1 | foo | baz |
那么进行上述操作后的数据库将是这样的:
id | foo | bar | 说明 |
---|---|---|---|
1 | hello | baz | 该行已经存在,只更新了 foo 字段 |
2 | world | bar | 插入了新行,其中 foo 字段取自传入的数据,bar 字段取自默认值 |
3 | null | barkoishi | 插入了新行,其中 bar 字段取自传入的数据,foo 字段取自默认值 |
如果想以非主键来索引要修改的数据,可以使用第三个参数:
// 以非主键为基准对数据表进行更新,你需要确保每一个元素都拥有 telegram 属性
await ctx.database.upsert('user', rows, 'telegram')
// 以复合键为基准对数据表进行更新,你需要确保每一个元素都拥有 platform 和 id 属性
await ctx.database.upsert('channel', rows, ['platform', 'id'])
remove
:删除数据
使用 database.remove()
方法以删除特定表中的数据。
// 从 schedule 表中删除特定 id 的数据行
// 第二个参数也可以使用上面介绍的查询表达式
await ctx.database.remove('schedule', [id])
获取改动行数
set
, upsert
和 remove
操作都会返回一个 WriteResult
对象,它包含了这次操作的结果。你可以通过 matched
属性来获取匹配的数据行数 (注意不是修改的函数),通过 inserted
属性来获取插入的数据行数 (仅限 upsert
操作)。
// 对某个用户的余额进行扣除
const result = await ctx.database.set(
'user',
{ id, money: { $gte: 100 } },
(row) => ({ money: $.sub(row.money, 100) }),
)
// 如果用户不存在或余额不足,此时 result.matched 为 0
if (!result.matched) {
throw new Error('用户不存在或余额不足!')
}
对比 set 和 upsert
set
与 upsert
方法都可以用于修改已经存在的数据,它们的区别如下表所示:
set | upsert | |
---|---|---|
作用范围 | 支持复杂的查询表达式 | 只能限定特定字段的值 |
插入行为 | 如果数据不存在则不会进行任何操作 | 如果数据不存在则会插入新行 |
对比 create 和 upsert
create
与 upsert
方法都可以用于插入新的数据,它们的区别如下表所示:
create | upsert | |
---|---|---|
插入数量 | 只能插入一条数据 | 可以批量插入多条数据 |
返回值 | 返回经过填充后的数据 | 没有返回值 |
冲突行为 | 如果数据已存在则会报错 | 如果数据已存在则会执行修改 |