mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-23 08:56:29 +00:00
Merge pull request #2146 from bradrydzewski/master
abstract the user interface
This commit is contained in:
commit
1b79d1fab0
20 changed files with 2008 additions and 174 deletions
|
@ -72,6 +72,12 @@ var flags = []cli.Flag{
|
|||
Name: "quic",
|
||||
Usage: "start the server with quic enabled",
|
||||
},
|
||||
cli.StringFlag{
|
||||
EnvVar: "DRONE_WWW",
|
||||
Name: "www",
|
||||
Usage: "serve the website from disk",
|
||||
Hidden: true,
|
||||
},
|
||||
cli.StringSliceFlag{
|
||||
EnvVar: "DRONE_ADMIN",
|
||||
Name: "admin",
|
||||
|
@ -463,8 +469,13 @@ func server(c *cli.Context) error {
|
|||
store_ := setupStore(c)
|
||||
setupEvilGlobals(c, store_, remote_)
|
||||
|
||||
// we are switching from gin to httpservermux|treemux,
|
||||
// so if this code looks strange, that is why.
|
||||
tree := setupTree(c)
|
||||
|
||||
// setup the server and start the listener
|
||||
handler := router.Load(
|
||||
tree,
|
||||
ginrus.Ginrus(logrus.StandardLogger(), time.RFC3339, true),
|
||||
middleware.Version,
|
||||
middleware.Config(c),
|
||||
|
|
|
@ -4,6 +4,7 @@ import (
|
|||
"fmt"
|
||||
|
||||
"github.com/cncd/queue"
|
||||
"github.com/dimfeld/httptreemux"
|
||||
"github.com/drone/drone/model"
|
||||
"github.com/drone/drone/plugins/registry"
|
||||
"github.com/drone/drone/plugins/secrets"
|
||||
|
@ -15,6 +16,7 @@ import (
|
|||
"github.com/drone/drone/remote/github"
|
||||
"github.com/drone/drone/remote/gitlab"
|
||||
"github.com/drone/drone/remote/gogs"
|
||||
"github.com/drone/drone/server/web"
|
||||
"github.com/drone/drone/store"
|
||||
"github.com/drone/drone/store/datastore"
|
||||
|
||||
|
@ -156,4 +158,14 @@ func setupCoding(c *cli.Context) (remote.Remote, error) {
|
|||
})
|
||||
}
|
||||
|
||||
func setupTree(c *cli.Context) *httptreemux.ContextMux {
|
||||
tree := httptreemux.NewContextMux()
|
||||
if path := c.String("www"); path == "" {
|
||||
web.New().Register(tree)
|
||||
} else {
|
||||
web.FromPath(path).Register(tree)
|
||||
}
|
||||
return tree
|
||||
}
|
||||
|
||||
func before(c *cli.Context) error { return nil }
|
||||
|
|
|
@ -3,6 +3,7 @@ package router
|
|||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/dimfeld/httptreemux"
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"github.com/drone/drone/router/middleware/header"
|
||||
|
@ -12,21 +13,22 @@ import (
|
|||
"github.com/drone/drone/server/debug"
|
||||
"github.com/drone/drone/server/metrics"
|
||||
"github.com/drone/drone/server/template"
|
||||
"github.com/drone/drone/server/web"
|
||||
)
|
||||
|
||||
// Load loads the router
|
||||
func Load(middleware ...gin.HandlerFunc) http.Handler {
|
||||
func Load(mux *httptreemux.ContextMux, middleware ...gin.HandlerFunc) http.Handler {
|
||||
|
||||
e := gin.New()
|
||||
e.Use(gin.Recovery())
|
||||
e.SetHTMLTemplate(template.T)
|
||||
|
||||
ui := server.NewWebsite()
|
||||
for _, path := range ui.Routes() {
|
||||
e.GET(path, func(c *gin.Context) {
|
||||
ui.File(c.Writer, c.Request)
|
||||
})
|
||||
}
|
||||
// ui := server.NewWebsite()
|
||||
// for _, path := range ui.Routes() {
|
||||
// e.GET(path, func(c *gin.Context) {
|
||||
// ui.File(c.Writer, c.Request)
|
||||
// })
|
||||
// }
|
||||
|
||||
e.Use(header.NoCache)
|
||||
e.Use(header.Options)
|
||||
|
@ -35,12 +37,19 @@ func Load(middleware ...gin.HandlerFunc) http.Handler {
|
|||
e.Use(session.SetUser())
|
||||
e.Use(token.Refresh)
|
||||
|
||||
e.GET("/logout", server.GetLogout)
|
||||
e.NoRoute(func(c *gin.Context) {
|
||||
u := session.User(c)
|
||||
ui.Page(c.Writer, c.Request, u)
|
||||
req := c.Request.WithContext(
|
||||
web.WithUser(
|
||||
c.Request.Context(),
|
||||
session.User(c),
|
||||
),
|
||||
)
|
||||
mux.ServeHTTP(c.Writer, req)
|
||||
})
|
||||
|
||||
e.GET("/logout", server.GetLogout)
|
||||
e.GET("/login", server.HandleLogin)
|
||||
|
||||
user := e.Group("/api/user")
|
||||
{
|
||||
user.Use(session.MustUser())
|
||||
|
@ -143,8 +152,8 @@ func Load(middleware ...gin.HandlerFunc) http.Handler {
|
|||
|
||||
auth := e.Group("/authorize")
|
||||
{
|
||||
auth.GET("", server.GetLogin)
|
||||
auth.POST("", server.GetLogin)
|
||||
auth.GET("", server.HandleAuth)
|
||||
auth.POST("", server.HandleAuth)
|
||||
auth.POST("/token", server.GetLoginToken)
|
||||
}
|
||||
|
||||
|
|
|
@ -145,7 +145,7 @@ func PostHook(c *gin.Context) {
|
|||
// fetch the build file from the database
|
||||
confb, err := remote_.File(user, repo, build, repo.Config)
|
||||
if err != nil {
|
||||
logrus.Errorf("failure to get build config for %s. %s", repo.FullName, err)
|
||||
logrus.Errorf("error: %s: cannot find %s in %s: %s", repo.FullName, repo.Config, build.Ref, err)
|
||||
c.AbortWithError(404, err)
|
||||
return
|
||||
}
|
||||
|
|
|
@ -16,7 +16,19 @@ import (
|
|||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func GetLogin(c *gin.Context) {
|
||||
func HandleLogin(c *gin.Context) {
|
||||
var (
|
||||
w = c.Writer
|
||||
r = c.Request
|
||||
)
|
||||
if err := r.FormValue("error"); err != "" {
|
||||
http.Redirect(w, r, "/login/error?code="+err, 303)
|
||||
} else {
|
||||
http.Redirect(w, r, "/authorize", 303)
|
||||
}
|
||||
}
|
||||
|
||||
func HandleAuth(c *gin.Context) {
|
||||
|
||||
// when dealing with redirects we may need to adjust the content type. I
|
||||
// cannot, however, remember why, so need to revisit this line.
|
||||
|
|
79
server/ui.go
79
server/ui.go
|
@ -1,79 +0,0 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
"github.com/drone/drone-ui/dist"
|
||||
"github.com/drone/drone/model"
|
||||
"github.com/drone/drone/server/template"
|
||||
"github.com/drone/drone/shared/token"
|
||||
)
|
||||
|
||||
// Website defines an interface to serve the user interface.
|
||||
type Website interface {
|
||||
Page(rw http.ResponseWriter, r *http.Request, u *model.User)
|
||||
File(rw http.ResponseWriter, r *http.Request)
|
||||
Routes() []string
|
||||
}
|
||||
|
||||
type website struct {
|
||||
fs http.Handler
|
||||
}
|
||||
|
||||
// NewWebsite returns a new website loader.
|
||||
func NewWebsite() Website {
|
||||
// TODO change to DRONE_WEB_PATH and add DRONE_WEB_PROXY
|
||||
path := os.Getenv("DRONE_WWW")
|
||||
if path != "" {
|
||||
return NewLocalWebsite(path)
|
||||
}
|
||||
return &website{
|
||||
fs: http.FileServer(dist.New()),
|
||||
}
|
||||
}
|
||||
|
||||
// Page serves a page in the user interface.
|
||||
func (w *website) Page(rw http.ResponseWriter, r *http.Request, u *model.User) {
|
||||
rw.WriteHeader(200)
|
||||
|
||||
path := r.URL.Path
|
||||
switch path {
|
||||
case "/login":
|
||||
if err := r.FormValue("error"); err != "" {
|
||||
// TODO login error
|
||||
} else {
|
||||
http.Redirect(rw, r, "/authorize", 303)
|
||||
}
|
||||
|
||||
default:
|
||||
var csrf string
|
||||
if u != nil {
|
||||
csrf, _ = token.New(
|
||||
token.CsrfToken,
|
||||
u.Login,
|
||||
).Sign(u.Hash)
|
||||
}
|
||||
params := map[string]interface{}{
|
||||
"user": u,
|
||||
"csrf": csrf,
|
||||
}
|
||||
|
||||
rw.Header().Set("Content-Type", "text/html; charset=UTF-8")
|
||||
template.T.ExecuteTemplate(rw, "index_polymer.html", params)
|
||||
}
|
||||
}
|
||||
|
||||
// File serves a static file for the user interface.
|
||||
func (w *website) File(rw http.ResponseWriter, r *http.Request) {
|
||||
w.fs.ServeHTTP(rw, r)
|
||||
}
|
||||
|
||||
func (w *website) Routes() []string {
|
||||
return []string{
|
||||
"/favicon-32x32.png",
|
||||
"/favicon-16x16.png",
|
||||
"/src/*filepath",
|
||||
"/bower_components/*filepath",
|
||||
}
|
||||
}
|
|
@ -1,80 +0,0 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/drone/drone/model"
|
||||
"github.com/drone/drone/server/template"
|
||||
"github.com/drone/drone/shared/token"
|
||||
)
|
||||
|
||||
type local struct {
|
||||
dir string
|
||||
fs http.Handler
|
||||
}
|
||||
|
||||
// NewLocalWebsite returns a new website loader.
|
||||
func NewLocalWebsite(path string) Website {
|
||||
return &local{
|
||||
dir: path,
|
||||
fs: http.FileServer(
|
||||
http.Dir(path),
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
// Page serves a page in the user interface.
|
||||
func (w *local) Page(rw http.ResponseWriter, r *http.Request, u *model.User) {
|
||||
rw.WriteHeader(200)
|
||||
|
||||
path := r.URL.Path
|
||||
switch path {
|
||||
case "/login":
|
||||
if err := r.FormValue("error"); err != "" {
|
||||
// TODO login error
|
||||
} else {
|
||||
http.Redirect(rw, r, "/authorize", 303)
|
||||
}
|
||||
|
||||
default:
|
||||
var csrf string
|
||||
if u != nil {
|
||||
csrf, _ = token.New(
|
||||
token.CsrfToken,
|
||||
u.Login,
|
||||
).Sign(u.Hash)
|
||||
}
|
||||
params := map[string]interface{}{
|
||||
"user": u,
|
||||
"csrf": csrf,
|
||||
}
|
||||
|
||||
index, err := ioutil.ReadFile(filepath.Join(w.dir, "index.html"))
|
||||
if err != nil {
|
||||
rw.WriteHeader(404)
|
||||
return
|
||||
}
|
||||
var buf bytes.Buffer
|
||||
template.T.ExecuteTemplate(&buf, "script.html", params)
|
||||
index = bytes.Replace(index, []byte("<!-- inject:js -->"), buf.Bytes(), 1)
|
||||
rw.Write(index)
|
||||
}
|
||||
}
|
||||
|
||||
// File serves a static file for the user interface.
|
||||
func (w *local) File(rw http.ResponseWriter, r *http.Request) {
|
||||
w.fs.ServeHTTP(rw, r)
|
||||
}
|
||||
|
||||
func (w *local) Routes() []string {
|
||||
return []string{
|
||||
"/favicon-32x32.png",
|
||||
"/favicon-16x16.png",
|
||||
"/src/*filepath",
|
||||
"/bower_components/*filepath",
|
||||
"/static/*filepath",
|
||||
}
|
||||
}
|
65
server/web/local.go
Normal file
65
server/web/local.go
Normal file
|
@ -0,0 +1,65 @@
|
|||
package web
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/drone/drone/server/template"
|
||||
"github.com/drone/drone/shared/token"
|
||||
"github.com/drone/drone/version"
|
||||
|
||||
"github.com/dimfeld/httptreemux"
|
||||
)
|
||||
|
||||
// FromPath returns the default website endpoint served from a local path.
|
||||
func FromPath(path string) Endpoint {
|
||||
return &local{path}
|
||||
}
|
||||
|
||||
type local struct {
|
||||
dir string
|
||||
}
|
||||
|
||||
func (l *local) Register(mux *httptreemux.ContextMux) {
|
||||
h := http.FileServer(
|
||||
http.Dir(l.dir),
|
||||
)
|
||||
h = resetCache(h)
|
||||
mux.Handler("GET", "/favicon-32x32.png", h)
|
||||
mux.Handler("GET", "/favicon-16x16.png", h)
|
||||
mux.Handler("GET", "/src/*filepath", h)
|
||||
mux.Handler("GET", "/static/*filepath", h)
|
||||
mux.Handler("GET", "/bundle/*filepath", h)
|
||||
mux.Handler("GET", "/bower_components/*filepath", h)
|
||||
mux.NotFoundHandler = l.handleIndexLocal
|
||||
}
|
||||
|
||||
func (l *local) handleIndexLocal(rw http.ResponseWriter, r *http.Request) {
|
||||
var csrf string
|
||||
|
||||
user, _ := ToUser(r.Context())
|
||||
if user != nil {
|
||||
csrf, _ = token.New(
|
||||
token.CsrfToken,
|
||||
user.Login,
|
||||
).Sign(user.Hash)
|
||||
}
|
||||
params := map[string]interface{}{
|
||||
"user": user,
|
||||
"csrf": csrf,
|
||||
"version": version.Version.String(),
|
||||
}
|
||||
|
||||
index, err := ioutil.ReadFile(filepath.Join(l.dir, "index.html"))
|
||||
if err != nil {
|
||||
rw.WriteHeader(404)
|
||||
return
|
||||
}
|
||||
var buf bytes.Buffer
|
||||
template.T.ExecuteTemplate(&buf, "script.html", params)
|
||||
index = bytes.Replace(index, []byte("<!-- inject:js -->"), buf.Bytes(), 1)
|
||||
rw.WriteHeader(200)
|
||||
rw.Write(index)
|
||||
}
|
102
server/web/web.go
Normal file
102
server/web/web.go
Normal file
|
@ -0,0 +1,102 @@
|
|||
package web
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/md5"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/drone/drone-ui/dist"
|
||||
"github.com/drone/drone/model"
|
||||
"github.com/drone/drone/server/template"
|
||||
"github.com/drone/drone/shared/token"
|
||||
"github.com/drone/drone/version"
|
||||
|
||||
"github.com/dimfeld/httptreemux"
|
||||
)
|
||||
|
||||
// Endpoint provides the website endpoints.
|
||||
type Endpoint interface {
|
||||
// Register registers the provider endpoints.
|
||||
Register(*httptreemux.ContextMux)
|
||||
}
|
||||
|
||||
// New returns the default website endpoint.
|
||||
func New() Endpoint {
|
||||
return new(website)
|
||||
}
|
||||
|
||||
type website struct{}
|
||||
|
||||
func (w *website) Register(mux *httptreemux.ContextMux) {
|
||||
r := dist.New()
|
||||
h := http.FileServer(r)
|
||||
h = setupCache(h)
|
||||
mux.Handler("GET", "/favicon-32x32.png", h)
|
||||
mux.Handler("GET", "/favicon-16x16.png", h)
|
||||
mux.Handler("GET", "/src/*filepath", h)
|
||||
mux.Handler("GET", "/bower_components/*filepath", h)
|
||||
mux.NotFoundHandler = handleIndex
|
||||
}
|
||||
|
||||
func handleIndex(rw http.ResponseWriter, r *http.Request) {
|
||||
rw.WriteHeader(200)
|
||||
|
||||
var csrf string
|
||||
var user, _ = ToUser(r.Context())
|
||||
if user != nil {
|
||||
csrf, _ = token.New(
|
||||
token.CsrfToken,
|
||||
user.Login,
|
||||
).Sign(user.Hash)
|
||||
}
|
||||
params := map[string]interface{}{
|
||||
"user": user,
|
||||
"csrf": csrf,
|
||||
"version": version.Version.String(),
|
||||
}
|
||||
rw.Header().Set("Content-Type", "text/html; charset=UTF-8")
|
||||
template.T.ExecuteTemplate(rw, "index_polymer.html", params)
|
||||
}
|
||||
|
||||
func setupCache(h http.Handler) http.Handler {
|
||||
data := []byte(time.Now().String())
|
||||
etag := fmt.Sprintf("%x", md5.Sum(data))
|
||||
|
||||
return http.HandlerFunc(
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Cache-Control", "public, max-age=31536000")
|
||||
w.Header().Del("Expires")
|
||||
w.Header().Set("ETag", etag)
|
||||
h.ServeHTTP(w, r)
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
func resetCache(h http.Handler) http.Handler {
|
||||
return http.HandlerFunc(
|
||||
func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Del("Cache-Control")
|
||||
w.Header().Del("Last-Updated")
|
||||
w.Header().Del("Expires")
|
||||
w.Header().Del("ETag")
|
||||
h.ServeHTTP(w, r)
|
||||
},
|
||||
)
|
||||
}
|
||||
|
||||
// WithUser returns a context with the current authenticated user.
|
||||
func WithUser(c context.Context, user *model.User) context.Context {
|
||||
return context.WithValue(c, userKey, user)
|
||||
}
|
||||
|
||||
// ToUser returns a user from the context.
|
||||
func ToUser(c context.Context) (*model.User, bool) {
|
||||
user, ok := c.Value(userKey).(*model.User)
|
||||
return user, ok
|
||||
}
|
||||
|
||||
type key int
|
||||
|
||||
const userKey key = 0
|
21
vendor/github.com/dimfeld/httptreemux/LICENSE
generated
vendored
Normal file
21
vendor/github.com/dimfeld/httptreemux/LICENSE
generated
vendored
Normal file
|
@ -0,0 +1,21 @@
|
|||
The MIT License (MIT)
|
||||
|
||||
Copyright (c) 2014,2015 Daniel Imfeld
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
240
vendor/github.com/dimfeld/httptreemux/README.md
generated
vendored
Normal file
240
vendor/github.com/dimfeld/httptreemux/README.md
generated
vendored
Normal file
|
@ -0,0 +1,240 @@
|
|||
httptreemux [![Build Status](https://travis-ci.org/dimfeld/httptreemux.png?branch=master)](https://travis-ci.org/dimfeld/httptreemux) [![GoDoc](https://godoc.org/github.com/dimfeld/httptreemux?status.svg)](https://godoc.org/github.com/dimfeld/httptreemux)
|
||||
===========
|
||||
|
||||
High-speed, flexible, tree-based HTTP router for Go.
|
||||
|
||||
This is inspired by [Julien Schmidt's httprouter](https://www.github.com/julienschmidt/httprouter), in that it uses a patricia tree, but the implementation is rather different. Specifically, the routing rules are relaxed so that a single path segment may be a wildcard in one route and a static token in another. This gives a nice combination of high performance with a lot of convenience in designing the routing patterns. In [benchmarks](https://github.com/julienschmidt/go-http-routing-benchmark), httptreemux is close to, but slightly slower than, httprouter.
|
||||
|
||||
Release notes may be found using the [Github releases tab](https://github.com/dimfeld/httptreemux/releases). Version numbers are compatible with the [Semantic Versioning 2.0.0](http://semver.org/) convention, and a new release is made after every change to the code.
|
||||
|
||||
## Why?
|
||||
There are a lot of good routers out there. But looking at the ones that were really lightweight, I couldn't quite get something that fit with the route patterns I wanted. The code itself is simple enough, so I spent an evening writing this.
|
||||
|
||||
## Handler
|
||||
The handler is a simple function with the prototype `func(w http.ResponseWriter, r *http.Request, params map[string]string)`. The params argument contains the parameters parsed from wildcards and catch-alls in the URL, as described below. This type is aliased as httptreemux.HandlerFunc.
|
||||
|
||||
### Using http.HandlerFunc
|
||||
Due to the inclusion of the [context](https://godoc.org/context) package as of Go 1.7, `httptreemux` now supports handlers of type [http.HandlerFunc](https://godoc.org/net/http#HandlerFunc). There are two ways to enable this support.
|
||||
|
||||
#### Adapting an Existing Router
|
||||
|
||||
The `UsingContext` method will wrap the router or group in a new group at the same path, but adapted for use with `context` and `http.HandlerFunc`.
|
||||
|
||||
```go
|
||||
router := httptreemux.New()
|
||||
|
||||
group := router.NewGroup("/api")
|
||||
group.GET("/v1/:id", func(w http.ResponseWriter, r *http.Request, params map[string]string) {
|
||||
id := params["id"]
|
||||
fmt.Fprintf(w, "GET /api/v1/%s", id)
|
||||
})
|
||||
|
||||
// UsingContext returns a version of the router or group with context support.
|
||||
ctxGroup := group.UsingContext() // sibling to 'group' node in tree
|
||||
ctxGroup.GET("/v2/:id", func(w http.ResponseWriter, r *http.Request) {
|
||||
params := httptreemux.ContextParams(r.Context())
|
||||
id := params["id"]
|
||||
fmt.Fprintf(w, "GET /api/v2/%s", id)
|
||||
})
|
||||
|
||||
http.ListenAndServe(":8080", router)
|
||||
```
|
||||
|
||||
#### New Router with Context Support
|
||||
|
||||
The `NewContextMux` function returns a router preconfigured for use with `context` and `http.HandlerFunc`.
|
||||
|
||||
```go
|
||||
router := httptreemux.NewContextMux()
|
||||
|
||||
router.GET("/:page", func(w http.ResponseWriter, r *http.Request) {
|
||||
params := httptreemux.ContextParams(r.Context())
|
||||
fmt.Fprintf(w, "GET /%s", params["page"])
|
||||
})
|
||||
|
||||
group := tree.NewGroup("/api")
|
||||
group.GET("/v1/:id", func(w http.ResponseWriter, r *http.Request) {
|
||||
params := httptreemux.ContextParams(r.Context())
|
||||
id := params["id"]
|
||||
fmt.Fprintf(w, "GET /api/v1/%s", id)
|
||||
})
|
||||
|
||||
http.ListenAndServe(":8080", router)
|
||||
```
|
||||
|
||||
|
||||
|
||||
## Routing Rules
|
||||
The syntax here is also modeled after httprouter. Each variable in a path may match on one segment only, except for an optional catch-all variable at the end of the URL.
|
||||
|
||||
Some examples of valid URL patterns are:
|
||||
* `/post/all`
|
||||
* `/post/:postid`
|
||||
* `/post/:postid/page/:page`
|
||||
* `/post/:postid/:page`
|
||||
* `/images/*path`
|
||||
* `/favicon.ico`
|
||||
* `/:year/:month/`
|
||||
* `/:year/:month/:post`
|
||||
* `/:page`
|
||||
|
||||
Note that all of the above URL patterns may exist concurrently in the router.
|
||||
|
||||
Path elements starting with `:` indicate a wildcard in the path. A wildcard will only match on a single path segment. That is, the pattern `/post/:postid` will match on `/post/1` or `/post/1/`, but not `/post/1/2`.
|
||||
|
||||
A path element starting with `*` is a catch-all, whose value will be a string containing all text in the URL matched by the wildcards. For example, with a pattern of `/images/*path` and a requested URL `images/abc/def`, path would contain `abc/def`.
|
||||
|
||||
#### Using : and * in routing patterns
|
||||
|
||||
The characters `:` and `*` can be used at the beginning of a path segment by escaping them with a backslash. A double backslash at the beginning of a segment is interpreted as a single backslash. These escapes are only checked at the very beginning of a path segment; they are not necessary or processed elsewhere in a token.
|
||||
|
||||
```go
|
||||
router.GET("/foo/\\*starToken", handler) // matches /foo/*starToken
|
||||
router.GET("/foo/star*inTheMiddle", handler) // matches /foo/star*inTheMiddle
|
||||
router.GET("/foo/starBackslash\\*", handler) // matches /foo/starBackslash\*
|
||||
router.GET("/foo/\\\\*backslashWithStar") // matches /foo/\*backslashWithStar
|
||||
```
|
||||
|
||||
### Routing Groups
|
||||
Lets you create a new group of routes with a given path prefix. Makes it easier to create clusters of paths like:
|
||||
* `/api/v1/foo`
|
||||
* `/api/v1/bar`
|
||||
|
||||
To use this you do:
|
||||
```go
|
||||
router = httptreemux.New()
|
||||
api := router.NewGroup("/api/v1")
|
||||
api.GET("/foo", fooHandler) // becomes /api/v1/foo
|
||||
api.GET("/bar", barHandler) // becomes /api/v1/bar
|
||||
```
|
||||
|
||||
### Routing Priority
|
||||
The priority rules in the router are simple.
|
||||
|
||||
1. Static path segments take the highest priority. If a segment and its subtree are able to match the URL, that match is returned.
|
||||
2. Wildcards take second priority. For a particular wildcard to match, that wildcard and its subtree must match the URL.
|
||||
3. Finally, a catch-all rule will match when the earlier path segments have matched, and none of the static or wildcard conditions have matched. Catch-all rules must be at the end of a pattern.
|
||||
|
||||
So with the following patterns adapted from [simpleblog](https://www.github.com/dimfeld/simpleblog), we'll see certain matches:
|
||||
```go
|
||||
router = httptreemux.New()
|
||||
router.GET("/:page", pageHandler)
|
||||
router.GET("/:year/:month/:post", postHandler)
|
||||
router.GET("/:year/:month", archiveHandler)
|
||||
router.GET("/images/*path", staticHandler)
|
||||
router.GET("/favicon.ico", staticHandler)
|
||||
```
|
||||
|
||||
#### Example scenarios
|
||||
|
||||
- `/abc` will match `/:page`
|
||||
- `/2014/05` will match `/:year/:month`
|
||||
- `/2014/05/really-great-blog-post` will match `/:year/:month/:post`
|
||||
- `/images/CoolImage.gif` will match `/images/*path`
|
||||
- `/images/2014/05/MayImage.jpg` will also match `/images/*path`, with all the text after `/images` stored in the variable path.
|
||||
- `/favicon.ico` will match `/favicon.ico`
|
||||
|
||||
### Special Method Behavior
|
||||
If TreeMux.HeadCanUseGet is set to true, the router will call the GET handler for a pattern when a HEAD request is processed, if no HEAD handler has been added for that pattern. This behavior is enabled by default.
|
||||
|
||||
Go's http.ServeContent and related functions already handle the HEAD method correctly by sending only the header, so in most cases your handlers will not need any special cases for it.
|
||||
|
||||
By default TreeMux.OptionsHandler is a null handler that doesn't affect your routing. If you set the handler, it will be called on OPTIONS requests to a path already registered by another method. If you set a path specific handler by using `router.OPTIONS`, it will override the global Options Handler for that path.
|
||||
|
||||
### Trailing Slashes
|
||||
The router has special handling for paths with trailing slashes. If a pattern is added to the router with a trailing slash, any matches on that pattern without a trailing slash will be redirected to the version with the slash. If a pattern does not have a trailing slash, matches on that pattern with a trailing slash will be redirected to the version without.
|
||||
|
||||
The trailing slash flag is only stored once for a pattern. That is, if a pattern is added for a method with a trailing slash, all other methods for that pattern will also be considered to have a trailing slash, regardless of whether or not it is specified for those methods too.
|
||||
However this behavior can be turned off by setting TreeMux.RedirectTrailingSlash to false. By default it is set to true.
|
||||
|
||||
One exception to this rule is catch-all patterns. By default, trailing slash redirection is disabled on catch-all patterns, since the structure of the entire URL and the desired patterns can not be predicted. If trailing slash removal is desired on catch-all patterns, set TreeMux.RemoveCatchAllTrailingSlash to true.
|
||||
|
||||
```go
|
||||
router = httptreemux.New()
|
||||
router.GET("/about", pageHandler)
|
||||
router.GET("/posts/", postIndexHandler)
|
||||
router.POST("/posts", postFormHandler)
|
||||
|
||||
GET /about will match normally.
|
||||
GET /about/ will redirect to /about.
|
||||
GET /posts will redirect to /posts/.
|
||||
GET /posts/ will match normally.
|
||||
POST /posts will redirect to /posts/, because the GET method used a trailing slash.
|
||||
```
|
||||
|
||||
### Custom Redirects
|
||||
|
||||
RedirectBehavior sets the behavior when the router redirects the request to the canonical version of the requested URL using RedirectTrailingSlash or RedirectClean. The default behavior is to return a 301 status, redirecting the browser to the version of the URL that matches the given pattern.
|
||||
|
||||
These are the values accepted for RedirectBehavior. You may also add these values to the RedirectMethodBehavior map to define custom per-method redirect behavior.
|
||||
|
||||
* Redirect301 - HTTP 301 Moved Permanently; this is the default.
|
||||
* Redirect307 - HTTP/1.1 Temporary Redirect
|
||||
* Redirect308 - RFC7538 Permanent Redirect
|
||||
* UseHandler - Don't redirect to the canonical path. Just call the handler instead.
|
||||
|
||||
#### Rationale/Usage
|
||||
On a POST request, most browsers that receive a 301 will submit a GET request to the redirected URL, meaning that any data will likely be lost. If you want to handle and avoid this behavior, you may use Redirect307, which causes most browsers to resubmit the request using the original method and request body.
|
||||
|
||||
Since 307 is supposed to be a temporary redirect, the new 308 status code has been proposed, which is treated the same, except it indicates correctly that the redirection is permanent. The big caveat here is that the RFC is relatively recent, and older or non-compliant browsers will not handle it. Therefore its use is not recommended unless you really know what you're doing.
|
||||
|
||||
Finally, the UseHandler value will simply call the handler function for the pattern, without redirecting to the canonical version of the URL.
|
||||
|
||||
### RequestURI vs. URL.Path
|
||||
|
||||
#### Escaped Slashes
|
||||
Go automatically processes escaped characters in a URL, converting + to a space and %XX to the corresponding character. This can present issues when the URL contains a %2f, which is unescaped to '/'. This isn't an issue for most applications, but it will prevent the router from correctly matching paths and wildcards.
|
||||
|
||||
For example, the pattern `/post/:post` would not match on `/post/abc%2fdef`, which is unescaped to `/post/abc/def`. The desired behavior is that it matches, and the `post` wildcard is set to `abc/def`.
|
||||
|
||||
Therefore, this router defaults to using the raw URL, stored in the Request.RequestURI variable. Matching wildcards and catch-alls are then unescaped, to give the desired behavior.
|
||||
|
||||
TL;DR: If a requested URL contains a %2f, this router will still do the right thing. Some Go HTTP routers may not due to [Go issue 3659](https://code.google.com/p/go/issues/detail?id=3659).
|
||||
|
||||
#### Escaped Characters
|
||||
|
||||
As mentioned above, characters in the URL are not unescaped when using RequestURI to determine the matched route. If this is a problem for you and you are unable to switch to URL.Path for the above reasons, you may set `router.EscapeAddedRoutes` to `true`. This option will run each added route through the `URL.EscapedPath` function, and add an additional route if the escaped version differs.
|
||||
|
||||
#### http Package Utility Functions
|
||||
|
||||
Although using RequestURI avoids the issue described above, certain utility functions such as `http.StripPrefix` modify URL.Path, and expect that the underlying router is using that field to make its decision. If you are using some of these functions, set the router's `PathSource` member to `URLPath`. This will give up the proper handling of escaped slashes described above, while allowing the router to work properly with these utility functions.
|
||||
|
||||
## Concurrency
|
||||
|
||||
The router contains an `RWMutex` that arbitrates access to the tree. This allows routes to be safely added from multiple goroutines at once.
|
||||
|
||||
No concurrency controls are needed when only reading from the tree, so the default behavior is to not use the `RWMutex` when serving a request. This avoids a theoretical slowdown under high-usage scenarios from competing atomic integer operations inside the `RWMutex`. If your application adds routes to the router after it has begun serving requests, you should avoid potential race conditions by setting `router.SafeAddRoutesWhileRunning` to `true` to use the `RWMutex` when serving requests.
|
||||
|
||||
## Error Handlers
|
||||
|
||||
### NotFoundHandler
|
||||
TreeMux.NotFoundHandler can be set to provide custom 404-error handling. The default implementation is Go's `http.NotFound` function.
|
||||
|
||||
### MethodNotAllowedHandler
|
||||
If a pattern matches, but the pattern does not have an associated handler for the requested method, the router calls the MethodNotAllowedHandler. The default
|
||||
version of this handler just writes the status code `http.StatusMethodNotAllowed` and sets the response header's `Allowed` field appropriately.
|
||||
|
||||
### Panic Handling
|
||||
TreeMux.PanicHandler can be set to provide custom panic handling. The `SimplePanicHandler` just writes the status code `http.StatusInternalServerError`. The function `ShowErrorsPanicHandler`, adapted from [gocraft/web](https://github.com/gocraft/web), will print panic errors to the browser in an easily-readable format.
|
||||
|
||||
## Unexpected Differences from Other Routers
|
||||
|
||||
This router is intentionally light on features in the name of simplicity and
|
||||
performance. When coming from another router that does heavier processing behind
|
||||
the scenes, you may encounter some unexpected behavior. This list is by no means
|
||||
exhaustive, but covers some nonobvious cases that users have encountered.
|
||||
|
||||
### gorilla/pat query string modifications
|
||||
|
||||
When matching on parameters in a route, the `gorilla/pat` router will modify
|
||||
`Request.URL.RawQuery` to make it appear like the parameters were in the
|
||||
query string. `httptreemux` does not do this. See [Issue #26](https://github.com/dimfeld/httptreemux/issues/26) for more details and a
|
||||
code snippet that can perform this transformation for you, should you want it.
|
||||
|
||||
## Middleware
|
||||
This package provides no middleware. But there are a lot of great options out there and it's pretty easy to write your own.
|
||||
|
||||
# Acknowledgements
|
||||
|
||||
* Inspiration from Julien Schmidt's [httprouter](https://github.com/julienschmidt/httprouter)
|
||||
* Show Errors panic handler from [gocraft/web](https://github.com/gocraft/web)
|
116
vendor/github.com/dimfeld/httptreemux/context.go
generated
vendored
Normal file
116
vendor/github.com/dimfeld/httptreemux/context.go
generated
vendored
Normal file
|
@ -0,0 +1,116 @@
|
|||
// +build go1.7
|
||||
|
||||
package httptreemux
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// ContextGroup is a wrapper around Group, with the purpose of mimicking its API, but with the use of http.HandlerFunc-based handlers.
|
||||
// Instead of passing a parameter map via the handler (i.e. httptreemux.HandlerFunc), the path parameters are accessed via the request
|
||||
// object's context.
|
||||
type ContextGroup struct {
|
||||
group *Group
|
||||
}
|
||||
|
||||
// UsingContext wraps the receiver to return a new instance of a ContextGroup.
|
||||
// The returned ContextGroup is a sibling to its wrapped Group, within the parent TreeMux.
|
||||
// The choice of using a *Group as the receiver, as opposed to a function parameter, allows chaining
|
||||
// while method calls between a TreeMux, Group, and ContextGroup. For example:
|
||||
//
|
||||
// tree := httptreemux.New()
|
||||
// group := tree.NewGroup("/api")
|
||||
//
|
||||
// group.GET("/v1", func(w http.ResponseWriter, r *http.Request, params map[string]string) {
|
||||
// w.Write([]byte(`GET /api/v1`))
|
||||
// })
|
||||
//
|
||||
// group.UsingContext().GET("/v2", func(w http.ResponseWriter, r *http.Request) {
|
||||
// w.Write([]byte(`GET /api/v2`))
|
||||
// })
|
||||
//
|
||||
// http.ListenAndServe(":8080", tree)
|
||||
//
|
||||
func (g *Group) UsingContext() *ContextGroup {
|
||||
return &ContextGroup{g}
|
||||
}
|
||||
|
||||
// NewContextGroup adds a child context group to its path.
|
||||
func (cg *ContextGroup) NewContextGroup(path string) *ContextGroup {
|
||||
return &ContextGroup{cg.group.NewGroup(path)}
|
||||
}
|
||||
|
||||
func (cg *ContextGroup) NewGroup(path string) *ContextGroup {
|
||||
return cg.NewContextGroup(path)
|
||||
}
|
||||
|
||||
// Handle allows handling HTTP requests via an http.HandlerFunc, as opposed to an httptreemux.HandlerFunc.
|
||||
// Any parameters from the request URL are stored in a map[string]string in the request's context.
|
||||
func (cg *ContextGroup) Handle(method, path string, handler http.HandlerFunc) {
|
||||
cg.group.Handle(method, path, func(w http.ResponseWriter, r *http.Request, params map[string]string) {
|
||||
if params != nil {
|
||||
r = r.WithContext(context.WithValue(r.Context(), paramsContextKey, params))
|
||||
}
|
||||
handler(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
// Handler allows handling HTTP requests via an http.Handler interface, as opposed to an httptreemux.HandlerFunc.
|
||||
// Any parameters from the request URL are stored in a map[string]string in the request's context.
|
||||
func (cg *ContextGroup) Handler(method, path string, handler http.Handler) {
|
||||
cg.group.Handle(method, path, func(w http.ResponseWriter, r *http.Request, params map[string]string) {
|
||||
if params != nil {
|
||||
r = r.WithContext(context.WithValue(r.Context(), paramsContextKey, params))
|
||||
}
|
||||
handler.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
// GET is convenience method for handling GET requests on a context group.
|
||||
func (cg *ContextGroup) GET(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("GET", path, handler)
|
||||
}
|
||||
|
||||
// POST is convenience method for handling POST requests on a context group.
|
||||
func (cg *ContextGroup) POST(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("POST", path, handler)
|
||||
}
|
||||
|
||||
// PUT is convenience method for handling PUT requests on a context group.
|
||||
func (cg *ContextGroup) PUT(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("PUT", path, handler)
|
||||
}
|
||||
|
||||
// DELETE is convenience method for handling DELETE requests on a context group.
|
||||
func (cg *ContextGroup) DELETE(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("DELETE", path, handler)
|
||||
}
|
||||
|
||||
// PATCH is convenience method for handling PATCH requests on a context group.
|
||||
func (cg *ContextGroup) PATCH(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("PATCH", path, handler)
|
||||
}
|
||||
|
||||
// HEAD is convenience method for handling HEAD requests on a context group.
|
||||
func (cg *ContextGroup) HEAD(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("HEAD", path, handler)
|
||||
}
|
||||
|
||||
// OPTIONS is convenience method for handling OPTIONS requests on a context group.
|
||||
func (cg *ContextGroup) OPTIONS(path string, handler http.HandlerFunc) {
|
||||
cg.Handle("OPTIONS", path, handler)
|
||||
}
|
||||
|
||||
// ContextParams returns the params map associated with the given context if one exists. Otherwise, an empty map is returned.
|
||||
func ContextParams(ctx context.Context) map[string]string {
|
||||
if p, ok := ctx.Value(paramsContextKey).(map[string]string); ok {
|
||||
return p
|
||||
}
|
||||
return map[string]string{}
|
||||
}
|
||||
|
||||
type contextKey int
|
||||
|
||||
// paramsContextKey is used to retrieve a path's params map from a request's context.
|
||||
const paramsContextKey contextKey = 0
|
195
vendor/github.com/dimfeld/httptreemux/group.go
generated
vendored
Normal file
195
vendor/github.com/dimfeld/httptreemux/group.go
generated
vendored
Normal file
|
@ -0,0 +1,195 @@
|
|||
package httptreemux
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/url"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Group struct {
|
||||
path string
|
||||
mux *TreeMux
|
||||
}
|
||||
|
||||
// Add a sub-group to this group
|
||||
func (g *Group) NewGroup(path string) *Group {
|
||||
if len(path) < 1 {
|
||||
panic("Group path must not be empty")
|
||||
}
|
||||
|
||||
checkPath(path)
|
||||
path = g.path + path
|
||||
//Don't want trailing slash as all sub-paths start with slash
|
||||
if path[len(path)-1] == '/' {
|
||||
path = path[:len(path)-1]
|
||||
}
|
||||
return &Group{path, g.mux}
|
||||
}
|
||||
|
||||
// Path elements starting with : indicate a wildcard in the path. A wildcard will only match on a
|
||||
// single path segment. That is, the pattern `/post/:postid` will match on `/post/1` or `/post/1/`,
|
||||
// but not `/post/1/2`.
|
||||
//
|
||||
// A path element starting with * is a catch-all, whose value will be a string containing all text
|
||||
// in the URL matched by the wildcards. For example, with a pattern of `/images/*path` and a
|
||||
// requested URL `images/abc/def`, path would contain `abc/def`.
|
||||
//
|
||||
// # Routing Rule Priority
|
||||
//
|
||||
// The priority rules in the router are simple.
|
||||
//
|
||||
// 1. Static path segments take the highest priority. If a segment and its subtree are able to match the URL, that match is returned.
|
||||
//
|
||||
// 2. Wildcards take second priority. For a particular wildcard to match, that wildcard and its subtree must match the URL.
|
||||
//
|
||||
// 3. Finally, a catch-all rule will match when the earlier path segments have matched, and none of the static or wildcard conditions have matched. Catch-all rules must be at the end of a pattern.
|
||||
//
|
||||
// So with the following patterns, we'll see certain matches:
|
||||
// router = httptreemux.New()
|
||||
// router.GET("/:page", pageHandler)
|
||||
// router.GET("/:year/:month/:post", postHandler)
|
||||
// router.GET("/:year/:month", archiveHandler)
|
||||
// router.GET("/images/*path", staticHandler)
|
||||
// router.GET("/favicon.ico", staticHandler)
|
||||
//
|
||||
// /abc will match /:page
|
||||
// /2014/05 will match /:year/:month
|
||||
// /2014/05/really-great-blog-post will match /:year/:month/:post
|
||||
// /images/CoolImage.gif will match /images/*path
|
||||
// /images/2014/05/MayImage.jpg will also match /images/*path, with all the text after /images stored in the variable path.
|
||||
// /favicon.ico will match /favicon.ico
|
||||
//
|
||||
// # Trailing Slashes
|
||||
//
|
||||
// The router has special handling for paths with trailing slashes. If a pattern is added to the
|
||||
// router with a trailing slash, any matches on that pattern without a trailing slash will be
|
||||
// redirected to the version with the slash. If a pattern does not have a trailing slash, matches on
|
||||
// that pattern with a trailing slash will be redirected to the version without.
|
||||
//
|
||||
// The trailing slash flag is only stored once for a pattern. That is, if a pattern is added for a
|
||||
// method with a trailing slash, all other methods for that pattern will also be considered to have a
|
||||
// trailing slash, regardless of whether or not it is specified for those methods too.
|
||||
//
|
||||
// This behavior can be turned off by setting TreeMux.RedirectTrailingSlash to false. By
|
||||
// default it is set to true. The specifics of the redirect depend on RedirectBehavior.
|
||||
//
|
||||
// One exception to this rule is catch-all patterns. By default, trailing slash redirection is
|
||||
// disabled on catch-all patterns, since the structure of the entire URL and the desired patterns
|
||||
// can not be predicted. If trailing slash removal is desired on catch-all patterns, set
|
||||
// TreeMux.RemoveCatchAllTrailingSlash to true.
|
||||
//
|
||||
// router = httptreemux.New()
|
||||
// router.GET("/about", pageHandler)
|
||||
// router.GET("/posts/", postIndexHandler)
|
||||
// router.POST("/posts", postFormHandler)
|
||||
//
|
||||
// GET /about will match normally.
|
||||
// GET /about/ will redirect to /about.
|
||||
// GET /posts will redirect to /posts/.
|
||||
// GET /posts/ will match normally.
|
||||
// POST /posts will redirect to /posts/, because the GET method used a trailing slash.
|
||||
func (g *Group) Handle(method string, path string, handler HandlerFunc) {
|
||||
g.mux.mutex.Lock()
|
||||
defer g.mux.mutex.Unlock()
|
||||
|
||||
addSlash := false
|
||||
addOne := func(thePath string) {
|
||||
node := g.mux.root.addPath(thePath[1:], nil, false)
|
||||
if addSlash {
|
||||
node.addSlash = true
|
||||
}
|
||||
node.setHandler(method, handler, false)
|
||||
|
||||
if g.mux.HeadCanUseGet && method == "GET" && node.leafHandler["HEAD"] == nil {
|
||||
node.setHandler("HEAD", handler, true)
|
||||
}
|
||||
}
|
||||
|
||||
checkPath(path)
|
||||
path = g.path + path
|
||||
if len(path) == 0 {
|
||||
panic("Cannot map an empty path")
|
||||
}
|
||||
|
||||
if len(path) > 1 && path[len(path)-1] == '/' && g.mux.RedirectTrailingSlash {
|
||||
addSlash = true
|
||||
path = path[:len(path)-1]
|
||||
}
|
||||
|
||||
if g.mux.EscapeAddedRoutes {
|
||||
u, err := url.ParseRequestURI(path)
|
||||
if err != nil {
|
||||
panic("URL parsing error " + err.Error() + " on url " + path)
|
||||
}
|
||||
escapedPath := unescapeSpecial(u.String())
|
||||
|
||||
if escapedPath != path {
|
||||
addOne(escapedPath)
|
||||
}
|
||||
}
|
||||
|
||||
addOne(path)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("GET", path, handler)
|
||||
func (g *Group) GET(path string, handler HandlerFunc) {
|
||||
g.Handle("GET", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("POST", path, handler)
|
||||
func (g *Group) POST(path string, handler HandlerFunc) {
|
||||
g.Handle("POST", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("PUT", path, handler)
|
||||
func (g *Group) PUT(path string, handler HandlerFunc) {
|
||||
g.Handle("PUT", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("DELETE", path, handler)
|
||||
func (g *Group) DELETE(path string, handler HandlerFunc) {
|
||||
g.Handle("DELETE", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("PATCH", path, handler)
|
||||
func (g *Group) PATCH(path string, handler HandlerFunc) {
|
||||
g.Handle("PATCH", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("HEAD", path, handler)
|
||||
func (g *Group) HEAD(path string, handler HandlerFunc) {
|
||||
g.Handle("HEAD", path, handler)
|
||||
}
|
||||
|
||||
// Syntactic sugar for Handle("OPTIONS", path, handler)
|
||||
func (g *Group) OPTIONS(path string, handler HandlerFunc) {
|
||||
g.Handle("OPTIONS", path, handler)
|
||||
}
|
||||
|
||||
func checkPath(path string) {
|
||||
// All non-empty paths must start with a slash
|
||||
if len(path) > 0 && path[0] != '/' {
|
||||
panic(fmt.Sprintf("Path %s must start with slash", path))
|
||||
}
|
||||
}
|
||||
|
||||
func unescapeSpecial(s string) string {
|
||||
// Look for sequences of \*, *, and \: that were escaped, and undo some of that escaping.
|
||||
|
||||
// Unescape /* since it references a wildcard token.
|
||||
s = strings.Replace(s, "/%2A", "/*", -1)
|
||||
|
||||
// Unescape /\: since it references a literal colon
|
||||
s = strings.Replace(s, "/%5C:", "/\\:", -1)
|
||||
|
||||
// Replace escaped /\\: with /\:
|
||||
s = strings.Replace(s, "/%5C%5C:", "/%5C:", -1)
|
||||
|
||||
// Replace escaped /\* with /*
|
||||
s = strings.Replace(s, "/%5C%2A", "/%2A", -1)
|
||||
|
||||
// Replace escaped /\\* with /\*
|
||||
s = strings.Replace(s, "/%5C%5C%2A", "/%5C%2A", -1)
|
||||
|
||||
return s
|
||||
}
|
211
vendor/github.com/dimfeld/httptreemux/panichandler.go
generated
vendored
Normal file
211
vendor/github.com/dimfeld/httptreemux/panichandler.go
generated
vendored
Normal file
|
@ -0,0 +1,211 @@
|
|||
package httptreemux
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"html/template"
|
||||
"net/http"
|
||||
"os"
|
||||
"runtime"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// SimplePanicHandler just returns error 500.
|
||||
func SimplePanicHandler(w http.ResponseWriter, r *http.Request, err interface{}) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
}
|
||||
|
||||
// ShowErrorsPanicHandler prints a nice representation of an error to the browser.
|
||||
// This was taken from github.com/gocraft/web, which adapted it from the Traffic project.
|
||||
func ShowErrorsPanicHandler(w http.ResponseWriter, r *http.Request, err interface{}) {
|
||||
const size = 4096
|
||||
stack := make([]byte, size)
|
||||
stack = stack[:runtime.Stack(stack, false)]
|
||||
renderPrettyError(w, r, err, stack)
|
||||
}
|
||||
|
||||
func makeErrorData(r *http.Request, err interface{}, stack []byte, filePath string, line int) map[string]interface{} {
|
||||
|
||||
data := map[string]interface{}{
|
||||
"Stack": string(stack),
|
||||
"Params": r.URL.Query(),
|
||||
"Method": r.Method,
|
||||
"FilePath": filePath,
|
||||
"Line": line,
|
||||
"Lines": readErrorFileLines(filePath, line),
|
||||
}
|
||||
|
||||
if e, ok := err.(error); ok {
|
||||
data["Error"] = e.Error()
|
||||
} else {
|
||||
data["Error"] = err
|
||||
}
|
||||
|
||||
return data
|
||||
}
|
||||
|
||||
func renderPrettyError(rw http.ResponseWriter, req *http.Request, err interface{}, stack []byte) {
|
||||
_, filePath, line, _ := runtime.Caller(5)
|
||||
|
||||
data := makeErrorData(req, err, stack, filePath, line)
|
||||
rw.Header().Set("Content-Type", "text/html")
|
||||
rw.WriteHeader(http.StatusInternalServerError)
|
||||
|
||||
tpl := template.Must(template.New("ErrorPage").Parse(panicPageTpl))
|
||||
tpl.Execute(rw, data)
|
||||
}
|
||||
|
||||
func ShowErrorsJsonPanicHandler(w http.ResponseWriter, r *http.Request, err interface{}) {
|
||||
const size = 4096
|
||||
stack := make([]byte, size)
|
||||
stack = stack[:runtime.Stack(stack, false)]
|
||||
|
||||
_, filePath, line, _ := runtime.Caller(4)
|
||||
data := makeErrorData(r, err, stack, filePath, line)
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
json.NewEncoder(w).Encode(data)
|
||||
}
|
||||
|
||||
func readErrorFileLines(filePath string, errorLine int) map[int]string {
|
||||
lines := make(map[int]string)
|
||||
|
||||
file, err := os.Open(filePath)
|
||||
if err != nil {
|
||||
return lines
|
||||
}
|
||||
|
||||
defer file.Close()
|
||||
|
||||
reader := bufio.NewReader(file)
|
||||
currentLine := 0
|
||||
for {
|
||||
line, err := reader.ReadString('\n')
|
||||
if err != nil || currentLine > errorLine+5 {
|
||||
break
|
||||
}
|
||||
|
||||
currentLine++
|
||||
|
||||
if currentLine >= errorLine-5 {
|
||||
lines[currentLine] = strings.Replace(line, "\n", "", -1)
|
||||
}
|
||||
}
|
||||
|
||||
return lines
|
||||
}
|
||||
|
||||
const panicPageTpl string = `
|
||||
<html>
|
||||
<head>
|
||||
<title>Panic</title>
|
||||
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
|
||||
<style>
|
||||
html, body{ padding: 0; margin: 0; }
|
||||
header { background: #C52F24; color: white; border-bottom: 2px solid #9C0606; }
|
||||
h1 { padding: 10px 0; margin: 0; }
|
||||
.container { margin: 0 20px; }
|
||||
.error { font-size: 18px; background: #FFCCCC; color: #9C0606; padding: 10px 0; }
|
||||
.file-info .file-name { font-weight: bold; }
|
||||
.stack { height: 300px; overflow-y: scroll; border: 1px solid #e5e5e5; padding: 10px; }
|
||||
|
||||
table.source {
|
||||
width: 100%;
|
||||
border-collapse: collapse;
|
||||
border: 1px solid #e5e5e5;
|
||||
}
|
||||
|
||||
table.source td {
|
||||
padding: 0;
|
||||
}
|
||||
|
||||
table.source .numbers {
|
||||
font-size: 14px;
|
||||
vertical-align: top;
|
||||
width: 1%;
|
||||
color: rgba(0,0,0,0.3);
|
||||
text-align: right;
|
||||
}
|
||||
|
||||
table.source .numbers .number {
|
||||
display: block;
|
||||
padding: 0 5px;
|
||||
border-right: 1px solid #e5e5e5;
|
||||
}
|
||||
|
||||
table.source .numbers .number.line-{{ .Line }} {
|
||||
border-right: 1px solid #ffcccc;
|
||||
}
|
||||
|
||||
table.source .numbers pre {
|
||||
white-space: pre-wrap;
|
||||
}
|
||||
|
||||
table.source .code {
|
||||
font-size: 14px;
|
||||
vertical-align: top;
|
||||
}
|
||||
|
||||
table.source .code .line {
|
||||
padding-left: 10px;
|
||||
display: block;
|
||||
}
|
||||
|
||||
table.source .numbers .number,
|
||||
table.source .code .line {
|
||||
padding-top: 1px;
|
||||
padding-bottom: 1px;
|
||||
}
|
||||
|
||||
table.source .code .line:hover {
|
||||
background-color: #f6f6f6;
|
||||
}
|
||||
|
||||
table.source .line-{{ .Line }},
|
||||
table.source line-{{ .Line }},
|
||||
table.source .code .line.line-{{ .Line }}:hover {
|
||||
background: #ffcccc;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<header>
|
||||
<div class="container">
|
||||
<h1>Error</h1>
|
||||
</div>
|
||||
</header>
|
||||
|
||||
<div class="error">
|
||||
<p class="container">{{ .Error }}</p>
|
||||
</div>
|
||||
|
||||
<div class="container">
|
||||
<p class="file-info">
|
||||
In <span class="file-name">{{ .FilePath }}:{{ .Line }}</span></p>
|
||||
</p>
|
||||
|
||||
<table class="source">
|
||||
<tr>
|
||||
<td class="numbers">
|
||||
<pre>{{ range $lineNumber, $line := .Lines }}<span class="number line-{{ $lineNumber }}">{{ $lineNumber }}</span>{{ end }}</pre>
|
||||
</td>
|
||||
<td class="code">
|
||||
<pre>{{ range $lineNumber, $line := .Lines }}<span class="line line-{{ $lineNumber }}">{{ $line }}<br /></span>{{ end }}</pre>
|
||||
</td>
|
||||
</tr>
|
||||
</table>
|
||||
<h2>Stack</h2>
|
||||
<pre class="stack">{{ .Stack }}</pre>
|
||||
<h2>Request</h2>
|
||||
<p><strong>Method:</strong> {{ .Method }}</p>
|
||||
<h3>Parameters:</h3>
|
||||
<ul>
|
||||
{{ range $key, $value := .Params }}
|
||||
<li><strong>{{ $key }}:</strong> {{ $value }}</li>
|
||||
{{ end }}
|
||||
</ul>
|
||||
</div>
|
||||
</body>
|
||||
</html>
|
||||
`
|
127
vendor/github.com/dimfeld/httptreemux/path.go
generated
vendored
Normal file
127
vendor/github.com/dimfeld/httptreemux/path.go
generated
vendored
Normal file
|
@ -0,0 +1,127 @@
|
|||
// Copyright 2013 Julien Schmidt. All rights reserved.
|
||||
// Based on the path package, Copyright 2009 The Go Authors.
|
||||
// Use of this source code is governed by a BSD-style license that can be found
|
||||
// in the LICENSE file.
|
||||
|
||||
package httptreemux
|
||||
|
||||
// Clean is the URL version of path.Clean, it returns a canonical URL path
|
||||
// for p, eliminating . and .. elements.
|
||||
//
|
||||
// The following rules are applied iteratively until no further processing can
|
||||
// be done:
|
||||
// 1. Replace multiple slashes with a single slash.
|
||||
// 2. Eliminate each . path name element (the current directory).
|
||||
// 3. Eliminate each inner .. path name element (the parent directory)
|
||||
// along with the non-.. element that precedes it.
|
||||
// 4. Eliminate .. elements that begin a rooted path:
|
||||
// that is, replace "/.." by "/" at the beginning of a path.
|
||||
//
|
||||
// If the result of this process is an empty string, "/" is returned
|
||||
func Clean(p string) string {
|
||||
if p == "" {
|
||||
return "/"
|
||||
}
|
||||
|
||||
n := len(p)
|
||||
var buf []byte
|
||||
|
||||
// Invariants:
|
||||
// reading from path; r is index of next byte to process.
|
||||
// writing to buf; w is index of next byte to write.
|
||||
|
||||
// path must start with '/'
|
||||
r := 1
|
||||
w := 1
|
||||
|
||||
if p[0] != '/' {
|
||||
r = 0
|
||||
buf = make([]byte, n+1)
|
||||
buf[0] = '/'
|
||||
}
|
||||
|
||||
trailing := n > 2 && p[n-1] == '/'
|
||||
|
||||
// A bit more clunky without a 'lazybuf' like the path package, but the loop
|
||||
// gets completely inlined (bufApp). So in contrast to the path package this
|
||||
// loop has no expensive function calls (except 1x make)
|
||||
|
||||
for r < n {
|
||||
switch {
|
||||
case p[r] == '/':
|
||||
// empty path element, trailing slash is added after the end
|
||||
r++
|
||||
|
||||
case p[r] == '.' && r+1 == n:
|
||||
trailing = true
|
||||
r++
|
||||
|
||||
case p[r] == '.' && p[r+1] == '/':
|
||||
// . element
|
||||
r++
|
||||
|
||||
case p[r] == '.' && p[r+1] == '.' && (r+2 == n || p[r+2] == '/'):
|
||||
// .. element: remove to last /
|
||||
r += 2
|
||||
|
||||
if w > 1 {
|
||||
// can backtrack
|
||||
w--
|
||||
|
||||
if buf == nil {
|
||||
for w > 1 && p[w] != '/' {
|
||||
w--
|
||||
}
|
||||
} else {
|
||||
for w > 1 && buf[w] != '/' {
|
||||
w--
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
default:
|
||||
// real path element.
|
||||
// add slash if needed
|
||||
if w > 1 {
|
||||
bufApp(&buf, p, w, '/')
|
||||
w++
|
||||
}
|
||||
|
||||
// copy element
|
||||
for r < n && p[r] != '/' {
|
||||
bufApp(&buf, p, w, p[r])
|
||||
w++
|
||||
r++
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// re-append trailing slash
|
||||
if trailing && w > 1 {
|
||||
bufApp(&buf, p, w, '/')
|
||||
w++
|
||||
}
|
||||
|
||||
// Turn empty string into "/"
|
||||
if w == 0 {
|
||||
return "/"
|
||||
}
|
||||
|
||||
if buf == nil {
|
||||
return p[:w]
|
||||
}
|
||||
return string(buf[:w])
|
||||
}
|
||||
|
||||
// internal helper to lazily create a buffer if necessary
|
||||
func bufApp(buf *[]byte, s string, w int, c byte) {
|
||||
if *buf == nil {
|
||||
if s[w] == c {
|
||||
return
|
||||
}
|
||||
|
||||
*buf = make([]byte, len(s))
|
||||
copy(*buf, s[:w])
|
||||
}
|
||||
(*buf)[w] = c
|
||||
}
|
290
vendor/github.com/dimfeld/httptreemux/router.go
generated
vendored
Normal file
290
vendor/github.com/dimfeld/httptreemux/router.go
generated
vendored
Normal file
|
@ -0,0 +1,290 @@
|
|||
// This is inspired by Julien Schmidt's httprouter, in that it uses a patricia tree, but the
|
||||
// implementation is rather different. Specifically, the routing rules are relaxed so that a
|
||||
// single path segment may be a wildcard in one route and a static token in another. This gives a
|
||||
// nice combination of high performance with a lot of convenience in designing the routing patterns.
|
||||
package httptreemux
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
// The params argument contains the parameters parsed from wildcards and catch-alls in the URL.
|
||||
type HandlerFunc func(http.ResponseWriter, *http.Request, map[string]string)
|
||||
type PanicHandler func(http.ResponseWriter, *http.Request, interface{})
|
||||
|
||||
// RedirectBehavior sets the behavior when the router redirects the request to the
|
||||
// canonical version of the requested URL using RedirectTrailingSlash or RedirectClean.
|
||||
// The default behavior is to return a 301 status, redirecting the browser to the version
|
||||
// of the URL that matches the given pattern.
|
||||
//
|
||||
// On a POST request, most browsers that receive a 301 will submit a GET request to
|
||||
// the redirected URL, meaning that any data will likely be lost. If you want to handle
|
||||
// and avoid this behavior, you may use Redirect307, which causes most browsers to
|
||||
// resubmit the request using the original method and request body.
|
||||
//
|
||||
// Since 307 is supposed to be a temporary redirect, the new 308 status code has been
|
||||
// proposed, which is treated the same, except it indicates correctly that the redirection
|
||||
// is permanent. The big caveat here is that the RFC is relatively recent, and older
|
||||
// browsers will not know what to do with it. Therefore its use is not recommended
|
||||
// unless you really know what you're doing.
|
||||
//
|
||||
// Finally, the UseHandler value will simply call the handler function for the pattern.
|
||||
type RedirectBehavior int
|
||||
|
||||
type PathSource int
|
||||
|
||||
const (
|
||||
Redirect301 RedirectBehavior = iota // Return 301 Moved Permanently
|
||||
Redirect307 // Return 307 HTTP/1.1 Temporary Redirect
|
||||
Redirect308 // Return a 308 RFC7538 Permanent Redirect
|
||||
UseHandler // Just call the handler function
|
||||
|
||||
RequestURI PathSource = iota // Use r.RequestURI
|
||||
URLPath // Use r.URL.Path
|
||||
)
|
||||
|
||||
// LookupResult contains information about a route lookup, which is returned from Lookup and
|
||||
// can be passed to ServeLookupResult if the request should be served.
|
||||
type LookupResult struct {
|
||||
// StatusCode informs the caller about the result of the lookup.
|
||||
// This will generally be `http.StatusNotFound` or `http.StatusMethodNotAllowed` for an
|
||||
// error case. On a normal success, the statusCode will be `http.StatusOK`. A redirect code
|
||||
// will also be used in the case
|
||||
StatusCode int
|
||||
handler HandlerFunc
|
||||
params map[string]string
|
||||
leafHandler map[string]HandlerFunc // Only has a value when StatusCode is MethodNotAllowed.
|
||||
}
|
||||
|
||||
// Dump returns a text representation of the routing tree.
|
||||
func (t *TreeMux) Dump() string {
|
||||
return t.root.dumpTree("", "")
|
||||
}
|
||||
|
||||
func (t *TreeMux) serveHTTPPanic(w http.ResponseWriter, r *http.Request) {
|
||||
if err := recover(); err != nil {
|
||||
t.PanicHandler(w, r, err)
|
||||
}
|
||||
}
|
||||
|
||||
func (t *TreeMux) redirectStatusCode(method string) (int, bool) {
|
||||
var behavior RedirectBehavior
|
||||
var ok bool
|
||||
if behavior, ok = t.RedirectMethodBehavior[method]; !ok {
|
||||
behavior = t.RedirectBehavior
|
||||
}
|
||||
switch behavior {
|
||||
case Redirect301:
|
||||
return http.StatusMovedPermanently, true
|
||||
case Redirect307:
|
||||
return http.StatusTemporaryRedirect, true
|
||||
case Redirect308:
|
||||
// Go doesn't have a constant for this yet. Yet another sign
|
||||
// that you probably shouldn't use it.
|
||||
return 308, true
|
||||
case UseHandler:
|
||||
return 0, false
|
||||
default:
|
||||
return http.StatusMovedPermanently, true
|
||||
}
|
||||
}
|
||||
|
||||
func redirectHandler(newPath string, statusCode int) HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request, params map[string]string) {
|
||||
redirect(w, r, newPath, statusCode)
|
||||
}
|
||||
}
|
||||
|
||||
func redirect(w http.ResponseWriter, r *http.Request, newPath string, statusCode int) {
|
||||
newURL := url.URL{
|
||||
Path: newPath,
|
||||
RawQuery: r.URL.RawQuery,
|
||||
Fragment: r.URL.Fragment,
|
||||
}
|
||||
http.Redirect(w, r, newURL.String(), statusCode)
|
||||
}
|
||||
|
||||
func (t *TreeMux) lookup(w http.ResponseWriter, r *http.Request) (result LookupResult, found bool) {
|
||||
result.StatusCode = http.StatusNotFound
|
||||
path := r.RequestURI
|
||||
pathLen := len(path)
|
||||
if pathLen > 0 && t.PathSource == RequestURI {
|
||||
rawQueryLen := len(r.URL.RawQuery)
|
||||
|
||||
if rawQueryLen != 0 || path[pathLen-1] == '?' {
|
||||
// Remove any query string and the ?.
|
||||
path = path[:pathLen-rawQueryLen-1]
|
||||
pathLen = len(path)
|
||||
}
|
||||
} else {
|
||||
// In testing with http.NewRequest,
|
||||
// RequestURI is not set so just grab URL.Path instead.
|
||||
path = r.URL.Path
|
||||
pathLen = len(path)
|
||||
}
|
||||
|
||||
trailingSlash := path[pathLen-1] == '/' && pathLen > 1
|
||||
if trailingSlash && t.RedirectTrailingSlash {
|
||||
path = path[:pathLen-1]
|
||||
}
|
||||
|
||||
n, handler, params := t.root.search(r.Method, path[1:])
|
||||
if n == nil {
|
||||
if t.RedirectCleanPath {
|
||||
// Path was not found. Try cleaning it up and search again.
|
||||
// TODO Test this
|
||||
cleanPath := Clean(path)
|
||||
n, handler, params = t.root.search(r.Method, cleanPath[1:])
|
||||
if n == nil {
|
||||
// Still nothing found.
|
||||
return
|
||||
}
|
||||
if statusCode, ok := t.redirectStatusCode(r.Method); ok {
|
||||
// Redirect to the actual path
|
||||
return LookupResult{statusCode, redirectHandler(cleanPath, statusCode), nil, nil}, true
|
||||
}
|
||||
} else {
|
||||
// Not found.
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if handler == nil {
|
||||
if r.Method == "OPTIONS" && t.OptionsHandler != nil {
|
||||
handler = t.OptionsHandler
|
||||
}
|
||||
|
||||
if handler == nil {
|
||||
result.leafHandler = n.leafHandler
|
||||
result.StatusCode = http.StatusMethodNotAllowed
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if !n.isCatchAll || t.RemoveCatchAllTrailingSlash {
|
||||
if trailingSlash != n.addSlash && t.RedirectTrailingSlash {
|
||||
if statusCode, ok := t.redirectStatusCode(r.Method); ok {
|
||||
var h HandlerFunc
|
||||
if n.addSlash {
|
||||
// Need to add a slash.
|
||||
h = redirectHandler(path+"/", statusCode)
|
||||
} else if path != "/" {
|
||||
// We need to remove the slash. This was already done at the
|
||||
// beginning of the function.
|
||||
h = redirectHandler(path, statusCode)
|
||||
}
|
||||
|
||||
if h != nil {
|
||||
return LookupResult{statusCode, h, nil, nil}, true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var paramMap map[string]string
|
||||
if len(params) != 0 {
|
||||
if len(params) != len(n.leafWildcardNames) {
|
||||
// Need better behavior here. Should this be a panic?
|
||||
panic(fmt.Sprintf("httptreemux parameter list length mismatch: %v, %v",
|
||||
params, n.leafWildcardNames))
|
||||
}
|
||||
|
||||
paramMap = make(map[string]string)
|
||||
numParams := len(params)
|
||||
for index := 0; index < numParams; index++ {
|
||||
paramMap[n.leafWildcardNames[numParams-index-1]] = params[index]
|
||||
}
|
||||
}
|
||||
|
||||
return LookupResult{http.StatusOK, handler, paramMap, nil}, true
|
||||
}
|
||||
|
||||
// Lookup performs a lookup without actually serving the request or mutating the request or response.
|
||||
// The return values are a LookupResult and a boolean. The boolean will be true when a handler
|
||||
// was found or the lookup resulted in a redirect which will point to a real handler. It is false
|
||||
// for requests which would result in a `StatusNotFound` or `StatusMethodNotAllowed`.
|
||||
//
|
||||
// Regardless of the returned boolean's value, the LookupResult may be passed to ServeLookupResult
|
||||
// to be served appropriately.
|
||||
func (t *TreeMux) Lookup(w http.ResponseWriter, r *http.Request) (LookupResult, bool) {
|
||||
if t.SafeAddRoutesWhileRunning {
|
||||
// In concurrency safe mode, we acquire a read lock on the mutex for any access.
|
||||
// This is optional to avoid potential performance loss in high-usage scenarios.
|
||||
t.mutex.RLock()
|
||||
}
|
||||
|
||||
result, found := t.lookup(w, r)
|
||||
|
||||
if t.SafeAddRoutesWhileRunning {
|
||||
t.mutex.RUnlock()
|
||||
}
|
||||
|
||||
return result, found
|
||||
}
|
||||
|
||||
// ServeLookupResult serves a request, given a lookup result from the Lookup function.
|
||||
func (t *TreeMux) ServeLookupResult(w http.ResponseWriter, r *http.Request, lr LookupResult) {
|
||||
if lr.handler == nil {
|
||||
if lr.StatusCode == http.StatusMethodNotAllowed && lr.leafHandler != nil {
|
||||
t.MethodNotAllowedHandler(w, r, lr.leafHandler)
|
||||
} else {
|
||||
t.NotFoundHandler(w, r)
|
||||
}
|
||||
} else {
|
||||
r = t.setDefaultRequestContext(r)
|
||||
lr.handler(w, r, lr.params)
|
||||
}
|
||||
}
|
||||
|
||||
func (t *TreeMux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
if t.PanicHandler != nil {
|
||||
defer t.serveHTTPPanic(w, r)
|
||||
}
|
||||
|
||||
if t.SafeAddRoutesWhileRunning {
|
||||
// In concurrency safe mode, we acquire a read lock on the mutex for any access.
|
||||
// This is optional to avoid potential performance loss in high-usage scenarios.
|
||||
t.mutex.RLock()
|
||||
}
|
||||
|
||||
result, _ := t.lookup(w, r)
|
||||
|
||||
if t.SafeAddRoutesWhileRunning {
|
||||
t.mutex.RUnlock()
|
||||
}
|
||||
|
||||
t.ServeLookupResult(w, r, result)
|
||||
}
|
||||
|
||||
// MethodNotAllowedHandler is the default handler for TreeMux.MethodNotAllowedHandler,
|
||||
// which is called for patterns that match, but do not have a handler installed for the
|
||||
// requested method. It simply writes the status code http.StatusMethodNotAllowed and fills
|
||||
// in the `Allow` header value appropriately.
|
||||
func MethodNotAllowedHandler(w http.ResponseWriter, r *http.Request,
|
||||
methods map[string]HandlerFunc) {
|
||||
|
||||
for m := range methods {
|
||||
w.Header().Add("Allow", m)
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
}
|
||||
|
||||
func New() *TreeMux {
|
||||
tm := &TreeMux{
|
||||
root: &node{path: "/"},
|
||||
NotFoundHandler: http.NotFound,
|
||||
MethodNotAllowedHandler: MethodNotAllowedHandler,
|
||||
HeadCanUseGet: true,
|
||||
RedirectTrailingSlash: true,
|
||||
RedirectCleanPath: true,
|
||||
RedirectBehavior: Redirect301,
|
||||
RedirectMethodBehavior: make(map[string]RedirectBehavior),
|
||||
PathSource: RequestURI,
|
||||
EscapeAddedRoutes: false,
|
||||
}
|
||||
tm.Group.mux = tm
|
||||
return tm
|
||||
}
|
341
vendor/github.com/dimfeld/httptreemux/tree.go
generated
vendored
Normal file
341
vendor/github.com/dimfeld/httptreemux/tree.go
generated
vendored
Normal file
|
@ -0,0 +1,341 @@
|
|||
package httptreemux
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/url"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type node struct {
|
||||
path string
|
||||
|
||||
priority int
|
||||
|
||||
// The list of static children to check.
|
||||
staticIndices []byte
|
||||
staticChild []*node
|
||||
|
||||
// If none of the above match, check the wildcard children
|
||||
wildcardChild *node
|
||||
|
||||
// If none of the above match, then we use the catch-all, if applicable.
|
||||
catchAllChild *node
|
||||
|
||||
// Data for the node is below.
|
||||
|
||||
addSlash bool
|
||||
isCatchAll bool
|
||||
// If true, the head handler was set implicitly, so let it also be set explicitly.
|
||||
implicitHead bool
|
||||
// If this node is the end of the URL, then call the handler, if applicable.
|
||||
leafHandler map[string]HandlerFunc
|
||||
|
||||
// The names of the parameters to apply.
|
||||
leafWildcardNames []string
|
||||
}
|
||||
|
||||
func (n *node) sortStaticChild(i int) {
|
||||
for i > 0 && n.staticChild[i].priority > n.staticChild[i-1].priority {
|
||||
n.staticChild[i], n.staticChild[i-1] = n.staticChild[i-1], n.staticChild[i]
|
||||
n.staticIndices[i], n.staticIndices[i-1] = n.staticIndices[i-1], n.staticIndices[i]
|
||||
i -= 1
|
||||
}
|
||||
}
|
||||
|
||||
func (n *node) setHandler(verb string, handler HandlerFunc, implicitHead bool) {
|
||||
if n.leafHandler == nil {
|
||||
n.leafHandler = make(map[string]HandlerFunc)
|
||||
}
|
||||
_, ok := n.leafHandler[verb]
|
||||
if ok && (verb != "HEAD" || !n.implicitHead) {
|
||||
panic(fmt.Sprintf("%s already handles %s", n.path, verb))
|
||||
}
|
||||
n.leafHandler[verb] = handler
|
||||
|
||||
if verb == "HEAD" {
|
||||
n.implicitHead = implicitHead
|
||||
}
|
||||
}
|
||||
|
||||
func (n *node) addPath(path string, wildcards []string, inStaticToken bool) *node {
|
||||
leaf := len(path) == 0
|
||||
if leaf {
|
||||
if wildcards != nil {
|
||||
// Make sure the current wildcards are the same as the old ones.
|
||||
// If not then we have an ambiguous path.
|
||||
if n.leafWildcardNames != nil {
|
||||
if len(n.leafWildcardNames) != len(wildcards) {
|
||||
// This should never happen.
|
||||
panic("Reached leaf node with differing wildcard array length. Please report this as a bug.")
|
||||
}
|
||||
|
||||
for i := 0; i < len(wildcards); i++ {
|
||||
if n.leafWildcardNames[i] != wildcards[i] {
|
||||
panic(fmt.Sprintf("Wildcards %v are ambiguous with wildcards %v",
|
||||
n.leafWildcardNames, wildcards))
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// No wildcards yet, so just add the existing set.
|
||||
n.leafWildcardNames = wildcards
|
||||
}
|
||||
}
|
||||
|
||||
return n
|
||||
}
|
||||
|
||||
c := path[0]
|
||||
nextSlash := strings.Index(path, "/")
|
||||
var thisToken string
|
||||
var tokenEnd int
|
||||
|
||||
if c == '/' {
|
||||
// Done processing the previous token, so reset inStaticToken to false.
|
||||
thisToken = "/"
|
||||
tokenEnd = 1
|
||||
} else if nextSlash == -1 {
|
||||
thisToken = path
|
||||
tokenEnd = len(path)
|
||||
} else {
|
||||
thisToken = path[0:nextSlash]
|
||||
tokenEnd = nextSlash
|
||||
}
|
||||
remainingPath := path[tokenEnd:]
|
||||
|
||||
if c == '*' && !inStaticToken {
|
||||
// Token starts with a *, so it's a catch-all
|
||||
thisToken = thisToken[1:]
|
||||
if n.catchAllChild == nil {
|
||||
n.catchAllChild = &node{path: thisToken, isCatchAll: true}
|
||||
}
|
||||
|
||||
if path[1:] != n.catchAllChild.path {
|
||||
panic(fmt.Sprintf("Catch-all name in %s doesn't match %s. You probably tried to define overlapping catchalls",
|
||||
path, n.catchAllChild.path))
|
||||
}
|
||||
|
||||
if nextSlash != -1 {
|
||||
panic("/ after catch-all found in " + path)
|
||||
}
|
||||
|
||||
if wildcards == nil {
|
||||
wildcards = []string{thisToken}
|
||||
} else {
|
||||
wildcards = append(wildcards, thisToken)
|
||||
}
|
||||
n.catchAllChild.leafWildcardNames = wildcards
|
||||
|
||||
return n.catchAllChild
|
||||
} else if c == ':' && !inStaticToken {
|
||||
// Token starts with a :
|
||||
thisToken = thisToken[1:]
|
||||
|
||||
if wildcards == nil {
|
||||
wildcards = []string{thisToken}
|
||||
} else {
|
||||
wildcards = append(wildcards, thisToken)
|
||||
}
|
||||
|
||||
if n.wildcardChild == nil {
|
||||
n.wildcardChild = &node{path: "wildcard"}
|
||||
}
|
||||
|
||||
return n.wildcardChild.addPath(remainingPath, wildcards, false)
|
||||
|
||||
} else {
|
||||
// if strings.ContainsAny(thisToken, ":*") {
|
||||
// panic("* or : in middle of path component " + path)
|
||||
// }
|
||||
|
||||
unescaped := false
|
||||
if len(thisToken) >= 2 && !inStaticToken {
|
||||
if thisToken[0] == '\\' && (thisToken[1] == '*' || thisToken[1] == ':' || thisToken[1] == '\\') {
|
||||
// The token starts with a character escaped by a backslash. Drop the backslash.
|
||||
c = thisToken[1]
|
||||
thisToken = thisToken[1:]
|
||||
unescaped = true
|
||||
}
|
||||
}
|
||||
|
||||
// Set inStaticToken to ensure that the rest of this token is not mistaken
|
||||
// for a wildcard if a prefix split occurs at a '*' or ':'.
|
||||
inStaticToken = (c != '/')
|
||||
|
||||
// Do we have an existing node that starts with the same letter?
|
||||
for i, index := range n.staticIndices {
|
||||
if c == index {
|
||||
// Yes. Split it based on the common prefix of the existing
|
||||
// node and the new one.
|
||||
child, prefixSplit := n.splitCommonPrefix(i, thisToken)
|
||||
|
||||
child.priority++
|
||||
n.sortStaticChild(i)
|
||||
if unescaped {
|
||||
// Account for the removed backslash.
|
||||
prefixSplit++
|
||||
}
|
||||
return child.addPath(path[prefixSplit:], wildcards, inStaticToken)
|
||||
}
|
||||
}
|
||||
|
||||
// No existing node starting with this letter, so create it.
|
||||
child := &node{path: thisToken}
|
||||
|
||||
if n.staticIndices == nil {
|
||||
n.staticIndices = []byte{c}
|
||||
n.staticChild = []*node{child}
|
||||
} else {
|
||||
n.staticIndices = append(n.staticIndices, c)
|
||||
n.staticChild = append(n.staticChild, child)
|
||||
}
|
||||
return child.addPath(remainingPath, wildcards, inStaticToken)
|
||||
}
|
||||
}
|
||||
|
||||
func (n *node) splitCommonPrefix(existingNodeIndex int, path string) (*node, int) {
|
||||
childNode := n.staticChild[existingNodeIndex]
|
||||
|
||||
if strings.HasPrefix(path, childNode.path) {
|
||||
// No split needs to be done. Rather, the new path shares the entire
|
||||
// prefix with the existing node, so the new node is just a child of
|
||||
// the existing one. Or the new path is the same as the existing path,
|
||||
// which means that we just move on to the next token. Either way,
|
||||
// this return accomplishes that
|
||||
return childNode, len(childNode.path)
|
||||
}
|
||||
|
||||
var i int
|
||||
// Find the length of the common prefix of the child node and the new path.
|
||||
for i = range childNode.path {
|
||||
if i == len(path) {
|
||||
break
|
||||
}
|
||||
if path[i] != childNode.path[i] {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
commonPrefix := path[0:i]
|
||||
childNode.path = childNode.path[i:]
|
||||
|
||||
// Create a new intermediary node in the place of the existing node, with
|
||||
// the existing node as a child.
|
||||
newNode := &node{
|
||||
path: commonPrefix,
|
||||
priority: childNode.priority,
|
||||
// Index is the first letter of the non-common part of the path.
|
||||
staticIndices: []byte{childNode.path[0]},
|
||||
staticChild: []*node{childNode},
|
||||
}
|
||||
n.staticChild[existingNodeIndex] = newNode
|
||||
|
||||
return newNode, i
|
||||
}
|
||||
|
||||
func (n *node) search(method, path string) (found *node, handler HandlerFunc, params []string) {
|
||||
// if test != nil {
|
||||
// test.Logf("Searching for %s in %s", path, n.dumpTree("", ""))
|
||||
// }
|
||||
pathLen := len(path)
|
||||
if pathLen == 0 {
|
||||
if len(n.leafHandler) == 0 {
|
||||
return nil, nil, nil
|
||||
} else {
|
||||
return n, n.leafHandler[method], nil
|
||||
}
|
||||
}
|
||||
|
||||
// First see if this matches a static token.
|
||||
firstChar := path[0]
|
||||
for i, staticIndex := range n.staticIndices {
|
||||
if staticIndex == firstChar {
|
||||
child := n.staticChild[i]
|
||||
childPathLen := len(child.path)
|
||||
if pathLen >= childPathLen && child.path == path[:childPathLen] {
|
||||
nextPath := path[childPathLen:]
|
||||
found, handler, params = child.search(method, nextPath)
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
// If we found a node and it had a valid handler, then return here. Otherwise
|
||||
// let's remember that we found this one, but look for a better match.
|
||||
if handler != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if n.wildcardChild != nil {
|
||||
// Didn't find a static token, so check for a wildcard.
|
||||
nextSlash := strings.IndexByte(path, '/')
|
||||
if nextSlash < 0 {
|
||||
nextSlash = pathLen
|
||||
}
|
||||
|
||||
thisToken := path[0:nextSlash]
|
||||
nextToken := path[nextSlash:]
|
||||
|
||||
if len(thisToken) > 0 { // Don't match on empty tokens.
|
||||
wcNode, wcHandler, wcParams := n.wildcardChild.search(method, nextToken)
|
||||
if wcHandler != nil || (found == nil && wcNode != nil) {
|
||||
unescaped, err := url.QueryUnescape(thisToken)
|
||||
if err != nil {
|
||||
unescaped = thisToken
|
||||
}
|
||||
|
||||
if wcParams == nil {
|
||||
wcParams = []string{unescaped}
|
||||
} else {
|
||||
wcParams = append(wcParams, unescaped)
|
||||
}
|
||||
|
||||
if wcHandler != nil {
|
||||
return wcNode, wcHandler, wcParams
|
||||
}
|
||||
|
||||
// Didn't actually find a handler here, so remember that we
|
||||
// found a node but also see if we can fall through to the
|
||||
// catchall.
|
||||
found = wcNode
|
||||
handler = wcHandler
|
||||
params = wcParams
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
catchAllChild := n.catchAllChild
|
||||
if catchAllChild != nil {
|
||||
// Hit the catchall, so just assign the whole remaining path if it
|
||||
// has a matching handler.
|
||||
handler = catchAllChild.leafHandler[method]
|
||||
// Found a handler, or we found a catchall node without a handler.
|
||||
// Either way, return it since there's nothing left to check after this.
|
||||
if handler != nil || found == nil {
|
||||
unescaped, err := url.QueryUnescape(path)
|
||||
if err != nil {
|
||||
unescaped = path
|
||||
}
|
||||
|
||||
return catchAllChild, handler, []string{unescaped}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return found, handler, params
|
||||
}
|
||||
|
||||
func (n *node) dumpTree(prefix, nodeType string) string {
|
||||
line := fmt.Sprintf("%s %02d %s%s [%d] %v wildcards %v\n", prefix, n.priority, nodeType, n.path,
|
||||
len(n.staticChild), n.leafHandler, n.leafWildcardNames)
|
||||
prefix += " "
|
||||
for _, node := range n.staticChild {
|
||||
line += node.dumpTree(prefix, "")
|
||||
}
|
||||
if n.wildcardChild != nil {
|
||||
line += n.wildcardChild.dumpTree(prefix, ":")
|
||||
}
|
||||
if n.catchAllChild != nil {
|
||||
line += n.catchAllChild.dumpTree(prefix, "*")
|
||||
}
|
||||
return line
|
||||
}
|
86
vendor/github.com/dimfeld/httptreemux/treemux_16.go
generated
vendored
Normal file
86
vendor/github.com/dimfeld/httptreemux/treemux_16.go
generated
vendored
Normal file
|
@ -0,0 +1,86 @@
|
|||
// +build !go1.7
|
||||
|
||||
package httptreemux
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type TreeMux struct {
|
||||
root *node
|
||||
mutex sync.RWMutex
|
||||
|
||||
Group
|
||||
|
||||
// The default PanicHandler just returns a 500 code.
|
||||
PanicHandler PanicHandler
|
||||
|
||||
// The default NotFoundHandler is http.NotFound.
|
||||
NotFoundHandler func(w http.ResponseWriter, r *http.Request)
|
||||
|
||||
// Any OPTIONS request that matches a path without its own OPTIONS handler will use this handler,
|
||||
// if set, instead of calling MethodNotAllowedHandler.
|
||||
OptionsHandler HandlerFunc
|
||||
|
||||
// MethodNotAllowedHandler is called when a pattern matches, but that
|
||||
// pattern does not have a handler for the requested method. The default
|
||||
// handler just writes the status code http.StatusMethodNotAllowed and adds
|
||||
// the required Allowed header.
|
||||
// The methods parameter contains the map of each method to the corresponding
|
||||
// handler function.
|
||||
MethodNotAllowedHandler func(w http.ResponseWriter, r *http.Request,
|
||||
methods map[string]HandlerFunc)
|
||||
|
||||
// HeadCanUseGet allows the router to use the GET handler to respond to
|
||||
// HEAD requests if no explicit HEAD handler has been added for the
|
||||
// matching pattern. This is true by default.
|
||||
HeadCanUseGet bool
|
||||
|
||||
// RedirectCleanPath allows the router to try clean the current request path,
|
||||
// if no handler is registered for it, using CleanPath from github.com/dimfeld/httppath.
|
||||
// This is true by default.
|
||||
RedirectCleanPath bool
|
||||
|
||||
// RedirectTrailingSlash enables automatic redirection in case router doesn't find a matching route
|
||||
// for the current request path but a handler for the path with or without the trailing
|
||||
// slash exists. This is true by default.
|
||||
RedirectTrailingSlash bool
|
||||
|
||||
// RemoveCatchAllTrailingSlash removes the trailing slash when a catch-all pattern
|
||||
// is matched, if set to true. By default, catch-all paths are never redirected.
|
||||
RemoveCatchAllTrailingSlash bool
|
||||
|
||||
// RedirectBehavior sets the default redirect behavior when RedirectTrailingSlash or
|
||||
// RedirectCleanPath are true. The default value is Redirect301.
|
||||
RedirectBehavior RedirectBehavior
|
||||
|
||||
// RedirectMethodBehavior overrides the default behavior for a particular HTTP method.
|
||||
// The key is the method name, and the value is the behavior to use for that method.
|
||||
RedirectMethodBehavior map[string]RedirectBehavior
|
||||
|
||||
// PathSource determines from where the router gets its path to search.
|
||||
// By default it pulls the data from the RequestURI member, but this can
|
||||
// be overridden to use URL.Path instead.
|
||||
//
|
||||
// There is a small tradeoff here. Using RequestURI allows the router to handle
|
||||
// encoded slashes (i.e. %2f) in the URL properly, while URL.Path provides
|
||||
// better compatibility with some utility functions in the http
|
||||
// library that modify the Request before passing it to the router.
|
||||
PathSource PathSource
|
||||
|
||||
// EscapeAddedRoutes controls URI escaping behavior when adding a route to the tree.
|
||||
// If set to true, the router will add both the route as originally passed, and
|
||||
// a version passed through URL.EscapedPath. This behavior is disabled by default.
|
||||
EscapeAddedRoutes bool
|
||||
|
||||
// SafeAddRoutesWhileRunning tells the router to protect all accesses to the tree with an RWMutex. This is only needed
|
||||
// if you are going to add routes after the router has already begun serving requests. There is a potential
|
||||
// performance penalty at high load.
|
||||
SafeAddRoutesWhileRunning bool
|
||||
}
|
||||
|
||||
func (t *TreeMux) setDefaultRequestContext(r *http.Request) *http.Request {
|
||||
// Nothing to do on Go 1.6 and before
|
||||
return r
|
||||
}
|
149
vendor/github.com/dimfeld/httptreemux/treemux_17.go
generated
vendored
Normal file
149
vendor/github.com/dimfeld/httptreemux/treemux_17.go
generated
vendored
Normal file
|
@ -0,0 +1,149 @@
|
|||
// +build go1.7
|
||||
|
||||
package httptreemux
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type TreeMux struct {
|
||||
root *node
|
||||
mutex sync.RWMutex
|
||||
|
||||
Group
|
||||
|
||||
// The default PanicHandler just returns a 500 code.
|
||||
PanicHandler PanicHandler
|
||||
|
||||
// The default NotFoundHandler is http.NotFound.
|
||||
NotFoundHandler func(w http.ResponseWriter, r *http.Request)
|
||||
|
||||
// Any OPTIONS request that matches a path without its own OPTIONS handler will use this handler,
|
||||
// if set, instead of calling MethodNotAllowedHandler.
|
||||
OptionsHandler HandlerFunc
|
||||
|
||||
// MethodNotAllowedHandler is called when a pattern matches, but that
|
||||
// pattern does not have a handler for the requested method. The default
|
||||
// handler just writes the status code http.StatusMethodNotAllowed and adds
|
||||
// the required Allowed header.
|
||||
// The methods parameter contains the map of each method to the corresponding
|
||||
// handler function.
|
||||
MethodNotAllowedHandler func(w http.ResponseWriter, r *http.Request,
|
||||
methods map[string]HandlerFunc)
|
||||
|
||||
// HeadCanUseGet allows the router to use the GET handler to respond to
|
||||
// HEAD requests if no explicit HEAD handler has been added for the
|
||||
// matching pattern. This is true by default.
|
||||
HeadCanUseGet bool
|
||||
|
||||
// RedirectCleanPath allows the router to try clean the current request path,
|
||||
// if no handler is registered for it, using CleanPath from github.com/dimfeld/httppath.
|
||||
// This is true by default.
|
||||
RedirectCleanPath bool
|
||||
|
||||
// RedirectTrailingSlash enables automatic redirection in case router doesn't find a matching route
|
||||
// for the current request path but a handler for the path with or without the trailing
|
||||
// slash exists. This is true by default.
|
||||
RedirectTrailingSlash bool
|
||||
|
||||
// RemoveCatchAllTrailingSlash removes the trailing slash when a catch-all pattern
|
||||
// is matched, if set to true. By default, catch-all paths are never redirected.
|
||||
RemoveCatchAllTrailingSlash bool
|
||||
|
||||
// RedirectBehavior sets the default redirect behavior when RedirectTrailingSlash or
|
||||
// RedirectCleanPath are true. The default value is Redirect301.
|
||||
RedirectBehavior RedirectBehavior
|
||||
|
||||
// RedirectMethodBehavior overrides the default behavior for a particular HTTP method.
|
||||
// The key is the method name, and the value is the behavior to use for that method.
|
||||
RedirectMethodBehavior map[string]RedirectBehavior
|
||||
|
||||
// PathSource determines from where the router gets its path to search.
|
||||
// By default it pulls the data from the RequestURI member, but this can
|
||||
// be overridden to use URL.Path instead.
|
||||
//
|
||||
// There is a small tradeoff here. Using RequestURI allows the router to handle
|
||||
// encoded slashes (i.e. %2f) in the URL properly, while URL.Path provides
|
||||
// better compatibility with some utility functions in the http
|
||||
// library that modify the Request before passing it to the router.
|
||||
PathSource PathSource
|
||||
|
||||
// EscapeAddedRoutes controls URI escaping behavior when adding a route to the tree.
|
||||
// If set to true, the router will add both the route as originally passed, and
|
||||
// a version passed through URL.EscapedPath. This behavior is disabled by default.
|
||||
EscapeAddedRoutes bool
|
||||
|
||||
// If present, override the default context with this one.
|
||||
DefaultContext context.Context
|
||||
|
||||
// SafeAddRoutesWhileRunning tells the router to protect all accesses to the tree with an RWMutex. This is only needed
|
||||
// if you are going to add routes after the router has already begun serving requests. There is a potential
|
||||
// performance penalty at high load.
|
||||
SafeAddRoutesWhileRunning bool
|
||||
}
|
||||
|
||||
func (t *TreeMux) setDefaultRequestContext(r *http.Request) *http.Request {
|
||||
if t.DefaultContext != nil {
|
||||
r = r.WithContext(t.DefaultContext)
|
||||
}
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
type ContextMux struct {
|
||||
*TreeMux
|
||||
*ContextGroup
|
||||
}
|
||||
|
||||
// NewContextMux returns a TreeMux preconfigured to work with standard http
|
||||
// Handler functions and context objects.
|
||||
func NewContextMux() *ContextMux {
|
||||
mux := New()
|
||||
cg := mux.UsingContext()
|
||||
|
||||
return &ContextMux{
|
||||
TreeMux: mux,
|
||||
ContextGroup: cg,
|
||||
}
|
||||
}
|
||||
|
||||
func (cm *ContextMux) NewGroup(path string) *ContextGroup {
|
||||
return cm.ContextGroup.NewGroup(path)
|
||||
}
|
||||
|
||||
// GET is convenience method for handling GET requests on a context group.
|
||||
func (cm *ContextMux) GET(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("GET", path, handler)
|
||||
}
|
||||
|
||||
// POST is convenience method for handling POST requests on a context group.
|
||||
func (cm *ContextMux) POST(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("POST", path, handler)
|
||||
}
|
||||
|
||||
// PUT is convenience method for handling PUT requests on a context group.
|
||||
func (cm *ContextMux) PUT(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("PUT", path, handler)
|
||||
}
|
||||
|
||||
// DELETE is convenience method for handling DELETE requests on a context group.
|
||||
func (cm *ContextMux) DELETE(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("DELETE", path, handler)
|
||||
}
|
||||
|
||||
// PATCH is convenience method for handling PATCH requests on a context group.
|
||||
func (cm *ContextMux) PATCH(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("PATCH", path, handler)
|
||||
}
|
||||
|
||||
// HEAD is convenience method for handling HEAD requests on a context group.
|
||||
func (cm *ContextMux) HEAD(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("HEAD", path, handler)
|
||||
}
|
||||
|
||||
// OPTIONS is convenience method for handling OPTIONS requests on a context group.
|
||||
func (cm *ContextMux) OPTIONS(path string, handler http.HandlerFunc) {
|
||||
cm.ContextGroup.Handle("OPTIONS", path, handler)
|
||||
}
|
8
vendor/vendor.json
vendored
8
vendor/vendor.json
vendored
|
@ -1,6 +1,6 @@
|
|||
{
|
||||
"comment": "",
|
||||
"ignore": "test github.com/drone/mq/ github.com/tidwall/redlog/ google.golang.org/appengine/ github.com/syndtr/goleveldb/ github.com/drone/drone-ui/ github.com/golang/protobuf/proto github.com/golang/protobuf/ptypes/any golang.org/x/net/context golang.org/x/net/context/ctxhttp",
|
||||
"ignore": "test github.com/drone/mq/ github.com/tidwall/redlog/ google.golang.org/appengine/ github.com/syndtr/goleveldb/ github.com/drone/drone-ui/ github.com/bradrydzewski/drone-ui-unit/dist/ github.com/golang/protobuf/proto github.com/golang/protobuf/ptypes/any golang.org/x/net/context golang.org/x/net/context/ctxhttp",
|
||||
"package": [
|
||||
{
|
||||
"path": "appengine/cloudsql",
|
||||
|
@ -143,6 +143,12 @@
|
|||
"revision": "c1da56349675b292d3200463e2c88b9aa5e02391",
|
||||
"revisionTime": "2015-09-04T14:24:56-07:00"
|
||||
},
|
||||
{
|
||||
"checksumSHA1": "sJwmO9GIFKXj4Ld94n9iAGSg9Jw=",
|
||||
"path": "github.com/dimfeld/httptreemux",
|
||||
"revision": "4faa6bbd32f9d5bad5f76288fa1a9e86574f5d46",
|
||||
"revisionTime": "2017-07-28T18:11:13Z"
|
||||
},
|
||||
{
|
||||
"checksumSHA1": "rSxOx+SnSLAxR4ST8fSz9hhJLdk=",
|
||||
"origin": "github.com/docker/docker/vendor/github.com/docker/distribution/reference",
|
||||
|
|
Loading…
Reference in a new issue