使用 main.go 中初始化的变量进行其他打包和测试仍是 Go 中解耦的项目结构的一部分吗？

小伙伴们对Golang编程感兴趣吗？是否正在学习相关知识点？如果是，那么本文《使用 main.go 中初始化的变量进行其他打包和测试仍是 Go 中解耦的项目结构的一部分吗？》，就很适合你，本篇文章讲解的知识点主要包括。在之后的文章中也会多多分享相关知识点，希望对大家的知识积累有所帮助！

我将从 python/django 切换到 go。在 django 中，我非常喜欢它的模块化应用程序项目结构设计，其中每个应用程序都有单独的业务模型、路由和视图。然后，所有应用程序都将在中心/项目的主路由系统等内进行通信。

django项目结构例如：

- myproject
    - myproject
        - urls.py
        - views.py
        ...
    - planner
        - urls.py
        - views.py
        - models.py
        ...

我正在尝试在go项目中实现类似的项目设计：

- myproject
        - cmd
            - api
                - main.go
                - routes.go
                - handlers.go
            - planner
                - routes.go
                - handlers.go
                - models.go

摘自 cmd/api/main.go：

package main

...

db, err := sql.open("pgx", cfg.db.dsn)
...
srv := &http.server{
    addr:         fmt.sprintf("localhost:%d", app.config.port),
    handler:      app.routes()
}
...

摘自 cmd/api/routes.go：

package main

func (app *application) routes() *httprouter.router {
    router := httprouter.new()

    planner.routes(router)

    return router
}

摘自 cmd/planner/routes.go：

package planner
...
func routes(router *httprouter.router) {
    router.handlerfunc(http.methodpost, "/v1/planner/todos", createtodohandler)
}

摘自 cmd/planner/models.go：

package planner

type plannermodel struct {
    db *sql.db
}

func (p plannermodel) inserttodo(todo *todo) error {
    query := `insert into todos (title, user_id)
    values ($1, $2)
    returning id, created_at`

    return p.db.queryrow(query, todo.title, todo.userid).scan(&todo.id, &todo.createdat)
}

现在的问题是我需要使用在 cmd/api/main.go 文件中初始化的数据库连接，从 package main 到 cmd/planner/handlers.go。由于变量来自主包，我无法将其导入到我的应用程序（规划器）处理函数中。

package planner

func CreateTodoHandler(w http.ResponseWriter, r *http.Request) {
    var input struct {
        Title  string `json:"title"`
        UserID int64  `json:"user_id"`
    }

    err := helpers.ReadJSON(w, r, &input)
    ...
    todo := &Todo{
        Title:  input.Title,
        UserID: input.UserID,
    }
    ...
    // How to use/inject DB connection dependency into the PlannerModel?
    pm := PlannerModel{
        DB:    // ????
    } 

    err = pm.InsertTodo(todo)
    ...
}

我认为拥有一个全局数据库变量可以解决问题，或者我发现的合理答案是在单独的包中声明该变量，并在 main.go 中初始化它。另一种方法是使 planner/handlers.go 成为 package main 并在主包中创建一个 application 结构来保存项目的所有模型并在处理程序中使用它，但我想这会破坏解耦架构设计。

我想知道什么是首选方法，或者是否有更好的方法来拥有类似 django 的解耦项目结构，并且仍然使用 main.go 中初始化的变量到其他包中并进行测试？

正确答案

当我从 python/django 切换到 go 时，我也有过类似的经历。

在每个应用程序中访问 db 连接的解决方案是在每个应用程序中定义具有 db 连接字段的结构，然后在 main 中创建 db 连接和所有应用程序结构。

// planner.go

func (t *todo) routes(router *httprouter.router) {
    router.handlerfunc(http.methodpost, "/v1/planner/todos", t.createtodohandler)
}

// handlers.go

struct Todo {
    DB: *sql.DB
}

func (t *Todo) CreateTodo(w http.ResponseWriter, r *http.Request) {
    var input struct {
        Title  string `json:"title"`
        UserID int64  `json:"user_id"`
    }

    err := helpers.ReadJSON(w, r, &input)
    ...
    todo := &Todo{
        Title:  input.Title,
        UserID: input.UserID,
    }
    ...

    pm := PlannerModel{
        DB: t.DB
    } 

    err = pm.InsertTodo(todo)
    ...
}

这可以解决您当前的问题，但如果您没有更好的应用程序设计，则会出现其他问题。

我建议阅读这两篇博客文章，以更好地了解在 go 中设计应用程序和构建代码。

https://www.gobeyond.dev/standard-package-layout/
https://www.gobeyond.dev/packages-as-layers/

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。