TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-06-09 04:52:35 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

API server now protected with rate limiter, fixed extra dot on URL on frontend

Browse source
This commit is contained in:
yusing 2024-11-03 10:11:27 +08:00
parent ddbee9ec19
commit 01b7ec2a99
2 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
internal/api/handler.go
View file

@ -9,6 +9,8 @@ import (
"github.com/yusing/go-proxy/internal/api/v1/auth" "github.com/yusing/go-proxy/internal/api/v1/auth"
. "github.com/yusing/go-proxy/internal/api/v1/utils" . "github.com/yusing/go-proxy/internal/api/v1/utils"
"github.com/yusing/go-proxy/internal/common" "github.com/yusing/go-proxy/internal/common"
"github.com/yusing/go-proxy/internal/logging"
"github.com/yusing/go-proxy/internal/net/http/middleware"
) )
type ServeMux struct{ *http.ServeMux } type ServeMux struct{ *http.ServeMux }
@ -18,7 +20,7 @@ func NewServeMux() ServeMux {
} }
func (mux ServeMux) HandleFunc(method, endpoint string, handler http.HandlerFunc) { func (mux ServeMux) HandleFunc(method, endpoint string, handler http.HandlerFunc) {
mux.ServeMux.HandleFunc(fmt.Sprintf("%s %s", method, endpoint), checkHost(handler)) mux.ServeMux.HandleFunc(fmt.Sprintf("%s %s", method, endpoint), checkHost(rateLimited(handler)))
} }
func NewHandler() http.Handler { func NewHandler() http.Handler {
@ -56,3 +58,16 @@ func checkHost(f http.HandlerFunc) http.HandlerFunc {
f(w, r) f(w, r)
} }
} }
func rateLimited(f http.HandlerFunc) http.HandlerFunc {
m, err := middleware.RateLimiter.WithOptionsClone(middleware.OptionsRaw{
"average": 10,
"burst": 10,
})
if err != nil {
logging.Fatal().Err(err).Msg("unable to create API rate limiter")
}
return func(w http.ResponseWriter, r *http.Request) {
m.ModifyRequest(f, w, r)
}
}

2
internal/config/query.go
View file

@ -89,7 +89,7 @@ func HomepageConfig() homepage.Config {
if item.URL == "" { if item.URL == "" {
if len(domains) > 0 { if len(domains) > 0 {
item.URL = fmt.Sprintf("%s://%s.%s:%s", proto, strings.ToLower(alias), domains[0], port) item.URL = fmt.Sprintf("%s://%s%s:%s", proto, strings.ToLower(alias), domains[0], port)
} }
} }
item.AltURL = r.TargetURL().String() item.AltURL = r.TargetURL().String()