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

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

2020-05-18 11:06xialeistudio Golang

这篇文章主要给大家介绍了关于golang利用不到20行代码实现路由调度的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

前言

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

项目地址

github

本项目依赖

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

为什么需要路由调度层

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

路由调度器定义

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

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

									}