forked from Fuyu/router
Compare commits
No commits in common. "master" and "v0.1.5" have entirely different histories.
4 changed files with 7 additions and 39 deletions
15
context.go
15
context.go
|
@ -105,17 +105,14 @@ func (c *Context) Get(key string) interface{} {
|
||||||
|
|
||||||
// RealIP uses proxy headers for the real ip, if none exist the IP of the current connection is returned
|
// RealIP uses proxy headers for the real ip, if none exist the IP of the current connection is returned
|
||||||
func (c *Context) RealIP() string {
|
func (c *Context) RealIP() string {
|
||||||
reqIP := c.Request.RemoteAddr
|
|
||||||
|
|
||||||
if ip := c.Request.Header.Get(`X-Forwarded-For`); ip != `` {
|
if ip := c.Request.Header.Get(`X-Forwarded-For`); ip != `` {
|
||||||
reqIP = strings.Split(ip, `, `)[0]
|
return strings.Split(ip, `, `)[0]
|
||||||
} else if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
|
|
||||||
reqIP = ip
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ra, _, _ := net.SplitHostPort(reqIP)
|
if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
|
||||||
if ra != `` {
|
return ip
|
||||||
reqIP = ra
|
|
||||||
}
|
}
|
||||||
return reqIP
|
|
||||||
|
ra, _, _ := net.SplitHostPort(c.Request.RemoteAddr)
|
||||||
|
return ra
|
||||||
}
|
}
|
||||||
|
|
5
go.mod
5
go.mod
|
@ -1,5 +0,0 @@
|
||||||
module git.fuyu.moe/Fuyu/router
|
|
||||||
|
|
||||||
go 1.13
|
|
||||||
|
|
||||||
require github.com/julienschmidt/httprouter v1.3.0
|
|
2
go.sum
2
go.sum
|
@ -1,2 +0,0 @@
|
||||||
github.com/julienschmidt/httprouter v1.3.0 h1:U0609e9tgbseu3rBINet9P48AI/D3oJs4dN7jwJOQ1U=
|
|
||||||
github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
|
|
24
router.go
24
router.go
|
@ -3,7 +3,6 @@ package router
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"net"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"reflect"
|
"reflect"
|
||||||
"time"
|
"time"
|
||||||
|
@ -39,7 +38,6 @@ type Router struct {
|
||||||
NotFoundHandler Handle
|
NotFoundHandler Handle
|
||||||
MethodNotAllowedHandler Handle
|
MethodNotAllowedHandler Handle
|
||||||
ErrorHandler ErrorHandle
|
ErrorHandler ErrorHandle
|
||||||
TrimTrailingSlashes bool
|
|
||||||
server *http.Server
|
server *http.Server
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -96,14 +94,6 @@ func (r *Router) OPTIONS(path string, handle Handle, middleware ...Middleware) {
|
||||||
r.routes = append(r.routes, route{`OPTIONS`, path, handle, middleware})
|
r.routes = append(r.routes, route{`OPTIONS`, path, handle, middleware})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Serve accepts incoming HTTP connections on the listener
|
|
||||||
func (r *Router) Serve(l net.Listener) error {
|
|
||||||
httpr := r.getHttpr()
|
|
||||||
|
|
||||||
r.server = &http.Server{Handler: httpr}
|
|
||||||
return r.server.Serve(l)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start starts the web server and binds to the given address
|
// Start starts the web server and binds to the given address
|
||||||
func (r *Router) Start(addr string) error {
|
func (r *Router) Start(addr string) error {
|
||||||
httpr := r.getHttpr()
|
httpr := r.getHttpr()
|
||||||
|
@ -137,7 +127,7 @@ func (r *Router) Stop() error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Router) getHttpr() http.Handler {
|
func (r *Router) getHttpr() *httprouter.Router {
|
||||||
httpr := httprouter.New()
|
httpr := httprouter.New()
|
||||||
|
|
||||||
for _, v := range r.routes {
|
for _, v := range r.routes {
|
||||||
|
@ -166,18 +156,6 @@ func (r *Router) getHttpr() http.Handler {
|
||||||
r.ErrorHandler(c, err)
|
r.ErrorHandler(c, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if r.TrimTrailingSlashes {
|
|
||||||
httpr.RedirectTrailingSlash = false
|
|
||||||
|
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
||||||
l := len(req.URL.Path)
|
|
||||||
if l > 1 && req.URL.Path[l-1] == '/' {
|
|
||||||
req.URL.Path = req.URL.Path[:l-1]
|
|
||||||
}
|
|
||||||
httpr.ServeHTTP(w, req)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return httpr
|
return httpr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue