七天实现web框架--上下文建立

对Web服务来说,无非是根据请求*http.Request,构造响应http.ResponseWriter
上下文必要性 【七天实现web框架--上下文建立】在书写web应用的时候,我们使用原始的库的时候,其中接受请求信息和返回的消息体是单独的存在,然后这样就会导致我们书写很多冗余的代码,然后在返回的时候我们需要返回json这个时候如果使用原生的库这个时候我们书写的代码是:

obj = map[string]interface{}{ "name": "geektutu", "password": "1234", } w.Header().Set("Content-Type", "application/json") w.WriteHeader(http.StatusOK) encoder := json.NewEncoder(w) if err := encoder.Encode(obj); err != nil { http.Error(w, err.Error(), 500) }

然后如果我们封装了过后我们书写的代码是:
c.JSON(http.StatusOK, gee.H{ "username": c.PostForm("username"), "password": c.PostForm("password"), })

这里我们可以明显的感受到我们的代码简单了很多,如果没有进行封装,这个时候我们使用起来是非常难受的。在这里我们需要知道的是,其实在web应用中,我们需要很多的工具,在处理请求的时候,例如,将来解析动态路由/hello/:name,参数:name的值放在哪呢?再比如,框架需要支持中间件,那中间件产生的信息放在哪呢?Context 随着每一个请求的出现而产生,请求的结束而销毁,和当前请求强相关的信息都应由 Context 承载。
也就是在这次会话的时候context就是一个会话的百宝箱,这个时候我们可以找到我们处理这次会话的所有工具。
实现代码
type H map[string]interface{}type Context struct { // origin objects Writer http.ResponseWriter Req*http.Request // request info Pathstring Method string // response info StatusCode int }func newContext(w http.ResponseWriter, req *http.Request) *Context { return &Context{ Writer: w, Req:req, Path:req.URL.Path, Method: req.Method, } }func (c *Context) PostForm(key string) string { return c.Req.FormValue(key) }func (c *Context) Query(key string) string { return c.Req.URL.Query().Get(key) }func (c *Context) Status(code int) { c.StatusCode = code c.Writer.WriteHeader(code) }func (c *Context) SetHeader(key string, value string) { c.Writer.Header().Set(key, value) }func (c *Context) String(code int, format string, values ...interface{}) { c.SetHeader("Content-Type", "text/plain") c.Status(code) c.Writer.Write([]byte(fmt.Sprintf(format, values...))) }func (c *Context) JSON(code int, obj interface{}) { c.SetHeader("Content-Type", "application/json") c.Status(code) encoder := json.NewEncoder(c.Writer) if err := encoder.Encode(obj); err != nil { http.Error(c.Writer, err.Error(), 500) } }func (c *Context) Data(code int, data []byte) { c.Status(code) c.Writer.Write(data) }func (c *Context) HTML(code int, html string) { c.SetHeader("Content-Type", "text/html") c.Status(code) c.Writer.Write([]byte(html)) }

在这里面我们将很多的工具进行了封装
代码最开头,给map[string]interface{}起了一个别名gee.H,构建JSON数据时,显得更简洁。
Context目前只包含了http.ResponseWriter和*http.Request,另外提供了对 Method 和 Path 这两个常用属性的直接访问。
提供了访问Query和PostForm参数的方法。
提供了快速构造String/Data/JSON/HTML响应的方法。
对处理函数的改装 昨天我们进行了路由解析的书写,我们知道了继承了handler后我们就可以进行请求的处理,然后我们将处理的函数进行一个改装
package geeimport ( "log" "net/http" )type router struct { handlers map[string]HandlerFunc }func newRouter() *router { return &router{handlers: make(map[string]HandlerFunc)} }func (r *router) addRoute(method string, pattern string, handler HandlerFunc) { log.Printf("Route %4s - %s", method, pattern) key := method + "-" + pattern r.handlers[key] = handler }func (r *router) handle(c *Context) { key := c.Method + "-" + c.Path if handler, ok := r.handlers[key]; ok { handler(c) } else { c.String(http.StatusNotFound, "404 NOT FOUND: %s\n", c.Path) } }

我们将增加处理函数的方法放在了这里,然后对昨天的代码进行一个改装
package geeimport ( "net/http" )// HandlerFunc defines the request handler used by gee type HandlerFunc func(*Context)// Engine implement the interface of ServeHTTP type Engine struct { router *router }// New is the constructor of gee.Engine //这里进行了router的创建 func New() *Engine { return &Engine{router: newRouter()} }func (engine *Engine) addRoute(method string, pattern string, handler HandlerFunc) { engine.router.addRoute(method, pattern, handler) }// GET defines the method to add GET request func (engine *Engine) GET(pattern string, handler HandlerFunc) { engine.addRoute("GET", pattern, handler) }// POST defines the method to add POST request func (engine *Engine) POST(pattern string, handler HandlerFunc) { engine.addRoute("POST", pattern, handler) }// Run defines the method to start a http server func (engine *Engine) Run(addr string) (err error) { return http.ListenAndServe(addr, engine) }//这里就是最后的处理函数,这里开始建立的上下文。 func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) { c := newContext(w, req) engine.router.handle(c) }

在这里我们结合了今天写的context进行了一个改装。 然后我们今天的上下文建立就建立好了。

    推荐阅读