Compare commits

..

No commits in common. "master" and "v0.1.5" have entirely different histories.

4 changed files with 7 additions and 39 deletions

View file

@ -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
View file

@ -1,5 +0,0 @@
module git.fuyu.moe/Fuyu/router
go 1.13
require github.com/julienschmidt/httprouter v1.3.0

2
go.sum
View file

@ -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=

View file

@ -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
} }