add: 模仿gin框架

This commit is contained in:
洪晓威 2022-02-13 11:46:56 +08:00
parent 6633420cf6
commit ea348b07a6
4 changed files with 71 additions and 4 deletions

View File

@ -13,9 +13,13 @@ func main() {
c.String(200, "hello,%v", c.Query("name"))
})
engine.POST("/login", func(c *gee.Context) {
//c.JSON(200, gee.H{
// "username": "xxxx",
// "password": "1234",
//})
c.JSON(200, gee.H{
"username": "xxxx",
"password": "1234",
"username": c.PostForm("username"),
"password": c.PostForm("password"),
})
})
_ = engine.Run(":9000")

View File

@ -1 +1,46 @@
package gee
import (
"fmt"
"log"
"net/http"
)
type HandlerFunc func(w http.ResponseWriter, r *http.Request)
type Engine struct {
router map[string]HandlerFunc
}
func New() *Engine {
return &Engine{
router: make(map[string]HandlerFunc),
}
}
func (e *Engine) addRoute(method string, pattern string, handler HandlerFunc) {
key := method + "-" + pattern
e.router[key] = handler
log.Println(e.router)
}
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) {
key := request.Method + "-" + request.URL.Path
if handler, ok := e.router[key]; ok {
handler(writer, request)
} else {
_, _ = fmt.Fprintf(writer, "404 not found:%v\n", request.URL)
}
}

View File

@ -1 +1,19 @@
package framework
package main
import (
"fmt"
"golanglearn/web/framework/framework/gee"
"net/http"
)
func main() {
engine := gee.New()
engine.GET("/", func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "hello web frame work")
})
engine.GET("/hello", func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "hello web frame work")
})
engine.Run(":9000")
}

View File

@ -11,7 +11,7 @@ type Engine struct {
// 实现Handler的ServerHTTP方法
func (e *Engine) ServeHTTP(w http.ResponseWriter, r *http.Request) {
switch r.RequestURI {
switch r.URL.Path {
case "/":
fmt.Fprint(w, "hello handler-basic")
case "/web":