每次数据库调用前初始化 gorm 连接

一分耕耘，一分收获！既然打开了这篇文章《每次数据库调用前初始化 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学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！