Skip to content

基本用法

TIP

ctx.database 并非内置服务,因此如果你的插件需要使用数据库功能,需要声明依赖

对于几乎所有大型机器人项目,数据库的使用都是不可或缺的。但如果每个插件都独立处理与数据库的交互,这将导致插件之间的兼容性非常差——用户要么选择同时安装多个数据库,要么只能放弃一些功能。为此,Koishi 设计了一整套对象关系映射 (ORM) 接口,它易于扩展并广泛地运用于各种插件中,足以应对绝大部分使用场景。

本页中涉及的完整 API:

get:查询数据

使用 database.get() 方法以获取特定表中的数据。下面是一个最基本的形式:

ts
// 获取 schedule 表中 id 为 1234 的数据行,返回一个数组
await ctx.database.get('schedule', 1234)

// 获取 schedule 表中 id 为 1234 或 5678 的数据行,返回一个数组
await ctx.database.get('schedule', [1234, 5678])

对于复杂的数据表,如果你只需要获取少数字段,你可以通过第三个参数手动指定要获取的字段:

ts
// 返回的数组中每个元素只会包含 command, time 属性
await ctx.database.get('schedule', [1234], ['command', 'time'])

你还可以向第二个参数传入一个对象,用来查询非主键上的数据或者同时指定多列的值:

ts
// 获取名为 schedule 的表中 assignee 为 123456 或 456789 的数据行
await ctx.database.get('schedule', {
  assignee: ['123456', '456789'],
})

对于需要进行复杂的数据库搜索的,你可以使用查询表达式:

ts
// 获取名为 schedule 的表中 id 大于 2 但是小于等于 5 的数据行
await ctx.database.get('schedule', {
  id: { $gt: 2, $lte: 5 },
})

为了处理更加复杂的需求,我们还引入了求值表达式。它表现为一个接受 row 参数的回调函数,并通过一系列 $ 上的运算返回一个 Boolean 类型的值。这种写法的表达能力更强,例如你可以让多个字段共同参与运算:

ts
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() 方法以插入数据。

ts
// 向 schedule 表中添加一行数据,data 是要添加的数据行
// 返回值是添加的行的完整数据 (包括自动填充的 id 和默认属性等)
await ctx.database.create('schedule', data)

如果你想要批量插入数据,可以使用下面介绍的 database.upsert() 方法。

set:修改数据

database.set() 方法需要传入三个参数:表名、查询条件和要修改的数据。

ts
// 第二个参数也可以使用上面介绍的查询表达式
await ctx.database.set('schedule', 1234, {
  assignee: 'telegram:123456',
  time: new Date(),
})

如果要修改的数据与已有数据相关,可以使用求值表达式:

ts
// 让所有日期为今天的数据行的 count 字段在原有基础上增加 1
await ctx.database.set('foo', { date: new Date() }, (row) => ({
  count: $.add(row.count, 1),
}))

upsert:修改或插入数据

database.upsert() 的逻辑稍微有些不同,需要你传入一个数组:

ts
// 用一个数组来对数据进行更新,你需要确保每一个元素都拥有这个数据表的主键
// 修改时只会用每一行中出现的键进行覆盖,不会影响未定义的字段
await ctx.database.upsert('foo', (row) => [
  { id: 1, foo: 'hello' },
  { id: 2, foo: 'world' },
  // 如果此列存在,则会按照 $.concat() 的行为进行修改
  // 如果此列不存在,row.bar 则会使用默认值
  { id: 3, bar: $.concat(row.bar, ['koishi']) },
])

如果初始的数据库是这样的:

idfoobar
(默认值)nullbar
1foobaz

那么进行上述操作后的数据库将是这样的:

idfoobar说明
1hellobaz该行已经存在,只更新了 foo 字段
2worldbar插入了新行,其中 foo 字段取自传入的数据,bar 字段取自默认值
3nullbarkoishi插入了新行,其中 bar 字段取自传入的数据,foo 字段取自默认值

如果想以非主键来索引要修改的数据,可以使用第三个参数:

ts
// 以非主键为基准对数据表进行更新,你需要确保每一个元素都拥有 telegram 属性
await ctx.database.upsert('user', rows, 'telegram')

// 以复合键为基准对数据表进行更新,你需要确保每一个元素都拥有 platform 和 id 属性
await ctx.database.upsert('channel', rows, ['platform', 'id'])

remove:删除数据

使用 database.remove() 方法以删除特定表中的数据。

ts
// 从 schedule 表中删除特定 id 的数据行
// 第二个参数也可以使用上面介绍的查询表达式
await ctx.database.remove('schedule', [id])

获取改动行数

set, upsertremove 操作都会返回一个 WriteResult 对象,它包含了这次操作的结果。你可以通过 matched 属性来获取匹配的数据行数 (注意不是修改的函数),通过 inserted 属性来获取插入的数据行数 (仅限 upsert 操作)。

ts
// 对某个用户的余额进行扣除
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

setupsert 方法都可以用于修改已经存在的数据,它们的区别如下表所示:

setupsert
作用范围支持复杂的查询表达式只能限定特定字段的值
插入行为如果数据不存在则不会进行任何操作如果数据不存在则会插入新行

对比 create 和 upsert

createupsert 方法都可以用于插入新的数据,它们的区别如下表所示:

createupsert
插入数量只能插入一条数据可以批量插入多条数据
返回值返回经过填充后的数据没有返回值
冲突行为如果数据已存在则会报错如果数据已存在则会执行修改