需要的知识点
为了防止你的心里不适,需要以下知识点:
- Go 基本知识
- Go 反射的深入理解
- 使用过框架
Go Web 服务器搭建
package main import ( "fmt" "net/http" ) func do(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Hello World!") //这个写入到w的是输出到客户端的 } func main() { http.HandleFunc("/", do) //设置访问的路由 http.ListenAndServe(":9090", nil) //设置监听的端口 }
上面的例子调用了http默认的DefaultServeMux来添加路由,需要提供两个参数,第一个参数是希望用户访问此资源的URL路径(保存在r.URL.Path),第二参数是即将要执行的函数,以提供用户访问的资源。
Go默认的路由添加是通过函数http.Handle
和http.HandleFunc
等来添加,底层都是调用了DefaultServeMux.Handle(pattern string, handler Handler)
,这个函数会把路由信息存储在一个map信息中map[string]muxEntry。
Go监听端口,然后接收到tcp连接会扔给Handler来处理,上面的例子默认nil即为http.DefaultServeMux
,通过DefaultServeMux.ServeHTTP
函数来进行调度,遍历之前存储的map路由信息,和用户访问的URL进行匹配,以查询对应注册的处理函数。
你可以通过文档查看 http.ListenAndServe 的方法,第二个参数是 Handler 类型的接口,只要实现 Handler 接口,就可以实现自定义路由。
func ListenAndServe(addr string, handler Handler) error
实现自定义路由:
package main import ( "fmt" "net/http" ) type MyMux struct { } func (p *MyMux) ServeHTTP(w http.ResponseWriter, r *http.Request) { if r.URL.Path == "/" { sayhelloName(w, r) return } http.NotFound(w, r) return } func sayhelloName(w http.ResponseWriter, r *http.Request) { fmt.Fprintf(w, "Hello myroute!") } func main() { mux := &MyMux{} http.ListenAndServe(":9090", mux) }
通过自定义路由,实现简单 MVC 框架
两个基本结构的定义:
type controllerInfo struct { url string controllerType reflect.Type } type ControllerRegistor struct { routers []*controllerInfo }
整体思路:
- controllerInfo url 是添加时候对应的路由, controllerType 反射类型。
- 通过 mux.Add("/", &DefaultController{}) 前台添加的信息,放到一个 routers []*controllerInfo 数组中
- 每一次请求都会遍历 routes, 判断当前的 r.URL.Path 是否与 routes 里面一个相等,如果相等, 通过类型反射,执行相应的方法。
流程图:
源码实现
package main import ( "fmt" "net/http" "reflect" )
type controllerInfo struct { url string controllerType reflect.Type } type ControllerRegistor struct { routers []*controllerInfo } type ControllerInterface interface { Do() } type UserController struct { } type DefaultController struct { } func (u *UserController) Do() { fmt.Println("I`m UserController") } func (d *DefaultController) Do() { fmt.Println("I`m DefaultController") } func (p *ControllerRegistor) Add(pattern string, c ControllerInterface) { //now create the Route t := reflect.TypeOf(c).Elem() route := &controllerInfo{} route.url = pattern route.controllerType = t p.routers = append(p.routers, route) } // AutoRoute func (p *ControllerRegistor) ServeHTTP(w http.ResponseWriter, r *http.Request) { var started bool requestPath := r.URL.Path fmt.Println(requestPath) //find a matching Route for _, route := range p.routers { if requestPath == route.url { vc := reflect.New(route.controllerType) method := vc.MethodByName("Do") method.Call(nil) started = true fmt.Fprintf(w, "Hello " + route.controllerType.Name()) break } } //if no matches to url, throw a not found exception if started == false { http.NotFound(w, r) } } func main() { mux := &ControllerRegistor{} mux.Add("/", &DefaultController{}) mux.Add("/user", &UserController{}) s := &http.Server{ Addr: ":9527", Handler: mux, } s.ListenAndServe() }
以上只是一个简陋的实现思路,可以优化加上具体的功能和通过参数调用方法等。
参考链接: