mysql - 如何使用Go的net/http基于鉴权向主从数据库分发请求？

Question

你好很棒的 Stackoverflow 社区，

为这个蹩脚的问题道歉。一段时间以来，我一直试图找到这个问题的答案，但谷歌搜索对我帮助不大。

我正在玩转 Go 的 net/http 包并关注 an article关于组织*sql.Db

通过它之后，我试图获得两种*sql.Db类型，一种用于master，一种用于slave。我的意图是，对于一个简单的路由 /home，如果 sessioncookie 已经设置，则使用主连接并让连接转到主服务器并使用 RW用户，如果不是，则使用与 RO 用户的从属连接。我已经在 *sql.DB 上有一个包装器结构 DB 并附加了一个 CheckSessionCookie 方法。

func (d *DB) CheckSessionCookie(w http.ResponseWriter, r *http.Request) (*http.Cookie) {
    c, err := r.Cookie("session")
    if err != nil {
    return nil
    }
    return c
}

我的路线看起来像

http.HandleFunc("/admin", envMaster.Admin) // Master connection's method
http.HandleFunc("/home", envSlave.Home) // Slave connection's method

但是，如果请求已经设置了 cookie，我想不出一种优雅的方法来使用不同的处理函数。我在想，如果我可以用 javascript 读取 cookie 并将 cookie 作为参数传递，并将路由用作

http.HandleFunc("/home", envMaster.Home)
http.HandleFunc("/home/:session", envSlave.Home)

但我不确定这是否是一个好方法，尤其是当 http only 标志设置为 cookie 时。

如果对“/home”的请求已经设置了 session cookie，我该如何使用主连接？

谢谢。

Answer 1

一种方法可能是这样的:

type Env struct {
    master *sql.DB
    slave *sql.DB
}

func (e *Env) DB(r *http.Request) *sql.DB {
    if _, err := r.Cookie("session"); err == http.ErrNoCookie {
            return e.slave
        }
        return e.master
}

func (e *Env) Home(w http.ResponseWriter, r *http.Request) {
    db := e.DB(r)

    // ...
}

---

注意: 以下是个人意见，因此并非真正重要。

虽然上面的示例可能适合您，但我个人认为在 Env 类型上添加处理程序是一个糟糕的设计决策。如果你有一个带有 3、5 或 10 条路由的小应用程序，它可能没问题，但如果你需要扩展到更多，比如 50 条路由，你最终会得到一个具有 50 多个方法的类型，而方法可能取决于环境它不是很明显，至少对我来说，为什么环境本身应该在其上定义行为。

然后还有可能不同的处理程序需要不同的配置类型，或者相同的配置类型但具有不同的值，Env 现在不仅有 50 多个方法，它还有许多字段具有彼此之间几乎没有共同点。这也允许处理程序触摸它没有业务接触的东西。

在我看来，Env 类型最多应该包含一组值，这些值表示应用程序将要运行的环境，仅此而已，没有行为，或者至少只是谓词以及实际上作为环境的一部分有意义的 setter/getter 。