Golang中的路由使用详解

type CopyRouter struct {
  router map[string]map[string]http.HandlerFunc
}

func (c *CopyRouter) HandleFunc(method, pattern string, handle http.HandlerFunc) {
  if method == "" {
    panic("Method can not be null!")
  }

  if pattern == "" {
    panic("Pattern can not be null!")
  }

  if _, ok := c.router[method][pattern]; ok {
    panic("Pattern Exists!")
  }

  if c.router == nil {
    c.router = make(map[string]map[string]http.HandlerFunc)
  }

  if c.router[method] == nil {
    c.router[method] = make(map[string]http.HandlerFunc)
  }
  c.router[method][pattern] = handle
}

func (c *CopyRouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  if f, ok := c.router[r.Method][r.URL.String()]; ok {
    f.ServeHTTP(w, r)
  }
}

func NewRouter() *CopyRouter {
  return new(CopyRouter)
}

func sayHi(w http.ResponseWriter, r *http.Request) {
  fmt.Fprint(w,"Hi")
}

func main() {
  copyRouter := copyrouter.NewRouter()
  copyRouter.HandleFunc("GET","/sayHi", sayHi)
  log.Fatal(http.ListenAndServe("localhost:8080", copyRouter))
}

func (c *CopyRouter) GET(pattern string, handler http.HandlerFunc){
  c.HandleFunc("GET", pattern, handler)
}

func (c *CopyRouter) POST(pattern string, handler http.HandlerFunc){
  c.HandleFunc("POST", pattern, handler)
}

...

copyRouter.GET("/sayHi",sayHi)

func (c *CopyRouter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  if f, ok := c.router[r.Method][r.URL.String()]; ok {
    func (handler http.Handler){
      start := time.Now()
      log.Printf(" 请求 [%s] 开始时间为 : %v\n", r.URL.String(), start)
      f.ServeHTTP(w, r)
      log.Printf(" 请求 [%s] 完成时间为 : %v\n", r.URL.String(), time.Since(start))
    }(f)
  }
}

copyRouter.GET("/sayHi",HandlerFunc(sayHi))

type HandlerFunc func(ResponseWriter, *Request)

func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
  f(w, r)
}