mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-23 17:41:05 +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
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package routeApi
|
|
|
|
import (
|
|
"net/http"
|
|
"slices"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
entrypoint "github.com/yusing/godoxy/internal/entrypoint/types"
|
|
"github.com/yusing/godoxy/internal/route"
|
|
"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
|
|
}
|
|
|
|
ep := entrypoint.FromCtx(c.Request.Context())
|
|
|
|
provider := c.Query("provider")
|
|
if provider == "" {
|
|
c.JSON(http.StatusOK, slices.Collect(ep.IterRoutes))
|
|
return
|
|
}
|
|
|
|
rts := make([]types.Route, 0, ep.NumRoutes())
|
|
for r := range ep.IterRoutes {
|
|
if r.ProviderName() == provider {
|
|
rts = append(rts, r)
|
|
}
|
|
}
|
|
c.JSON(http.StatusOK, rts)
|
|
}
|
|
|
|
func RoutesWS(c *gin.Context) {
|
|
ep := entrypoint.FromCtx(c.Request.Context())
|
|
|
|
provider := c.Query("provider")
|
|
if provider == "" {
|
|
websocket.PeriodicWrite(c, 3*time.Second, func() (any, error) {
|
|
return slices.Collect(ep.IterRoutes), nil
|
|
})
|
|
return
|
|
}
|
|
|
|
websocket.PeriodicWrite(c, 3*time.Second, func() (any, error) {
|
|
rts := make([]types.Route, 0, ep.NumRoutes())
|
|
for r := range ep.IterRoutes {
|
|
if r.ProviderName() == provider {
|
|
rts = append(rts, r)
|
|
}
|
|
}
|
|
return rts, nil
|
|
})
|
|
}
|