我在 golang 中构建自己的路由器以理解逻辑,但是我收到 404 错误,通常找不到,我能够运行到服务器但是我在 hello 名称中编写了函数,它没有运行。这是什么原因?
package main
import (
"fmt"
"log"
"net/http"
"strings"
"time"
)
var Session *http.Server
var r Router
func Run(port string) {
Session = &http.Server{
Addr: port,
Handler: &r,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
MaxHeaderBytes: 1 << 20,
}
log.Fatal(Session.ListenAndServe())
}
type Handle func(http.ResponseWriter, *http.Request)
type Router struct {
mux map[string]Handle
}
func newRouter() *Router {
return &Router{
mux: make(map[string]Handle),
}
}
func (r *Router) Add(path string, handle Handle) {
r.mux[path] = handle
}
func GetHeader(url string) string {
sl := strings.Split(url, "/")
return fmt.Sprintf("/%s", sl[1])
}
func (rt *Router) ServeHTTP(w http.ResponseWriter, r *http.Request) {
head := GetHeader(r.URL.Path)
h, ok := rt.mux[head]
if ok {
h(w, r)
return
}
http.NotFound(w, r)
}
func hello(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "%s", "hello world")
}
func main() {
r := newRouter()
r.Add("/hello", hello)
Run(":8080")
}
最佳答案
您的主函数通过使用 :=
表示法声明并初始化一个名为 r 的新局部变量,其作用域不同于您列表顶部的全局变量。
这意味着您要添加处理程序的 r
变量不是您尝试分配给 Http.Server
结构的全局变量,因此您的 ServeHTTP
根本找不到任何映射到它自己的接收器 r
中。
一种解决方案是将 r Router
声明为指向 Router
的指针,如下所示:
var r *Router
并且在您的处理程序中,您可以直接传递指针而无需任何取消引用:
Session = &http.Server{
Addr: port,
Handler: r,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
MaxHeaderBytes: 1 << 20,
}
当然,请记住通过分配指针值而不是声明一个新的本地变量来初始化变量:
func main() {
r = newRouter()
r.Add("/hello", hello)
Run(":8080")
}
关于go - 如何在 golang 中构建自己的简单路由器?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40070168/