登录
首页 >  Golang >  Go问答

在运行时切换数据库连接时的 SQLC 用法指南

来源:stackoverflow

时间:2024-02-17 08:36:24 458浏览 收藏

golang学习网今天将给大家带来《在运行时切换数据库连接时的 SQLC 用法指南》,感兴趣的朋友请继续看下去吧!以下内容将会涉及到等等知识点,如果你是正在学习Golang或者已经是大佬级别了,都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家!

问题内容

我知道这个问题已经被问过很多次了,但我没有找到适合我的情况的好的答案。我正在使用 sqlc 生成查询数据库的方法。使用开始时初始化的一个连接时,一切工作正常。现在我需要在多租户环境中设置它,其中每个租户都有一个单独的数据库。现在,我想从连接租户与数据库连接的连接映射(map[string]*sql.db)开始。我的问题是关于在运行时覆盖/选择连接。通过一个连接,存储库的初始化如下:

type Repository interface {
    GetCustomerById(ctx context.Context, id int64) (Customer, error)
    ListCustomers(ctx context.Context) ([]Customer, error)
}

type repoSvc struct {
    *Queries
    db *sql.DB
}

func NewRepository(dbconn *sql.DB) Repository {
    return &repoSvc{
        Queries: New(dbconn),
        db:      dbconn,
    }
}

customerRepo := customerRepo.NewRepository(conn)

getcustomerbyid 是 sqlc 生成的方法 conn 是数据库连接

如何根据参数(来自 cookie 或上下文)建立连接?


正确答案


假设您使用单独的数据库,最简单的方法是维护 map[tenantid]repository,其中 tenantid 是区分租户的方式(例如,包含租户 id 的 stringuint)。

这样你就可以在运行时做所有事情:

  • 当您需要添加租户时,只需实例化该租户的 repository 并将其添加到地图即可
  • 当您需要删除租户时,只需从地图中删除其 repository 并关闭数据库连接即可
  • 当需要查询某个租户时,可以在地图中查找对应的repository,并使用它来对该租户执行查询

如果上述操作可能同时发生,请确保您使用某种同步机制来避免访问地图时出现数据争用(例如 sync.mapsync.rwmutex)。

如果您有一个存储租户及其数据库连接 uri 的数据库表,您仍然可以使用此方法:当您需要执行查询检查 repository 是否存在于 map 中时:如果缺少,则查询租户表并将该租户的 repository 添加到地图中。然后,您可以定期扫描 map 并删除一段时间未使用的任何 repository

为了使这一切变得更容易,您还可以将整个机器包装到 multitenantrepository 接口中,该接口与 repository 接口相同,但在每个方法上接受额外的 tenantid 参数:

type MultitenantRepository interface {
    GetCustomerById(ctx context.Context, tenant tenantID, id int64) (Customer, error)
    ListCustomers(ctx context.Context, tenant tenantID) ([]Customer, error)
}

这将避免将多租户设置的所有复杂性暴露给您的业务逻辑。

理论要掌握,实操不能落!以上关于《在运行时切换数据库连接时的 SQLC 用法指南》的详细介绍,大家都掌握了吧!如果想要继续提升自己的能力,那么就来关注golang学习网公众号吧!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>