mirror of
https://github.com/yusing/godoxy.git
synced 2026-02-23 02:44:52 +01:00
31b4fedf72
- Introduced `NewTestRoute` function to simplify route creation in benchmark tests. - Replaced direct route validation and starting with error handling using `require.NoError`. - Updated server retrieval to use `common.ProxyHTTPAddr` for consistency. - Improved logging for HTTP route addition errors in `AddRoute` method. * fix(tcp): wrap proxy proto listener before acl * refactor(entrypoint): propagate errors from route registration and stream serving * fix(docs): correct swagger and package README
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package entrypoint
|
|
|
|
import (
|
|
"github.com/yusing/godoxy/internal/types"
|
|
)
|
|
|
|
// GetHealthInfo returns a map of route name to health info.
|
|
//
|
|
// The health info is for all routes, including excluded routes.
|
|
func (ep *Entrypoint) GetHealthInfo() map[string]types.HealthInfo {
|
|
healthMap := make(map[string]types.HealthInfo, ep.NumRoutes())
|
|
for r := range ep.IterRoutes {
|
|
healthMap[r.Name()] = getHealthInfo(r)
|
|
}
|
|
return healthMap
|
|
}
|
|
|
|
// GetHealthInfoWithoutDetail returns a map of route name to health info without detail.
|
|
//
|
|
// The health info is for all routes, including excluded routes.
|
|
func (ep *Entrypoint) GetHealthInfoWithoutDetail() map[string]types.HealthInfoWithoutDetail {
|
|
healthMap := make(map[string]types.HealthInfoWithoutDetail, ep.NumRoutes())
|
|
for r := range ep.IterRoutes {
|
|
healthMap[r.Name()] = getHealthInfoWithoutDetail(r)
|
|
}
|
|
return healthMap
|
|
}
|
|
|
|
// GetHealthInfoSimple returns a map of route name to health status.
|
|
//
|
|
// The health status is for all routes, including excluded routes.
|
|
func (ep *Entrypoint) GetHealthInfoSimple() map[string]types.HealthStatus {
|
|
healthMap := make(map[string]types.HealthStatus, ep.NumRoutes())
|
|
for r := range ep.IterRoutes {
|
|
healthMap[r.Name()] = getHealthInfoSimple(r)
|
|
}
|
|
return healthMap
|
|
}
|
|
|
|
// RoutesByProvider returns a map of provider name to routes.
|
|
//
|
|
// The routes are all routes, including excluded routes.
|
|
func (ep *Entrypoint) RoutesByProvider() map[string][]types.Route {
|
|
rts := make(map[string][]types.Route)
|
|
for r := range ep.IterRoutes {
|
|
rts[r.ProviderName()] = append(rts[r.ProviderName()], r)
|
|
}
|
|
return rts
|
|
}
|
|
|
|
func getHealthInfo(r types.Route) types.HealthInfo {
|
|
mon := r.HealthMonitor()
|
|
if mon == nil {
|
|
return types.HealthInfo{
|
|
HealthInfoWithoutDetail: types.HealthInfoWithoutDetail{
|
|
Status: types.StatusUnknown,
|
|
},
|
|
Detail: "n/a",
|
|
}
|
|
}
|
|
return types.HealthInfo{
|
|
HealthInfoWithoutDetail: types.HealthInfoWithoutDetail{
|
|
Status: mon.Status(),
|
|
Uptime: mon.Uptime(),
|
|
Latency: mon.Latency(),
|
|
},
|
|
Detail: mon.Detail(),
|
|
}
|
|
}
|
|
|
|
func getHealthInfoWithoutDetail(r types.Route) types.HealthInfoWithoutDetail {
|
|
mon := r.HealthMonitor()
|
|
if mon == nil {
|
|
return types.HealthInfoWithoutDetail{
|
|
Status: types.StatusUnknown,
|
|
}
|
|
}
|
|
return types.HealthInfoWithoutDetail{
|
|
Status: mon.Status(),
|
|
Uptime: mon.Uptime(),
|
|
Latency: mon.Latency(),
|
|
}
|
|
}
|
|
|
|
func getHealthInfoSimple(r types.Route) types.HealthStatus {
|
|
mon := r.HealthMonitor()
|
|
if mon == nil {
|
|
return types.StatusUnknown
|
|
}
|
|
return mon.Status()
|
|
}
|