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

时间:2021-09-11 04:15:02

前言

本文主要介绍了关于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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
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

控制器"基类"

?
1
2
3
4
5
6
7
8
package app
 
import "net/http"
 
type Controller struct {
 Response http.ResponseWriter
 Request *http.Request
}

controller/site.go

具体业务逻辑类

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
package controllers
 
import (
 "fmt"
 "app"
)
 
type SiteController struct {
 app.Controller
}
 
func (p SiteController) Index() {
 fmt.Fprint(p.Response, p.Request.RequestURI)
}

main.go

入口文件

?
1
2
3
4
5
6
7
8
9
10
11
12
13
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框架!

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

原文链接:https://segmentfault.com/a/1190000015742742