Gee微框架学习记录(第二天)

// map[string]interface{} 设置别名，够减JSON更方便
type H map[string]interface{}

// 定义Context 结构体，方便后续拓展
// 加入相关参数，冗余Path 和 Method、StatusCode方便处理
type Context struct {
	// origin objects
	Writer http.ResponseWriter
	Req    *http.Request
	// request info
	Path   string
	Method string
	// response info
	StatusCode int
}

// new Context 方法，给相关参数赋初值
func newContext(w http.ResponseWriter, req *http.Request) *Context {
	return &Context{
		Writer: w,
		Req:    req,
		Path:   req.URL.Path,
		Method: req.Method,
	}
}

// 以下都是实现Context的方法，方便请求处理数据

// 表单处理 如：账号密码提交
func (c *Context) PostForm(key string) string {
	return c.Req.FormValue(key)
}

// 从 URL ? 后的查询参数中获取 key 对应的值 如 /home?q=golang&page=1
func (c *Context) Query(key string) string {
	return c.Req.URL.Query().Get(key)
}

// Status赋值
func (c *Context) Status(code int) {
	c.StatusCode = code
	c.Writer.WriteHeader(code)
}

// 设置Header
func (c *Context) SetHeader(key string, value string) {
	c.Writer.Header().Set(key, value)
}

// String方法处理
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...)))
}

// JSON 方法处理
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)
}

// HTML内容处理
func (c *Context) HTML(code int, html string) {
	c.SetHeader("Content-Type", "text/html")
	c.Status(code)
	c.Writer.Write([]byte(html))
}

// 定义路由结构体，内部为map数据结构对应的路由映射表
type router struct {
	handlers map[string]HandlerFunc
}

// new 方法，初始化router内部参数
func newRouter() *router {
	return &router{handlers: make(map[string]HandlerFunc)}
}

// addRoute方法变更
func (r *router) addRoute(method string, pattern string, handler HandlerFunc) {
	log.Printf("Route %4s - %s", method, pattern)
	key := method + "-" + pattern
	r.handlers[key] = handler
}

// handle 参数变更 重点，这里原来是ServeHTTP的处理逻辑。被提取出来到这里作为单独方法使用。
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)
	}
}

// HandlerFunc defines the request handler used by gee
// HandlerFunc参数更改为 Context
type HandlerFunc func(*Context)

// Engine implement the interface of ServeHTTP
// 原来的router 映射表改为 定义的rouer结构体
type Engine struct {
	router *router
}

// New is the constructor of gee.Engine
// new 方法也要做更新
func New() *Engine {
	return &Engine{router: newRouter()}
}

// engine 的 addRoute更新成调用router的addRoute方法，简化了
func (engine *Engine) addRoute(method string, pattern string, handler HandlerFunc) {
	engine.router.addRoute(method, pattern, handler)
}

// 省略 GET 和 POST 、Run 无变更

// 由于抽离了handle到router，这里直接new Context 后使用router的handle就可以了
func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
	c := newContext(w, req)
	engine.router.handle(c)
}

func main() {
	r := gee.New()
    // Handlerfunc 参数变更成Context
	r.GET("/", func(c *gee.Context) {
        // 调用Context的HTML响应方法
		c.HTML(http.StatusOK, "<h1>Hello Gee</h1>")
	})
    // Handlerfunc 参数变更Context
	r.GET("/hello", func(c *gee.Context) {
		// expect /hello?name=geektutu
        // 调用Context的Query响应方法
		c.String(http.StatusOK, "hello %s, you're at %s\n", c.Query("name"), c.Path)
	})
	// Handlerfunc 参数变更Context
	r.POST("/login", func(c *gee.Context) {
        // 调用Context的JSON响应方法，使用设置的别名新建一个JSON
		c.JSON(http.StatusOK, gee.H{
			"username": c.PostForm("username"),
			"password": c.PostForm("password"),
		})
	})

	r.Run(":9999")
}