如何在 NestJS 中的 TypeORM 中编写 setval ？

在NestJS中使用TypeORM时，如果你需要调整序列的当前值，例如在测试期间或者在重新调整数据库时，你可能会想要使用类似PostgreSQL的 setval() 函数。setval() 函数是PostgreSQL特有的，用于设置序列的当前值。在TypeORM中，你可以通过执行原生SQL语句来完成这个操作。

以下是一个具体的步骤和示例，说明如何在NestJS中使用TypeORM来调用 setval()：

步骤 1: 注入EntityManager

首先，确保你的服务中注入了 EntityManager。EntityManager 允许你执行原生SQL查询。

typescript
import { Injectable } from '@nestjs/common';
import { EntityManager } from 'typeorm';

@Injectable()
export class YourService {
    constructor(private entityManager: EntityManager) {}
}

步骤 2: 执行原生SQL来设置序列值

使用 entityManager 的 query() 方法执行原生SQL。你需要提供序列名和你想要设置的新值。

typescript
async setSequenceValue(sequenceName: string, newValue: number): Promise<void> {
    await this.entityManager.query(`SELECT setval('${sequenceName}', ${newValue}, false)`);
}

这里 setval() 的第三个参数设置为 false 表示序列的下一个 nextval() 调用将返回设定的这个新值。如果设置为 true，nextval() 将返回这个新值加上序列的步长（通常是1）。

示例：调整用户ID序列

假设你有一个用户表 user，其ID是自增的。在某些情况下（比如数据迁移后），你可能需要重设这个ID的序列值。

typescript
@Injectable()
export class UserService {
    constructor(private entityManager: EntityManager) {}

    async resetUserIdSequence(nextId: number): Promise<void> {
        // 假设序列名为 'user_id_seq'
        await this.setSequenceValue('user_id_seq', nextId - 1);
    }
}

在这个例子中，如果你想让下一个用户ID为100，你应该调用 resetUserIdSequence(100)。此函数会将序列设置为99，因此，下一个通过 nextval() 生成的ID将会是100。

结论

通过上述方式，你可以在使用NestJS和TypeORM的环境中灵活地调整序列值，这对于数据库维护和特定的应用场景非常有用。不过请注意，直接操作数据库序列可能会引入风险，尤其是在并发高的生产环境中，因此需要谨慎操作。