From 6633420cf67b9966525c1a28041ae41d01f6d0a1 Mon Sep 17 00:00:00 2001
From: xiaowei <403828237@qq.com>
Date: Sun, 13 Feb 2022 11:34:46 +0800
Subject: [PATCH] =?UTF-8?q?add:=20=E6=A8=A1=E4=BB=BFgin=E6=A1=86=E6=9E=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../framework-context/gee/context.go          | 67 +++++++++++++++++++
 web/framework/framework-context/gee/gee.go    | 38 +++++++++++
 web/framework/framework-context/gee/router.go | 29 ++++++++
 web/framework/framework-context/main.go       | 22 ++++++
 web/framework/framework/gee/gee.go            |  1 +
 web/framework/framework/main.go               |  1 +
 6 files changed, 158 insertions(+)
 create mode 100644 web/framework/framework-context/gee/context.go
 create mode 100644 web/framework/framework-context/gee/gee.go
 create mode 100644 web/framework/framework-context/gee/router.go
 create mode 100644 web/framework/framework-context/main.go
 create mode 100644 web/framework/framework/gee/gee.go
 create mode 100644 web/framework/framework/main.go

diff --git a/web/framework/framework-context/gee/context.go b/web/framework/framework-context/gee/context.go
new file mode 100644
index 0000000..902d33e
--- /dev/null
+++ b/web/framework/framework-context/gee/context.go
@@ -0,0 +1,67 @@
+package gee
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+)
+
+type H map[string]interface{}
+
+type Context struct {
+	Writer     http.ResponseWriter
+	Req        *http.Request
+	Path       string
+	Method     string
+	StatusCode int
+}
+
+func newContext(w http.ResponseWriter, r *http.Request) *Context {
+	return &Context{
+		Writer: w,
+		Req:    r,
+		Path:   r.URL.Path,
+		Method: r.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))
+}
diff --git a/web/framework/framework-context/gee/gee.go b/web/framework/framework-context/gee/gee.go
new file mode 100644
index 0000000..ea7e87d
--- /dev/null
+++ b/web/framework/framework-context/gee/gee.go
@@ -0,0 +1,38 @@
+package gee
+
+import (
+	"net/http"
+)
+
+type HandlerFunc func(*Context)
+
+type Engine struct {
+	router *router
+}
+
+func New() *Engine {
+	return &Engine{
+		router: newRouter(),
+	}
+}
+
+func (e *Engine) addRoute(method string, pattern string, handler HandlerFunc) {
+	e.router.addRoute(method, pattern, handler)
+}
+
+func (e *Engine) GET(pattern string, handler HandlerFunc) {
+	e.addRoute("GET", pattern, handler)
+}
+
+func (e *Engine) POST(pattern string, handler HandlerFunc) {
+	e.addRoute("POST", pattern, handler)
+}
+
+func (e *Engine) Run(addr string) error {
+	return http.ListenAndServe(addr, e)
+}
+
+func (e *Engine) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
+	context := newContext(writer, request)
+	e.router.handle(context)
+}
diff --git a/web/framework/framework-context/gee/router.go b/web/framework/framework-context/gee/router.go
new file mode 100644
index 0000000..2d0baad
--- /dev/null
+++ b/web/framework/framework-context/gee/router.go
@@ -0,0 +1,29 @@
+package gee
+
+import (
+	"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)
+	}
+}
diff --git a/web/framework/framework-context/main.go b/web/framework/framework-context/main.go
new file mode 100644
index 0000000..d4e1740
--- /dev/null
+++ b/web/framework/framework-context/main.go
@@ -0,0 +1,22 @@
+package main
+
+import (
+	"golanglearn/web/framework/framework-context/gee"
+)
+
+func main() {
+	engine := gee.New()
+	engine.GET("/", func(c *gee.Context) {
+		c.HTML(200, "<h1>hello</h1>")
+	})
+	engine.GET("/hello", func(c *gee.Context) {
+		c.String(200, "hello,%v", c.Query("name"))
+	})
+	engine.POST("/login", func(c *gee.Context) {
+		c.JSON(200, gee.H{
+			"username": "xxxx",
+			"password": "1234",
+		})
+	})
+	_ = engine.Run(":9000")
+}
diff --git a/web/framework/framework/gee/gee.go b/web/framework/framework/gee/gee.go
new file mode 100644
index 0000000..2b44a0d
--- /dev/null
+++ b/web/framework/framework/gee/gee.go
@@ -0,0 +1 @@
+package gee
diff --git a/web/framework/framework/main.go b/web/framework/framework/main.go
new file mode 100644
index 0000000..83af475
--- /dev/null
+++ b/web/framework/framework/main.go
@@ -0,0 +1 @@
+package framework