在运行时切换数据库连接时的 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 的 string
或 uint
)。
这样你就可以在运行时做所有事情:
- 当您需要添加租户时,只需实例化该租户的
repository
并将其添加到地图即可 - 当您需要删除租户时,只需从地图中删除其
repository
并关闭数据库连接即可 - 当需要查询某个租户时,可以在地图中查找对应的
repository
,并使用它来对该租户执行查询
如果上述操作可能同时发生,请确保您使用某种同步机制来避免访问地图时出现数据争用(例如 sync.map
或 sync.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学习网公众号吧!
-
502 收藏
-
502 收藏
-
501 收藏
-
501 收藏
-
501 收藏
-
139 收藏
-
204 收藏
-
325 收藏
-
477 收藏
-
486 收藏
-
439 收藏
-
357 收藏
-
352 收藏
-
101 收藏
-
440 收藏
-
212 收藏
-
143 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 508次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 497次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 484次学习