mirror of
https://github.com/yusing/godoxy.git
synced 2026-01-11 21:10:30 +01:00
dba8441e8a
- Updated route iteration to include all routes, including excluded ones. - Renamed existing functions for clarity. - Adjusted health info retrieval to reflect changes in route iteration. - Improved route management by adding health monitoring capabilities for excluded routes.
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package routeApi
|
|
|
|
import (
|
|
"net/http"
|
|
"slices"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yusing/godoxy/internal/route"
|
|
"github.com/yusing/godoxy/internal/route/routes"
|
|
"github.com/yusing/godoxy/internal/types"
|
|
"github.com/yusing/goutils/http/httpheaders"
|
|
"github.com/yusing/goutils/http/websocket"
|
|
)
|
|
|
|
type RouteType route.Route // @name Route
|
|
|
|
// @x-id "routes"
|
|
// @BasePath /api/v1
|
|
// @Summary List routes
|
|
// @Description List routes
|
|
// @Tags route,websocket
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param provider query string false "Provider"
|
|
// @Success 200 {array} RouteType
|
|
// @Failure 403 {object} apitypes.ErrorResponse
|
|
// @Router /route/list [get]
|
|
func Routes(c *gin.Context) {
|
|
if httpheaders.IsWebsocket(c.Request.Header) {
|
|
RoutesWS(c)
|
|
return
|
|
}
|
|
|
|
provider := c.Query("provider")
|
|
if provider == "" {
|
|
c.JSON(http.StatusOK, slices.Collect(routes.IterAll))
|
|
return
|
|
}
|
|
|
|
rts := make([]types.Route, 0, routes.NumAllRoutes())
|
|
for r := range routes.IterAll {
|
|
if r.ProviderName() == provider {
|
|
rts = append(rts, r)
|
|
}
|
|
}
|
|
c.JSON(http.StatusOK, rts)
|
|
}
|
|
|
|
func RoutesWS(c *gin.Context) {
|
|
provider := c.Query("provider")
|
|
if provider == "" {
|
|
websocket.PeriodicWrite(c, 3*time.Second, func() (any, error) {
|
|
return slices.Collect(routes.IterAll), nil
|
|
})
|
|
return
|
|
}
|
|
|
|
websocket.PeriodicWrite(c, 3*time.Second, func() (any, error) {
|
|
rts := make([]types.Route, 0, routes.NumAllRoutes())
|
|
for r := range routes.IterAll {
|
|
if r.ProviderName() == provider {
|
|
rts = append(rts, r)
|
|
}
|
|
}
|
|
return rts, nil
|
|
})
|
|
}
|