每次数据库调用前初始化 gorm 连接
来源:stackoverflow
时间:2024-02-19 11:06:23 427浏览 收藏
一分耕耘,一分收获!既然打开了这篇文章《每次数据库调用前初始化 gorm 连接》,就坚持看下去吧!文中内容包含等等知识点...希望你能在阅读本文后,能真真实实学到知识或者帮你解决心中的疑惑,也欢迎大佬或者新人朋友们多留言评论,多给建议!谢谢!
问题内容
我有一个使用 swagger 创建的 api,它通过 gorm 与 mariadb 实例进行通信。我对处理数据库连接模块以及传递数据库处理程序的方式不满意。我怎样才能使这个更加健壮,并且可能合并接口,并且每次发出数据库请求时总是调用 db.open
函数。
这是我由 swagger 创建的配置文件
func configureapi(api *operations.hairdooapi) http.handler { // configure the api here api.serveerror = errors.serveerror // set your custom logger if needed. default one is log.printf // expected interface func(string, ...interface{}) // // example: //api.logger = log.printf api.useswaggerui() // to continue using redoc as your ui, uncomment the following line // api.useredoc() api.jsonconsumer = runtime.jsonconsumer() //api.urlformconsumer = runtime.discardconsumer api.jsonproducer = runtime.jsonproducer() api.companyaddcompanyhandler = company.addcompanyhandlerfunc(func(params company.addcompanyparams) middleware.responder { //log.debugf("add commpay handler called") //log.infof("payload data : %v", params.body) response, err := handlers.addcompany(params) if err != nil { return company.newaddcompanybadrequest().withpayload(&models.errorreponse400{ error: err.error(), }) } return company.newaddcompanycreated().withpayload(response) })
这是我对 addcompany
函数的实现。
package handlers import ( "hairdoo.com/m/v2/db" "hairdoo.com/m/v2/models" "hairdoo.com/m/v2/restapi/operations/company" ) func addcompany(params company.addcompanyparams) (*models.company, error) { result, err := db.addcompany(params.body) if err != nil { return nil, err } return result, nil }
最后是数据库包的实现
package db import ( "fmt" "time" "gorm.io/gorm" "hairdoo.com/m/v2/models" ) type Company struct { gorm.Model Name string `gorm:"uniqueIndex"` Phone string Email string ContatctPerson string CategoryID int Category *Category Status string } type Category struct { ID int Name string `gorm:"uniqueIndex"` } func AddCompany(companyObj *models.Company) (*models.Company, error) { db := Open() var result *gorm.DB category := &Category{} db.Where("name = ?", companyObj.Category.Name).First(category) company := convertToDBCompany(companyObj, false) if category.ID != 0 { result = db.Exec("INSERT INTO companies (created_at, updated_at, name, phone, email, contatct_person, category_id) values (?, ?, ?, ?, ?, ?, ?);", time.Now(), time.Now(), companyObj.Name, companyObj.Phone, companyObj.Email, companyObj.ContatctPerson, category.ID) } else { result = db.Create(company) } if result.RowsAffected < 1 { return nil, result.Error } // add employee to db if object present var employees []*models.EmployeeItems0 if companyObj.Employee != nil { for _, employeeInput := range companyObj.Employee { result, err := AddEmployee(employeeInput) if err != nil { return nil, err } employees = append(employees, result) } } return convertToModelCompany(company, employees), nil }
解决方案
实现此目的的一种方法是创建一个结构体,该结构体存储数据库连接并在该结构体上放置方法,例如:
type dbService struct { *gorm.DB } func InitDB() *dbService { db, err := gorm.Open(...) if err != nil { panic("failed to connect database") } db.AutoMigrate(&Category{}, &Company{}) return &dbService{db} } func (db *dbService) AddCompany(companyObj *models.Company) (*models.Company, error) { company := convertToDBCompany(companyObj, false) db.FirstOrInit(company.Category, &Category{Name: companyObj.Category.Name}) result := db.Create(&company) if result.RowsAffected < 1 { return nil, result.Error } ... }
如果您已经在使用 gorm,我建议您实际利用它的 orm 功能;)
今天关于《每次数据库调用前初始化 gorm 连接》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!
声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
-
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次学习