81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
|
package framework
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Serve struct {
|
|||
|
|
Addr string
|
|||
|
|
//Handler Hander
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Core struct {
|
|||
|
|
router map[string]*Tree
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewCore() *Core {
|
|||
|
|
router := map[string]*Tree{}
|
|||
|
|
router["GET"] = NewTree()
|
|||
|
|
router["POST"] = NewTree()
|
|||
|
|
router["PUT"] = NewTree()
|
|||
|
|
router["DELETE"] = NewTree()
|
|||
|
|
|
|||
|
|
return &Core{router: router}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) Get(url string, handler ControllerHandler) {
|
|||
|
|
if err := c.router["GET"].AddRouter(url, handler); err != nil {
|
|||
|
|
log.Fatal("add router error: ", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) POST(url string, handler ControllerHandler) {
|
|||
|
|
if err := c.router["POST"].AddRouter(url, handler); err != nil {
|
|||
|
|
log.Fatal("add router error: ", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) PUT(url string, handler ControllerHandler) {
|
|||
|
|
if err := c.router["PUT"].AddRouter(url, handler); err != nil {
|
|||
|
|
log.Fatal("add router error: ", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) DELETE(url string, handler ControllerHandler) {
|
|||
|
|
if err := c.router["DELETE"].AddRouter(url, handler); err != nil {
|
|||
|
|
log.Fatal("add router error: ", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
log.Println("core.serveHTTP")
|
|||
|
|
ctx := NewContext(r, w)
|
|||
|
|
// 寻找路由
|
|||
|
|
router := c.FindRouteByRequest(r)
|
|||
|
|
if router == nil { // 如果没有找到，这里打印日志
|
|||
|
|
ctx.Json(404, "not found")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 调用路由函数，如果返回err 代表存在内部错误，返回500状态码
|
|||
|
|
if err := router(ctx); err != nil {
|
|||
|
|
ctx.Json(500, "inner error")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Core) FindRouteByRequest(request *http.Request) ControllerHandler {
|
|||
|
|
uri := request.URL.Path
|
|||
|
|
method := request.Method
|
|||
|
|
upperMethod := strings.ToUpper(method)
|
|||
|
|
upperUri := strings.ToUpper(uri)
|
|||
|
|
|
|||
|
|
if methodHandlers, ok := c.router[upperMethod]; ok {
|
|||
|
|
return methodHandlers.FindHandler(upperUri)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|