mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-18 23:33:51 +01:00
db2eda49f1
Moved non-agent-specific logic from agent/pkg/agent/ to internal/agentpool/: - pool.go: Agent pool management (Get, Add, Remove, List, Iter, etc.) - http_requests.go: HTTP utilities (health checks, forwarding, websockets, reverse proxy) - agent.go: Agent struct with HTTP client management This separates general-purpose pool management from agent-specific configuration, improving code organization and making the agent package focused on agent config only.
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
package metrics
|
|
|
|
import (
|
|
"io"
|
|
"maps"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
agentPkg "github.com/yusing/godoxy/agent/pkg/agent"
|
|
"github.com/yusing/godoxy/internal/agentpool"
|
|
"github.com/yusing/godoxy/internal/metrics/period"
|
|
"github.com/yusing/godoxy/internal/metrics/systeminfo"
|
|
apitypes "github.com/yusing/goutils/apitypes"
|
|
"github.com/yusing/goutils/http/httpheaders"
|
|
"github.com/yusing/goutils/synk"
|
|
)
|
|
|
|
type SystemInfoRequest struct {
|
|
AgentAddr string `query:"agent_addr"`
|
|
AgentName string `query:"agent_name"`
|
|
Aggregate systeminfo.SystemInfoAggregateMode `query:"aggregate"`
|
|
Period period.Filter `query:"period"`
|
|
} // @name SystemInfoRequest
|
|
|
|
type SystemInfoAggregate period.ResponseType[systeminfo.Aggregated] // @name SystemInfoAggregate
|
|
|
|
// @x-id "system_info"
|
|
// @BasePath /api/v1
|
|
// @Summary Get system info
|
|
// @Description Get system info
|
|
// @Tags metrics,websocket
|
|
// @Produce json
|
|
// @Param request query SystemInfoRequest false "Request"
|
|
// @Success 200 {object} systeminfo.SystemInfo "no period specified"
|
|
// @Success 200 {object} SystemInfoAggregate "period specified"
|
|
// @Failure 400 {object} apitypes.ErrorResponse
|
|
// @Failure 403 {object} apitypes.ErrorResponse
|
|
// @Failure 404 {object} apitypes.ErrorResponse
|
|
// @Failure 500 {object} apitypes.ErrorResponse
|
|
// @Router /metrics/system_info [get]
|
|
func SystemInfo(c *gin.Context) {
|
|
query := c.Request.URL.Query()
|
|
agentAddr := query.Get("agent_addr")
|
|
agentName := query.Get("agent_name")
|
|
query.Del("agent_addr")
|
|
query.Del("agent_name")
|
|
if agentAddr == "" && agentName == "" {
|
|
systeminfo.Poller.ServeHTTP(c)
|
|
return
|
|
}
|
|
c.Request.URL.RawQuery = query.Encode()
|
|
|
|
agent, ok := agentpool.Get(agentAddr)
|
|
if !ok {
|
|
agent, ok = agentpool.GetAgent(agentName)
|
|
}
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, apitypes.Error("agent_addr or agent_name not found"))
|
|
return
|
|
}
|
|
|
|
isWS := httpheaders.IsWebsocket(c.Request.Header)
|
|
if !isWS {
|
|
resp, err := agent.Forward(c.Request, agentPkg.EndpointSystemInfo)
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to forward request to agent"))
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
maps.Copy(c.Writer.Header(), resp.Header)
|
|
c.Status(resp.StatusCode)
|
|
|
|
pool := synk.GetSizedBytesPool()
|
|
buf := pool.GetSized(16384)
|
|
_, err = io.CopyBuffer(c.Writer, resp.Body, buf)
|
|
pool.Put(buf)
|
|
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to copy response to client"))
|
|
return
|
|
}
|
|
} else {
|
|
agent.ReverseProxy(c.Writer, c.Request, agentPkg.EndpointSystemInfo)
|
|
}
|
|
}
|