在 Golang 中如何将变量传递至 chi 路由器

在 Chi 路由器中，使用全局变量来传递数据库连接字符串和会话存储不是最佳实践。本文提供了将这些变量传递给 Chi mount 函数的解决方案。通过创建一个自定义资源结构，我们可以将这些变量作为其字段，然后在创建路由时传递该结构。这种方法有助于提高代码的可测试性和可维护性，同时避免了对全局变量的依赖。

我正在使用 chi 路由器编写一个 web 应用程序，并为数据库连接字符串和 gorilla 会话 mysqlstore 提供全局变量。我想将这些传递给 chi mount 函数而不是使用全局变量，但不知道如何做到这一点。

这是我的代码，为简洁起见进行了编辑。

var dBSource string

var store *mysqlstore.MySQLStore

func main() {

  var err error

  dBSource = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8", dbUser, dbPass, dbHost, dbPort, dbName)

  store, err = mysqlstore.NewMySQLStore(endpoint, tableName, path, maxAge, codecs)

  defer store.Close()

  r := chi.NewRouter()

  r.Mount("/users", usersResource{}.Routes())
}

type usersResource struct{}

func (rs usersResource) Routes() chi.Router {
    r := chi.NewRouter()

    r.Get("/", rs.List)

    return r
}

func (rs usersResource) List(w http.ResponseWriter, r *http.Request) {

    session, err := store.Get(r, "session-data")

    db, err := sql.Open("mysql", dBSource)
}

解决方案

通常，我们这样做。

func main() {

......
  r := chi.newrouter()

  // keep db connection alive.
  db, err := sql.open("mysql", dbsource)

  ur := newuserresource(store, db)

  r.mount("/users", ur.routes())
}

type usersresource struct{
  store *mysqlstore.mysqlstore
// and more...
  db *yourdb
}

// new function to pass global var
func newuserresource(store *mysqlstore.mysqlstore, db *yourdb) *usersresource{
  return &usersresource {
    store: store,
    db: db,
  }
}

func (rs *usersresource) list(w http.responsewriter, r *http.request) {
    // don't use global var
    session, err := ur.store.get(r, "session-data")

    ur.db.getwhatever()

}

为了完整起见，以下是我如何实现可接受的解决方案。希望它可以为另一个新手节省大量时间。

type appResource struct {
    tmpl  *template.Template // net/http
    store *mysqlstore.MySQLStore
    db    *sql.DB // database/sql
}

func newAppResource(store *mysqlstore.MySQLStore, db *sql.DB, tmpl *template.Template) *appResource {
    return &appResource{
        tmpl:  tmpl,
        store: store,
        db:    db,
    }
}

func main() {

  var err error

  err = godotenv.Load()
  if err != nil {
     log.Fatal("Error loading .env file")
  }

  dbUser := os.Getenv("dbUser")
  dbName := os.Getenv("dbName")
  dbPass := os.Getenv("dbPass")
  dbHost := os.Getenv("dbHost")
  dbPort := os.Getenv("dbPort")

  dBSource := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8", dbUser, dbPass, dbHost, dbPort, dbName)

  db, err := sql.Open("mysql", dBSource)
  if err != nil {
     log.Fatal(err)
  }

  sessionKey := os.Getenv("sessionKey")

  endpoint := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true&loc=Local", dbUser, dbPass, dbHost, dbPort, dbName)
  tableName := "sessions"
  path := "/"
  maxAge := 3600
  codecs := []byte(sessionKey)

  store, err := mysqlstore.NewMySQLStore(endpoint, tableName, path, maxAge, codecs)
  if err != nil {
    log.Println("SESSIONS STORE error")
    log.Fatal(err)
  }

  r := chi.NewRouter()

  r.Get("/", rs.adminIndex)

  r.Mount("/users", rs.userRoutes())

  httpPort := os.Getenv("httpPort")

  http.ListenAndServe(httpPort, r)
}

func (rs *appResource) adminIndex(w http.ResponseWriter, r *http.Request) {
    session, err := rs.store.Get(r, "admin-data")
    if err != nil {
        log.Fatal(err)
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    var flash map[string]interface{}

    if flashes := session.Flashes(); len(flashes) > 0 {
        v := map[string]interface{}{
            "FlashedMessages": session.Flashes(),
        }
        flash = v
    }

    if err := rs.tmpl.ExecuteTemplate(w, "admin-index", flash); err != nil {
        log.Println(err.Error())
        http.Error(w, http.StatusText(500), 500)
    }
}

到这里，我们也就讲完了《在 Golang 中如何将变量传递至 chi 路由器》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！