时间戳

首页 > 解决方案 > 如何使用依赖注入在golang中正确设置defer .close

mysql - 如何使用依赖注入在golang中正确设置defer .close

问题描述

我想defer xxx.Close()在 golang 中设置关闭连接。
但是如果我defer xxx.Close()在INSERTSQL 方法之后设置，它的结果是空的。如果我在SQL 方法之后
删除，它的结果会返回正确的值。 我设置以避免连接限制和崩溃。但是现在，如果我删除.defer xxx.Close()INSERT
defer xxx.Close()defer xxx.Close()

到go

package dao

import (
    "database/sql"
    "log"
)

type Dao struct {
    database *sql.DB
}

func NewDao(database *sql.DB) *Dao {
    objs := &Dao{database: database}
    return objs
}

type DaoInterface interface {
    GetEmployeeListDao() *sql.Rows
    RegisterEmployeeDao(employee model.RegisterEmployee, uu string, encrypted string)
}

func (d *Dao) GetEmployeeListDao() *sql.Rows {

    results, err := d.database.Query("SELECT id, name, email, isDisplay FROM employee")
    if err != nil {
        panic(err.Error())
    }
    //it returns nothing.
    //If removing defer results.Close(), results returns correct value
    defer results.Close()
    return results
}

func (d *Dao) RegisterEmployeeDao(employee model.RegisterEmployee, uu string, encrypted string) {
    ins, err := d.database.Prepare("INSERT INTO employee(name, email, password, isDisplay) VALUES(?,?,?,?)")
    if err != nil {
        log.Fatal(err)
    }
    //same problem will occure here.
    defer ins.Close()
    _, err = ins.Exec(uu, employee.NAME, employee.EMAIL, encrypted, employee.ISDISPLAY)
    if err != nil {
        log.Fatal(err)
    }
}

服务.go

package service

import (
    "crypto/aes"
    "crypto/cipher"
    "crypto/rand"
    "encoding/base64"
    "io"

    "github.com/gin-gonic/gin"
)

type Service struct {
    dao dao.DaoInterface
}

func NewService(dao dao.DaoInterface) *Service {
    return &Service{dao: dao}
}

type ServiceInterface interface {
    GetEmployeeListService() []model.Employee
    RegisterEmployeeService(c *gin.Context)
}

func (s Service) GetEmployeeListService() []model.Employee {
    var employee []model.Employee

    results := s.dao.GetEmployeeListDao()

    singleEmployee := model.Employee{}
    for results.Next() {
        err := results.Scan(&singleEmployee.ID, &singleEmployee.NAME, &singleEmployee.EMAIL, &singleEmployee.ISDISPLAY)
        if err != nil {
            panic(err.Error())
        } else {
            employee = append(employee, singleEmployee)
        }
    }

    defer results.Close()

    var displayEmployee []model.Employee

    for _, e := range employee {
        if e.ISDISPLAY == true {
            displayEmployee = append(displayEmployee, e)
        }
    }

    return displayEmployee
}

func (s Service) RegisterEmployeeService(c *gin.Context) {
    var employee model.RegisterEmployee
    c.BindJSON(&employee)

    block, err := aes.NewCipher([]byte("aes-secret-key-1"))
    if err != nil {
        panic(err)
    }
    b := base64.StdEncoding.EncodeToString([]byte(employee.PASSWORD))
    cipt := make([]byte, aes.BlockSize+len(b))
    iv := cipt[:aes.BlockSize]
    if _, err := io.ReadFull(rand.Reader, iv); err != nil {
        panic(err)
    }
    cfb := cipher.NewCFBEncrypter(block, iv)
    cfb.XORKeyStream(cipt[aes.BlockSize:], []byte(b))
    encrypted := base64.StdEncoding.EncodeToString(cipt)

    s.dao.RegisterEmployeeDao(employee, encrypted)
}

我在 api 中设置了依赖注入样式。main.go

package main

import (
    "os"
    _ "github.com/go-sql-driver/mysql"
)

func main() {
    db := db.NewDatabase(os.Getenv("MYSQL_USER"), os.Getenv("MYSQL_PASSWORD"), os.Getenv("MYSQL_HOST"))
    dao := dao.NewDao(db.DATABASE)
    service := service.NewService(dao)
    cntlr := controller.NewController(service)
    router := handler.NewHandler(cntlr)

    router.Router.Run(":8080")
}

我希望defer xxx.Close()没有问题，并且 insert 和 select sql 返回正确的值。
但问题是因为defer xxx.Close()insert 和 select sql 没有返回正确的值。
我想知道如何避免连接限制和没有这个问题的崩溃。

标签： mysqlsqlgodependency-injection

解决方案

我认为您所期望的是结果集以数组或类似的形式立即返回。这不会发生，因为客户端是懒惰的- 在您明确尝试使用Next(). 这是有充分理由的——如果结果集中有数百万行，通过网络连接传输数据会非常慢，甚至可能耗尽运行查询的机器的内存。为了避免这种情况，结果集实际上更像是一个“句柄”，它允许您在需要时从数据库中获取更多行，而不是结果本身。

当您调用Close()结果集时，它表明您不再需要任何数据。

这里也可能有些混乱defer。defer在函数结束时触发，所以它会在你返回时触发：

defer results.Close()
return results  // defer is triggered here, which will close the result set

然后，当您尝试从中读取时，它是一个封闭的结果集，因此您实际上无法从中读取任何内容。

作为参考，这里Close()是有关调用结果集时究竟发生了什么的文档。

推荐阅读