当前位置 博文首页 > golang利用不到20行代码实现路由调度详解

    golang利用不到20行代码实现路由调度详解

    作者:xialeistudio 时间:2021-07-02 17:48

    前言

    本文主要介绍了关于golang实现路由调度的相关内容,分享出来供大家参考学习,下面话不多说了,来一起看看详细的介绍吧

    项目地址

    github (本地下载)

    本项目依赖

    使用标准库实现,无额外依赖

    为什么需要路由调度层

    golang http标准库只能精确匹配请求的URI,然后执行handler。现在一般web项目都至少有个Controller层,以struct实现,根据不同的请求路径派发到不同的方法中去。

    路由调度器定义

    由于golang暂时还不可以动态创建对象(比如java的Class.forName("xxx").newInstance(),xxx是任意存在的class名称)。所以需要手动注册一下controller关系。

    1. 定义routes保存controller指针
    2. 解析请求过来的URL查询参数,暂定a为action名称,c为controller名称,本文偷了下懒,没对PATH_INFO做处理,也没有对actionName的首字母自动大写,这个不影响本文要传达的核心内容,有兴趣的读者可以自行实现。
    3. 根据URL中的controllerName找到对应的controller
    4. 使用反射将当前请求对象的*http.Request和http.ResponseWriter设置到该Controller
    5. 使用反射以及actionName对应该controller的方法

    由于golang的继承不是一般的OOP,所以也没有父子类这种说法,路由注册那里只能使用interface{}

    代码实现

    app/app.go

    该文件为核心调度文件

    package app

    import (
     "net/http"
     "reflect"
     "fmt"
    )
    
    type application struct {
     routes map[string]interface{}
    }
    
    func New() *application {
     return &application{
      routes: make(map[string]interface{}),
     }
    }
    
    func (p *application) ServeHTTP(w http.ResponseWriter, r *http.Request) {
     controllerName := r.URL.Query().Get("c")
     actionName := r.URL.Query().Get("a")
     if controllerName == "" || actionName == "" {
      http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
      return
     }
     route, ok := p.routes[controllerName]
     if !ok {
      http.Error(w, "Controller Not Found", http.StatusNotFound)
      return
     }
     ele := reflect.ValueOf(route).Elem()
     ele.FieldByName("Request").Set(reflect.ValueOf(r))
     ele.FieldByName("Response").Set(reflect.ValueOf(w))
     ele.MethodByName(actionName).Call([]reflect.Value{})
    }
    
    func (p *application) printRoutes() {
     for route, controller := range p.routes {
      ele := reflect.ValueOf(controller).Type().String()
      fmt.Printf("%s %s\n", route, ele)
     }
    }
    
    func (p *application) Get(route string, controller interface{}) {
     p.routes[route] = controller
    }
    
    func (p *application) Run(addr string) error {
     p.printRoutes()
     fmt.Printf("listen on %s\n", addr)
     return http.ListenAndServe(addr, p)
    }

    app/controller.go

    控制器"基类"

    package app
    
    import "net/http"
    
    type Controller struct {
     Response http.ResponseWriter
     Request *http.Request
    }

    controller/site.go

    具体业务逻辑类

    package controllers
    
    import (
     "fmt"
     "app"
    )
    
    type SiteController struct {
     app.Controller
    }
    
    func (p SiteController) Index() {
     fmt.Fprint(p.Response, p.Request.RequestURI)
    }

    main.go

    入口文件

    package main
    
    import (
     _ "github.com/go-sql-driver/mysql"
     "app"
     "controllers"
    )
    
    func main() {
     application := app.New()
     application.Get("site", &controllers.SiteController{})
     application.Run(":8080")
    }

    运行项目

    启动进程

    访问http://localhost:8080?c=site&a=Index会输出/?c=site&a=Index

    写在最后

    希望这个小小的项目能启发到各位读者,早日开发出适合自己的Web框架!

    好了,以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对站长博客的支持。

    jsjbwy
    下一篇:没有了