-1

我是初学者，现在想创建一个大项目我的问题是应该在哪里创建连接以及在 http 请求中关闭连接现在我在主函数中声明 db 并在所有请求中使用并且不关闭连接：

package main

import (
    "fmt"
    "github.com/jinzhu/gorm"
    "github.com/kataras/iris"
    "github.com/kataras/iris/context"
    "github.com/kataras/iris/middleware/recover"
    _ "github.com/go-sql-driver/mysql"
)

var db *gorm.DB

func main() {


    port := "8080"

    app := iris.New()
    app.Configure()
    app.Logger().SetLevel("debug")

    app.Use(recover.New())

    db1, err := gorm.Open("mysql", "root:@/database?charset=utf8&parseTime=True&loc=Local")
    //db1, err := gorm.Open("mysql", "payro:AEkCpNhd@/payro_db?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        fmt.Println(err.Error())
        panic(err)
    }

    db=db1

    app.Get("/", func(i context.Context) {
        db.Exec("update tbl1 set col1=''")
        i.Next()
    })
    app.Get("/test", func(i context.Context) {
        db.Exec("update tbl2 set col2=''")
        i.Next()
    })
    _ = app.Run(iris.Addr(":"+port), iris.WithConfiguration(iris.Configuration{
        //DisableBodyConsumptionOnUnmarshal:true,

    }), iris.WithoutServerError(iris.ErrServerClosed))
}

这段代码可以吗？

4

1 回答 1

1

您的代码可能是测试/POC 代码。在生产项目中，您可以根据需要使用 MVC 或任何其他类型的架构。很难确定项目的确切结构。但至少，您需要创建一个 db 包，它为所有与 DB 相关的交互声明一个接口。例如

type UserDBRepo interface{
   AddUser(context.Context, *User)
   GetUser(context.Context, uint64)
}

type userDBRepo struct{ //implements UserDBRepo
   *sql.DB // or whatever type gorm.Open returns
}

func NewUserDBRepo(db *sql.DB) DBRepo{
  return &dbRepo{DB: db}
}

以上基本上代表了此示例的单个 RDBMS 表。数据库表可能有n这样的文件。n现在NewUserDBRepo从 main.go 调用并将这个实例传递给所有需要这个数据库的服务。

于 2019-01-16T10:04:22.460 回答