BeegoAutoRouter工作原理解析

import "GitHub.com/beego/beego/v2/server/web"
type ReganYueController struct {
    web.Controller
}

import "github.com/beego/beego/v2/server/web"
type ReganYueController struct {
  web.Controller
}
func (u *ReganYueController) HelloWorld()  {
  u.Ctx.WriteString("Welcome, Regan Yue")
}
func main() {
  web.AutoRouter(&ReganYueController{})
  web.Run()
}

beego.Router("/", &IndexController{})

func (u *ReganYueController) HelloWorld()  {
  u.Ctx.WriteString("Welcome, Regan Yue")
}

// AutoRouter see httpserver.AutoRouter
func AutoRouter(c ControllerInterface) *HttpServer {
  return BeeApp.AutoRouter(c)
}

// AutoRouter adds defined controller handler to BeeApp.
// it's same to HttpServer.AutoRouter.
// if beego.AddAuto(&MainContorlller{}) and MainController has methods List and Page,
// visit the url /main/list to exec List function or /main/page to exec Page function.
func (app *HttpServer) AutoRouter(c ControllerInterface) *HttpServer {
  app.Handlers.AddAuto(c)
  return app
}

// NewHttpServerWithCfg will create an sever with specific cfg
func NewHttpServerWithCfg(cfg *Config) *HttpServer {
  cr := NewControllerReGISterWithCfg(cfg)
  app := &HttpServer{
    Handlers: cr,
    Server:   &http.Server{},
    Cfg:      cfg,
  }
  return app
}

func (p *ControllerRegister) AddAuto(c ControllerInterface) {
  p.AddAutoPrefix("/", c)
}

beego.AddAutoPrefix("/admin",&MainContorlller{})

func (p *ControllerRegister) AddAutoPrefix(prefix string, c ControllerInterface) {
    //对传入的Controller做反射
  reflectVal := reflect.ValueOf(c)
    //获取传入的Controller的类型
  rt := reflectVal.Type()
    //因为c是指针，所以要用Indirect方法获取指针指向的变量类型
  ct := reflect.Indirect(reflectVal).Type()
    //使用Beego注册controller的名称后面有Controller，这里把它去掉得到controllerName。
  controllerName := strings.TrimSuffix(ct.Name(), "Controller")
    //
  for i := 0; i < rt.NumMethod(); i++ {
    if !utils.InSlice(rt.Method(i).Name, exceptMethod) {
      route := &ControllerInfo{}
      route.routerType = routerTypeBeego
      route.methods = map[string]string{"*": rt.Method(i).Name}
      route.controllerType = ct
      pattern := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(rt.Method(i).Name), "*")
      patternInit := path.Join(prefix, controllerName, rt.Method(i).Name, "*")
      patternFix := path.Join(prefix, strings.ToLower(controllerName), strings.ToLower(rt.Method(i).Name))
      patternFixInit := path.Join(prefix, controllerName, rt.Method(i).Name)
      route.pattern = pattern
      for m := range HTTPMETHOD {
        p.addToRouter(m, pattern, route)
        p.addToRouter(m, patternInit, route)
        p.addToRouter(m, patternFix, route)
        p.addToRouter(m, patternFixInit, route)
      }
    }
  }
}

type iface struct {
        tab  *itab          //类型信息
        data unsafe.Pointer //实际对象指针
}
type itab struct {
        inter *interfacetype //接口类型
        _type *_type         //实际对象类型
        hash  uint32
        _     [4]byte
        fun   [1]uintptr     //实际对象方法地址
}

func InSlice(v string, sl []string) bool {
  for _, vv := range sl {
    if vv == v {
      return true
    }
  }
  return false
}

exceptMethod = []string{"Init", "Prepare", "Finish", "Render", "RenderString",
    "RenderBytes", "Redirect", "Abort", "StopRun", "UrlFor", "ServeJSON", "ServejsONP",
    "ServeYAML", "ServeXML", "Input", "ParseFORM", "GetString", "GetStrings", "GetInt", "GetBool",
    "GetFloat", "GetFile", "SaveToFile", "StartSession", "SetSession", "GetSession",
    "DelSession", "SessionRegenerateID", "DestroySession", "Isajax", "GetSecureCookie",
    "SetSecureCookie", "XsrfToken", "CheckXsrfCookie", "XsrfFormhtml",
    "GetControllerAndAction", "ServeFormatted"}

func (p *ControllerRegister) addToRouter(method, pattern string, r *ControllerInfo) {
  if !p.cfg.RouterCaseSensitive {
    pattern = strings.ToLower(pattern)
  }
  if t, ok := p.routers[method]; ok {
    t.AddRouter(pattern, r)
  } else {
    t := NewTree()
    t.AddRouter(pattern, r)
    p.routers[method] = t
  }
}

t := NewTree()
t.AddRouter(pattern, r)
p.routers[method] = t