mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 04:42:33 +02:00
1a5f3735cf
* cleanup code for URL type * fix makefile for trace mode * refactor, merge Entry, RawEntry and Route into one. * Implement fileserver. * refactor: rename HTTPRoute to ReverseProxyRoute to avoid confusion * refactor: move metrics logger to middleware package - fix prometheus metrics for load balanced routes - route will now fail when health monitor fail to start * fix extra output of ls-* commands by defer initializaing stuff, speed up start time * add test for path traversal attack, small fix on FileServer.Start method * rename rule.on.bypass to pass * refactor and fixed map-to-map deserialization * updated route loading logic * schemas: add "add_prefix" option to modify_request middleware * updated route JSONMarshalling --------- Co-authored-by: yusing <yusing@6uo.me>
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package provider
|
|
|
|
import (
|
|
R "github.com/yusing/go-proxy/internal/route"
|
|
"github.com/yusing/go-proxy/internal/route/provider/types"
|
|
route "github.com/yusing/go-proxy/internal/route/types"
|
|
"github.com/yusing/go-proxy/internal/watcher/health"
|
|
)
|
|
|
|
type (
|
|
RouteStats struct {
|
|
Total uint16 `json:"total"`
|
|
NumHealthy uint16 `json:"healthy"`
|
|
NumUnhealthy uint16 `json:"unhealthy"`
|
|
NumNapping uint16 `json:"napping"`
|
|
NumError uint16 `json:"error"`
|
|
NumUnknown uint16 `json:"unknown"`
|
|
}
|
|
ProviderStats struct {
|
|
Total uint16 `json:"total"`
|
|
RPs RouteStats `json:"reverse_proxies"`
|
|
Streams RouteStats `json:"streams"`
|
|
Type types.ProviderType `json:"type"`
|
|
}
|
|
)
|
|
|
|
func (stats *RouteStats) Add(r *R.Route) {
|
|
stats.Total++
|
|
mon := r.HealthMonitor()
|
|
if mon == nil {
|
|
stats.NumUnknown++
|
|
return
|
|
}
|
|
switch mon.Status() {
|
|
case health.StatusHealthy:
|
|
stats.NumHealthy++
|
|
case health.StatusUnhealthy:
|
|
stats.NumUnhealthy++
|
|
case health.StatusNapping:
|
|
stats.NumNapping++
|
|
case health.StatusError:
|
|
stats.NumError++
|
|
default:
|
|
stats.NumUnknown++
|
|
}
|
|
}
|
|
|
|
func (stats *RouteStats) AddOther(other RouteStats) {
|
|
stats.Total += other.Total
|
|
stats.NumHealthy += other.NumHealthy
|
|
stats.NumUnhealthy += other.NumUnhealthy
|
|
stats.NumNapping += other.NumNapping
|
|
stats.NumError += other.NumError
|
|
stats.NumUnknown += other.NumUnknown
|
|
}
|
|
|
|
func (p *Provider) Statistics() ProviderStats {
|
|
var rps, streams RouteStats
|
|
for _, r := range p.routes {
|
|
switch r.Type() {
|
|
case route.RouteTypeHTTP:
|
|
rps.Add(r)
|
|
case route.RouteTypeStream:
|
|
streams.Add(r)
|
|
}
|
|
}
|
|
return ProviderStats{
|
|
Total: rps.Total + streams.Total,
|
|
RPs: rps,
|
|
Streams: streams,
|
|
Type: p.t,
|
|
}
|
|
}
|