在go-nunu项目中优雅实现数据库初始化脚本

在Go语言项目开发中，数据库迁移和初始化数据是常见的需求。本文将详细介绍在go-nunu项目中如何优雅地实现数据库初始化脚本。

初始化脚本的存放位置

在go-nunu项目中，推荐两种方式来存放数据库初始化脚本：

1. 直接放在server/migration.go文件中：这是最简单直接的方式，适合初始化逻辑不太复杂的情况。你可以在这个文件中编写初始化数据的逻辑，然后在适当的时候调用。

2. 创建internal/migrate目录：当初始化逻辑比较复杂时，可以创建一个专门的目录来组织这些代码。这种方式类似于项目中task和job的实现方式，提供了更好的代码组织和可维护性。

实现建议

对于简单的初始化逻辑，可以直接在migration.go文件中添加类似如下的代码：

func initData(ctx context.Context, svc *service.Service) error {
    // 初始化用户数据
    if err := svc.UserRepo.Create(ctx, &model.User{
        Username: "admin",
        Password: "123456",
    }); err != nil {
        return err
    }
    
    // 初始化角色数据
    // ...
    
    return nil
}

然后在数据库迁移完成后调用这个函数。

对于复杂的初始化逻辑，建议采用类似task和job的实现方式：

1. 创建internal/migrate目录
2. 在该目录下创建init_data.go文件
3. 实现一个InitData结构体和方法
4. 通过依赖注入的方式使用服务层

这种方式的好处是：

• 代码结构清晰
• 易于维护和扩展
• 可以复用项目中的服务层代码
• 方便进行单元测试

最佳实践

1. 幂等性：确保初始化脚本可以多次执行而不会产生重复数据或错误。

2. 事务处理：对于重要的初始化数据，使用事务确保数据一致性。

3. 环境区分：考虑不同环境（开发、测试、生产）可能需要不同的初始化数据。

4. 日志记录：记录初始化过程中的关键操作，便于排查问题。

5. 配置化：考虑将部分初始化数据放在配置文件中，提高灵活性。

通过以上方式，你可以在go-nunu项目中优雅地实现数据库初始化脚本，既保证了代码的可维护性，又能满足不同复杂度的初始化需求。