context add RelativePath

This commit is contained in:
王守虎 2019-02-15 15:02:44 +08:00
parent 17031001f0
commit d370c327b2
4 changed files with 8 additions and 30 deletions

View File

@ -20,7 +20,6 @@ import (
"github.com/gin-contrib/sse" "github.com/gin-contrib/sse"
"github.com/gin-gonic/gin/binding" "github.com/gin-gonic/gin/binding"
"github.com/gin-gonic/gin/render" "github.com/gin-gonic/gin/render"
"fmt"
) )
// Content-Type MIME of the most common data formats. // Content-Type MIME of the most common data formats.
@ -54,6 +53,8 @@ type Context struct {
// Keys is a key/value pair exclusively for the context of each request. // Keys is a key/value pair exclusively for the context of each request.
Keys map[string]interface{} Keys map[string]interface{}
RelativePath string
// Errors is a list of errors attached to all the handlers/middlewares who used this context. // Errors is a list of errors attached to all the handlers/middlewares who used this context.
Errors errorMsgs Errors errorMsgs
@ -635,10 +636,6 @@ func (c *Context) ShouldBindBodyWith(obj interface{}, bb binding.BindingBody) (e
// X-Real-IP and X-Forwarded-For in order to work properly with reverse-proxies such us: nginx or haproxy. // X-Real-IP and X-Forwarded-For in order to work properly with reverse-proxies such us: nginx or haproxy.
// Use X-Forwarded-For before X-Real-Ip as nginx uses X-Real-Ip with the proxy's IP. // Use X-Forwarded-For before X-Real-Ip as nginx uses X-Real-Ip with the proxy's IP.
func (c *Context) ClientIP() string { func (c *Context) ClientIP() string {
root:=c.engine.trees[0].root
fmt.Println("wsh2:",(*root.children[0]).allPath,root.indices)
fmt.Println("wsh002:",(*(root.children[1])),root.indices)
fmt.Println("wsh02:",len(root.children))
if c.engine.ForwardedByClientIP { if c.engine.ForwardedByClientIP {
clientIP := c.requestHeader("X-Forwarded-For") clientIP := c.requestHeader("X-Forwarded-For")
clientIP = strings.TrimSpace(strings.Split(clientIP, ",")[0]) clientIP = strings.TrimSpace(strings.Split(clientIP, ",")[0])

View File

@ -30,7 +30,6 @@ func debugPrintRoute(httpMethod, absolutePath string, handlers HandlersChain) {
nuHandlers := len(handlers) nuHandlers := len(handlers)
handlerName := nameOfFunction(handlers.Last()) handlerName := nameOfFunction(handlers.Last())
if DebugPrintRouteFunc == nil { if DebugPrintRouteFunc == nil {
fmt.Println("wsh1:",absolutePath)
debugPrint("%-6s %-25s --> %s (%d handlers)\n", httpMethod, absolutePath, handlerName, nuHandlers) debugPrint("%-6s %-25s --> %s (%d handlers)\n", httpMethod, absolutePath, handlerName, nuHandlers)
} else { } else {
DebugPrintRouteFunc(httpMethod, absolutePath, handlerName, nuHandlers) DebugPrintRouteFunc(httpMethod, absolutePath, handlerName, nuHandlers)

4
gin.go
View File

@ -376,11 +376,11 @@ func (engine *Engine) handleHTTPRequest(c *Context) {
} }
root := t[i].root root := t[i].root
// Find route in tree // Find route in tree
handlers, params, tsr,_ := root.getValue(path, c.Params, unescape) handlers, params, tsr,relativePath := root.getValue(path, c.Params, unescape)
if handlers != nil { if handlers != nil {
c.handlers = handlers c.handlers = handlers
c.Params = params c.Params = params
fmt.Println("params:",params) c.RelativePath=relativePath
c.Next() c.Next()
c.writermem.WriteHeaderNow() c.writermem.WriteHeaderNow()
return return

26
tree.go
View File

@ -8,7 +8,6 @@ import (
"net/url" "net/url"
"strings" "strings"
"unicode" "unicode"
"fmt"
) )
// Param is a single URL parameter, consisting of a key and a value. // Param is a single URL parameter, consisting of a key and a value.
@ -95,7 +94,6 @@ type node struct {
nType nodeType nType nodeType
maxParams uint8 maxParams uint8
wildChild bool wildChild bool
allPath string
} }
// increments priority of the given child and reorders if necessary. // increments priority of the given child and reorders if necessary.
@ -125,7 +123,6 @@ func (n *node) incrementChildPrio(pos int) int {
// addRoute adds a node with the given handle to the path. // addRoute adds a node with the given handle to the path.
// Not concurrency-safe! // Not concurrency-safe!
func (n *node) addRoute(path string, handlers HandlersChain) { func (n *node) addRoute(path string, handlers HandlersChain) {
fmt.Println("wsh66:",path)
fullPath := path fullPath := path
n.priority++ n.priority++
numParams := countParams(path) numParams := countParams(path)
@ -156,7 +153,6 @@ func (n *node) addRoute(path string, handlers HandlersChain) {
indices: n.indices, indices: n.indices,
children: n.children, children: n.children,
handlers: n.handlers, handlers: n.handlers,
allPath:path,
priority: n.priority - 1, priority: n.priority - 1,
} }
@ -172,7 +168,6 @@ func (n *node) addRoute(path string, handlers HandlersChain) {
n.indices = string([]byte{n.path[i]}) n.indices = string([]byte{n.path[i]})
n.path = path[:i] n.path = path[:i]
n.handlers = nil n.handlers = nil
n.allPath=path
n.wildChild = false n.wildChild = false
} }
@ -240,14 +235,12 @@ func (n *node) addRoute(path string, handlers HandlersChain) {
n = child n = child
} }
n.insertChild(numParams, path, fullPath, handlers) n.insertChild(numParams, path, fullPath, handlers)
n.allPath=path
return return
} else if i == len(path) { // Make node a (in-path) leaf } else if i == len(path) { // Make node a (in-path) leaf
if n.handlers != nil { if n.handlers != nil {
panic("handlers are already registered for path '" + fullPath + "'") panic("handlers are already registered for path '" + fullPath + "'")
} }
n.allPath=path
n.handlers = handlers n.handlers = handlers
} }
return return
@ -255,7 +248,6 @@ func (n *node) addRoute(path string, handlers HandlersChain) {
} else { // Empty tree } else { // Empty tree
n.insertChild(numParams, path, fullPath, handlers) n.insertChild(numParams, path, fullPath, handlers)
n.nType = root n.nType = root
n.allPath=path
} }
} }
@ -368,7 +360,6 @@ func (n *node) insertChild(numParams uint8, path string, fullPath string, handle
} }
// insert remaining path part and handle to the leaf // insert remaining path part and handle to the leaf
fmt.Println("wsh21:",path[offset:])
n.path = path[offset:] n.path = path[offset:]
n.handlers = handlers n.handlers = handlers
} }
@ -378,11 +369,7 @@ func (n *node) insertChild(numParams uint8, path string, fullPath string, handle
// If no handle can be found, a TSR (trailing slash redirect) recommendation is // If no handle can be found, a TSR (trailing slash redirect) recommendation is
// made if a handle exists with an extra (without the) trailing slash for the // made if a handle exists with an extra (without the) trailing slash for the
// given path. // given path.
func (n *node) getValue(path string, po Params, unescape bool) (handlers HandlersChain, p Params, tsr bool,regPath string) { func (n *node) getValue(path string, po Params, unescape bool) (handlers HandlersChain, p Params, tsr bool,relativePath string) {
regPath=""
defer func() {
fmt.Println("tsr:",tsr,"regpath:",regPath)
}()
p = po p = po
walk: // Outer loop for walking the tree walk: // Outer loop for walking the tree
for { for {
@ -392,13 +379,12 @@ walk: // Outer loop for walking the tree
// If this node does not have a wildcard (param or catchAll) // If this node does not have a wildcard (param or catchAll)
// child, we can just look up the next child node and continue // child, we can just look up the next child node and continue
// to walk down the tree // to walk down the tree
relativePath+=n.path
if !n.wildChild { if !n.wildChild {
c := path[0] c := path[0]
for i := 0; i < len(n.indices); i++ { for i := 0; i < len(n.indices); i++ {
if c == n.indices[i] { if c == n.indices[i] {
n = n.children[i] n = n.children[i]
regPath+=n.path
fmt.Println("path4:",n.path,regPath)
continue walk continue walk
} }
} }
@ -430,8 +416,7 @@ walk: // Outer loop for walking the tree
val := path[:end] val := path[:end]
regPath+=n.path relativePath+=n.path
fmt.Println("path6:",n.path,regPath)
if unescape { if unescape {
var err error var err error
if p[i].Value, err = url.QueryUnescape(val); err != nil { if p[i].Value, err = url.QueryUnescape(val); err != nil {
@ -475,7 +460,6 @@ walk: // Outer loop for walking the tree
i := len(p) i := len(p)
p = p[:i+1] // expand slice within preallocated capacity p = p[:i+1] // expand slice within preallocated capacity
p[i].Key = n.path[2:] p[i].Key = n.path[2:]
fmt.Println("path5:",n.path,n.allPath)
if unescape { if unescape {
var err error var err error
if p[i].Value, err = url.QueryUnescape(path); err != nil { if p[i].Value, err = url.QueryUnescape(path); err != nil {
@ -493,8 +477,7 @@ walk: // Outer loop for walking the tree
} }
} }
} else if path == n.path { } else if path == n.path {
fmt.Println("path2:",n.path) relativePath+=n.path
regPath+=n.path
// We should have reached the node containing the handle. // We should have reached the node containing the handle.
// Check if this node has a handle registered. // Check if this node has a handle registered.
if handlers = n.handlers; handlers != nil { if handlers = n.handlers; handlers != nil {
@ -510,7 +493,6 @@ walk: // Outer loop for walking the tree
// trailing slash exists for trailing slash recommendation // trailing slash exists for trailing slash recommendation
for i := 0; i < len(n.indices); i++ { for i := 0; i < len(n.indices); i++ {
if n.indices[i] == '/' { if n.indices[i] == '/' {
fmt.Println("path3:",n.children[i].path)
n = n.children[i] n = n.children[i]
tsr = (len(n.path) == 1 && n.handlers != nil) || tsr = (len(n.path) == 1 && n.handlers != nil) ||
(n.nType == catchAll && n.children[0].handlers != nil) (n.nType == catchAll && n.children[0].handlers != nil)