gin/gin.go

371 lines
11 KiB
Go
Raw Normal View History

2014-08-29 17:49:50 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
2014-06-17 23:42:34 +00:00
package gin
import (
"html/template"
2015-05-18 22:45:08 +00:00
"net"
2014-06-17 23:42:34 +00:00
"net/http"
2015-05-18 22:45:08 +00:00
"os"
2014-07-06 19:09:23 +00:00
"sync"
2015-03-23 03:41:29 +00:00
"github.com/gin-gonic/gin/render"
2014-06-17 23:42:34 +00:00
)
2016-04-14 23:16:46 +00:00
// Version is Framework's version
const Version = "v1.1.4"
2015-05-22 14:55:16 +00:00
2015-03-31 19:39:06 +00:00
var default404Body = []byte("404 page not found")
var default405Body = []byte("405 method not allowed")
2014-06-17 23:42:34 +00:00
2015-06-26 14:05:09 +00:00
type HandlerFunc func(*Context)
type HandlersChain []HandlerFunc
// Last returns the last handler in the chain. ie. the last handler is the main own.
2015-06-26 14:05:09 +00:00
func (c HandlersChain) Last() HandlerFunc {
length := len(c)
if length > 0 {
return c[length-1]
}
return nil
}
2014-06-17 23:42:34 +00:00
type (
2015-06-26 14:05:09 +00:00
RoutesInfo []RouteInfo
RouteInfo struct {
Method string
Path string
Handler string
}
2014-06-17 23:42:34 +00:00
// Engine is the framework's instance, it contains the muxer, middleware and configuration settings.
// Create an instance of Engine, by using New() or Default()
2014-06-17 23:42:34 +00:00
Engine struct {
2015-04-07 10:22:38 +00:00
RouterGroup
2015-05-18 13:45:24 +00:00
HTMLRender render.HTMLRender
2015-05-07 09:30:01 +00:00
allNoRoute HandlersChain
allNoMethod HandlersChain
noRoute HandlersChain
noMethod HandlersChain
2015-05-29 19:03:28 +00:00
pool sync.Pool
trees methodTrees
2015-03-31 19:39:06 +00:00
// Enables automatic redirection if the current route can't be matched but a
// handler for the path with (without) the trailing slash exists.
// For example if /foo/ is requested but a route only exists for /foo, the
// client is redirected to /foo with http status code 301 for GET requests
// and 307 for all other request methods.
RedirectTrailingSlash bool
// If enabled, the router tries to fix the current request path, if no
// handle is registered for it.
// First superfluous path elements like ../ or // are removed.
// Afterwards the router does a case-insensitive lookup of the cleaned path.
// If a handle can be found for this route, the router makes a redirection
// to the corrected path with status code 301 for GET requests and 307 for
// all other request methods.
// For example /FOO and /..//Foo could be redirected to /foo.
// RedirectTrailingSlash is independent of this option.
RedirectFixedPath bool
// If enabled, the router checks if another method is allowed for the
// current route, if the current request can not be routed.
// If this is the case, the request is answered with 'Method Not Allowed'
// and HTTP status code 405.
// If no other Method is allowed, the request is delegated to the NotFound
// handler.
HandleMethodNotAllowed bool
2015-06-07 11:51:13 +00:00
ForwardedByClientIP bool
2014-06-17 23:42:34 +00:00
}
)
var _ IRouter = &Engine{}
2015-06-26 14:01:35 +00:00
// New returns a new blank Engine instance without any middleware attached.
// By default the configuration is:
// - RedirectTrailingSlash: true
// - RedirectFixedPath: false
// - HandleMethodNotAllowed: false
// - ForwardedByClientIP: true
2014-07-06 19:09:23 +00:00
func New() *Engine {
debugPrintWARNINGNew()
2015-03-31 19:39:06 +00:00
engine := &Engine{
2015-04-07 10:22:38 +00:00
RouterGroup: RouterGroup{
Handlers: nil,
basePath: "/",
2015-06-10 23:02:38 +00:00
root: true,
2015-04-07 10:22:38 +00:00
},
2015-03-31 19:39:06 +00:00
RedirectTrailingSlash: true,
RedirectFixedPath: false,
HandleMethodNotAllowed: false,
2015-06-07 11:51:13 +00:00
ForwardedByClientIP: true,
trees: make(methodTrees, 0, 9),
2015-03-31 19:39:06 +00:00
}
2015-04-07 10:22:38 +00:00
engine.RouterGroup.engine = engine
2014-10-08 19:37:26 +00:00
engine.pool.New = func() interface{} {
2015-03-25 18:33:17 +00:00
return engine.allocateContext()
}
2014-06-17 23:42:34 +00:00
return engine
}
// Default returns an Engine instance with the Logger and Recovery middleware already attached.
2014-06-17 23:42:34 +00:00
func Default() *Engine {
engine := New()
engine.Use(Logger(), Recovery())
2014-06-17 23:42:34 +00:00
return engine
}
2015-06-07 11:51:13 +00:00
func (engine *Engine) allocateContext() *Context {
return &Context{engine: engine}
2015-03-25 18:33:17 +00:00
}
2014-07-15 15:41:56 +00:00
func (engine *Engine) LoadHTMLGlob(pattern string) {
2014-10-08 19:37:26 +00:00
if IsDebugging() {
2015-08-03 15:28:12 +00:00
debugPrintLoadTemplate(template.Must(template.ParseGlob(pattern)))
2015-05-18 13:45:24 +00:00
engine.HTMLRender = render.HTMLDebug{Glob: pattern}
2014-08-20 23:04:35 +00:00
} else {
templ := template.Must(template.ParseGlob(pattern))
engine.SetHTMLTemplate(templ)
}
2014-07-15 15:41:56 +00:00
}
func (engine *Engine) LoadHTMLFiles(files ...string) {
2014-10-08 19:37:26 +00:00
if IsDebugging() {
2015-05-18 13:45:24 +00:00
engine.HTMLRender = render.HTMLDebug{Files: files}
} else {
2014-08-20 23:04:35 +00:00
templ := template.Must(template.ParseFiles(files...))
engine.SetHTMLTemplate(templ)
}
}
func (engine *Engine) SetHTMLTemplate(templ *template.Template) {
2015-06-12 16:09:44 +00:00
if len(engine.trees) > 0 {
debugPrintWARNINGSetHTMLTemplate()
2015-06-12 16:09:44 +00:00
}
2015-05-18 13:45:24 +00:00
engine.HTMLRender = render.HTMLProduction{Template: templ}
2014-06-17 23:42:34 +00:00
}
2016-04-14 23:16:46 +00:00
// NoRoute adds handlers for NoRoute. It return a 404 code by default.
2014-07-17 21:42:23 +00:00
func (engine *Engine) NoRoute(handlers ...HandlerFunc) {
2014-07-17 22:29:44 +00:00
engine.noRoute = handlers
2014-10-08 19:37:26 +00:00
engine.rebuild404Handlers()
2014-07-17 22:29:44 +00:00
}
2016-04-14 23:16:46 +00:00
// NoMethod sets the handlers called when... TODO
func (engine *Engine) NoMethod(handlers ...HandlerFunc) {
engine.noMethod = handlers
engine.rebuild405Handlers()
}
2016-04-14 23:16:46 +00:00
// Use attachs a global middleware to the router. ie. the middleware attached though Use() will be
2015-05-29 19:03:41 +00:00
// included in the handlers chain for every single request. Even 404, 405, static files...
// For example, this is the right place for a logger or error management middleware.
func (engine *Engine) Use(middleware ...HandlerFunc) IRoutes {
engine.RouterGroup.Use(middleware...)
2014-10-08 19:37:26 +00:00
engine.rebuild404Handlers()
engine.rebuild405Handlers()
2015-06-10 23:02:38 +00:00
return engine
2014-10-08 19:37:26 +00:00
}
func (engine *Engine) rebuild404Handlers() {
2015-03-25 15:53:58 +00:00
engine.allNoRoute = engine.combineHandlers(engine.noRoute)
}
func (engine *Engine) rebuild405Handlers() {
2015-03-25 15:53:58 +00:00
engine.allNoMethod = engine.combineHandlers(engine.noMethod)
2014-06-17 23:42:34 +00:00
}
2015-05-19 21:22:35 +00:00
func (engine *Engine) addRoute(method, path string, handlers HandlersChain) {
2016-01-27 23:35:09 +00:00
assert1(path[0] == '/', "path must begin with '/'")
assert1(len(method) > 0, "HTTP method can not be empty")
assert1(len(handlers) > 0, "there must be at least one handler")
2015-04-09 10:15:02 +00:00
2016-01-27 23:35:09 +00:00
debugPrintRoute(method, path, handlers)
2015-06-03 23:54:36 +00:00
root := engine.trees.get(method)
2015-03-31 19:39:06 +00:00
if root == nil {
root = new(node)
2016-01-27 23:35:09 +00:00
engine.trees = append(engine.trees, methodTree{method: method, root: root})
2015-03-31 19:39:06 +00:00
}
root.addRoute(path, handlers)
}
// Routes returns a slice of registered routes, including some useful information, such as:
// the http method, path and the handler name.
2015-06-18 15:17:22 +00:00
func (engine *Engine) Routes() (routes RoutesInfo) {
2015-06-07 02:20:39 +00:00
for _, tree := range engine.trees {
routes = iterate("", tree.method, routes, tree.root)
2015-06-07 02:20:39 +00:00
}
return routes
}
2015-06-18 15:17:22 +00:00
func iterate(path, method string, routes RoutesInfo, root *node) RoutesInfo {
2015-06-07 02:20:39 +00:00
path += root.path
if len(root.handlers) > 0 {
routes = append(routes, RouteInfo{
Method: method,
Path: path,
Handler: nameOfFunction(root.handlers.Last()),
})
2015-06-07 02:20:39 +00:00
}
2015-07-02 16:42:33 +00:00
for _, child := range root.children {
routes = iterate(path, method, routes, child)
2015-06-07 02:20:39 +00:00
}
return routes
}
// Run attaches the router to a http.Server and starts listening and serving HTTP requests.
2015-05-29 19:03:41 +00:00
// It is a shortcut for http.ListenAndServe(addr, router)
// Note: this method will block the calling goroutine indefinitely unless an error happens.
func (engine *Engine) Run(addr ...string) (err error) {
2015-05-18 22:45:08 +00:00
defer func() { debugPrintError(err) }()
2015-05-09 01:34:43 +00:00
address := resolveAddress(addr)
debugPrint("Listening and serving HTTP on %s\n", address)
err = http.ListenAndServe(address, engine)
2015-05-09 01:34:43 +00:00
return
2015-04-07 10:22:38 +00:00
}
// RunTLS attaches the router to a http.Server and starts listening and serving HTTPS (secure) requests.
2015-05-29 19:03:41 +00:00
// It is a shortcut for http.ListenAndServeTLS(addr, certFile, keyFile, router)
// Note: this method will block the calling goroutine indefinitely unless an error happens.
func (engine *Engine) RunTLS(addr string, certFile string, keyFile string) (err error) {
2015-04-07 10:22:38 +00:00
debugPrint("Listening and serving HTTPS on %s\n", addr)
2015-05-18 22:45:08 +00:00
defer func() { debugPrintError(err) }()
2015-05-09 01:34:43 +00:00
err = http.ListenAndServeTLS(addr, certFile, keyFile, engine)
2015-05-09 01:34:43 +00:00
return
2015-04-07 10:22:38 +00:00
}
// RunUnix attaches the router to a http.Server and starts listening and serving HTTP requests
// through the specified unix socket (ie. a file).
// Note: this method will block the calling goroutine indefinitely unless an error happens.
2015-05-18 22:45:08 +00:00
func (engine *Engine) RunUnix(file string) (err error) {
debugPrint("Listening and serving HTTP on unix:/%s", file)
defer func() { debugPrintError(err) }()
os.Remove(file)
2015-05-18 22:48:19 +00:00
listener, err := net.Listen("unix", file)
2015-05-18 22:45:08 +00:00
if err != nil {
return
}
defer listener.Close()
err = http.Serve(listener, engine)
return
}
2015-05-29 19:03:41 +00:00
// Conforms to the http.Handler interface.
2015-03-31 19:39:06 +00:00
func (engine *Engine) ServeHTTP(w http.ResponseWriter, req *http.Request) {
2015-05-18 18:50:46 +00:00
c := engine.pool.Get().(*Context)
c.writermem.reset(w)
c.Request = req
c.reset()
2015-04-07 10:22:38 +00:00
2015-05-28 01:22:34 +00:00
engine.handleHTTPRequest(c)
2015-05-18 18:50:46 +00:00
engine.pool.Put(c)
2015-04-07 10:22:38 +00:00
}
2015-05-18 18:50:46 +00:00
func (engine *Engine) handleHTTPRequest(context *Context) {
2015-04-07 10:22:38 +00:00
httpMethod := context.Request.Method
path := context.Request.URL.Path
// Find root of the tree for the given HTTP method
2015-05-29 19:03:28 +00:00
t := engine.trees
for i, tl := 0, len(t); i < tl; i++ {
if t[i].method == httpMethod {
root := t[i].root
2015-05-29 19:03:28 +00:00
// Find route in tree
handlers, params, tsr := root.getValue(path, context.Params)
2015-05-29 19:03:28 +00:00
if handlers != nil {
context.handlers = handlers
context.Params = params
context.Next()
context.writermem.WriteHeaderNow()
2015-04-07 10:22:38 +00:00
return
2015-05-29 19:03:28 +00:00
} else if httpMethod != "CONNECT" && path != "/" {
if tsr && engine.RedirectTrailingSlash {
redirectTrailingSlash(context)
return
}
if engine.RedirectFixedPath && redirectFixedPath(context, root, engine.RedirectFixedPath) {
2015-05-29 19:03:28 +00:00
return
}
2015-04-07 10:22:38 +00:00
}
2015-06-03 23:54:36 +00:00
break
2015-03-31 19:39:06 +00:00
}
}
2015-05-29 19:03:28 +00:00
// TODO: unit test
2015-04-07 10:22:38 +00:00
if engine.HandleMethodNotAllowed {
2015-05-29 19:03:28 +00:00
for _, tree := range engine.trees {
if tree.method != httpMethod {
if handlers, _, _ := tree.root.getValue(path, nil); handlers != nil {
2015-04-07 10:22:38 +00:00
context.handlers = engine.allNoMethod
serveError(context, 405, default405Body)
return
}
}
}
}
2015-04-08 12:24:49 +00:00
context.handlers = engine.allNoRoute
2015-04-07 10:22:38 +00:00
serveError(context, 404, default404Body)
2014-06-17 23:42:34 +00:00
}
var mimePlain = []string{MIMEPlain}
func serveError(c *Context, code int, defaultMessage []byte) {
c.writermem.status = code
c.Next()
if !c.writermem.Written() {
if c.writermem.Status() == code {
c.writermem.Header()["Content-Type"] = mimePlain
c.Writer.Write(defaultMessage)
} else {
c.writermem.WriteHeaderNow()
}
}
}
func redirectTrailingSlash(c *Context) {
2015-04-07 10:22:38 +00:00
req := c.Request
path := req.URL.Path
code := 301 // Permanent redirect, request with GET method
if req.Method != "GET" {
code = 307
}
if len(path) > 1 && path[len(path)-1] == '/' {
req.URL.Path = path[:len(path)-1]
} else {
req.URL.Path = path + "/"
}
debugPrint("redirecting request %d: %s --> %s", code, path, req.URL.String())
http.Redirect(c.Writer, req, req.URL.String(), code)
c.writermem.WriteHeaderNow()
}
func redirectFixedPath(c *Context, root *node, trailingSlash bool) bool {
req := c.Request
path := req.URL.Path
fixedPath, found := root.findCaseInsensitivePath(
cleanPath(path),
trailingSlash,
)
if found {
code := 301 // Permanent redirect, request with GET method
if req.Method != "GET" {
code = 307
2015-04-07 10:22:38 +00:00
}
req.URL.Path = string(fixedPath)
debugPrint("redirecting request %d: %s --> %s", code, path, req.URL.String())
2015-04-07 10:22:38 +00:00
http.Redirect(c.Writer, req, req.URL.String(), code)
2015-05-05 14:37:33 +00:00
c.writermem.WriteHeaderNow()
2015-04-07 10:22:38 +00:00
return true
}
return false
2014-06-17 23:42:34 +00:00
}