go - 在 Golang Web 服务器中使用映射处理程序

Question

我需要为我的 Golang 网络服务器中的特定请求定义请求处理程序。我目前的做法如下

package main

import "net/http"

type apiFunc func(rg string, w http.ResponseWriter, r *http.Request)

func h1(rg string, w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Bonjour"))
}

func h2(rg string, w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Guten Tag!"))
}

func h3(rg string, w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Good Morning!"))
}

type gHandlers map[string]apiFunc

var handlers gHandlers

func handleConnection(w http.ResponseWriter, r *http.Request) {
    hh := r.URL.Query().Get("handler")
    handlers[hh]("rg", w, r)
}

func main() {
    handlers = make(map[string]apiFunc, 3)
    handlers["h1"] = h1
    handlers["h2"] = h2
    handlers["h3"] = h3
    http.HandleFunc("/", handleConnection)
    http.ListenAndServe(":8080", nil)
}

这很好用。然而，我仍然是 Golang 的新手，所以它可能不是“正确”的做事方式。对于任何能够指出是否是实现此结果的更好方法的人，我将非常感激

Answer 1

在 handleConnection 中使用 switch 语句如何？

switch hh {
case "h1":
    h1("rg", w, r)
case "h2":
    h2("rg", w, r)
case "h3":
    h3("rg", w, r)
default:
    // return HTTP 400 here
}

优点是:

• 更易于理解的代码:
  • 没有 apiFunc 和 gHandlers 类型
  • 无需浏览源代码即可了解路由逻辑，一切尽在一处
• 更灵活:您可以调用具有不同参数的函数，并在必要时实现更复杂的路由规则。