mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-26 11:01:07 +01:00
b4646b665f
Implement a new API endpoint to retrieve real-time statistics for Proxmox LXC containers, similar to `docker stats` functionality. Changes: - Add `GET /api/v1/proxmox/stats/:node/:vmid` endpoint with HTTP and WebSocket support - Implement resource polling loop to cache VM metadata every 3 seconds - Create `LXCStats()` method with streaming (websocket) and single-shot modes - Format output as: STATUS|CPU%|MEM USAGE/LIMIT|MEM%|NET I/O|BLOCK I/O - Add `GetResource()` method for efficient VM resource lookup by kind and ID - Fix task creation bug using correct client reference Example response: running|31.1%|9.6GiB/20GiB|48.87%|4.7GiB/3.3GiB|25GiB/36GiB
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package proxmoxapi
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/yusing/godoxy/internal/proxmox"
|
|
"github.com/yusing/goutils/apitypes"
|
|
"github.com/yusing/goutils/http/httpheaders"
|
|
"github.com/yusing/goutils/http/websocket"
|
|
)
|
|
|
|
type StatsRequest struct {
|
|
Node string `uri:"node" binding:"required"`
|
|
VMID int `uri:"vmid" binding:"required"`
|
|
}
|
|
|
|
// @x-id "stats"
|
|
// @BasePath /api/v1
|
|
// @Summary Get proxmox stats
|
|
// @Description Get proxmox stats in format of "STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET I/O|BLOCK I/O"
|
|
// @Tags proxmox,websocket
|
|
// @Accept json
|
|
// @Produce application/json
|
|
// @Param path path StatsRequest true "Request"
|
|
// @Success 200 string plain "Stats output"
|
|
// @Failure 400 {object} apitypes.ErrorResponse "Invalid request"
|
|
// @Failure 403 {object} apitypes.ErrorResponse "Unauthorized"
|
|
// @Failure 404 {object} apitypes.ErrorResponse "Node not found"
|
|
// @Failure 500 {object} apitypes.ErrorResponse "Internal server error"
|
|
// @Router /api/v1/proxmox/stats/{node}/{vmid} [get]
|
|
func Stats(c *gin.Context) {
|
|
var request StatsRequest
|
|
if err := c.ShouldBindUri(&request); err != nil {
|
|
c.JSON(http.StatusBadRequest, apitypes.Error("invalid request", err))
|
|
return
|
|
}
|
|
|
|
node, ok := proxmox.Nodes.Get(request.Node)
|
|
if !ok {
|
|
c.JSON(http.StatusNotFound, apitypes.Error("node not found"))
|
|
return
|
|
}
|
|
|
|
isWs := httpheaders.IsWebsocket(c.Request.Header)
|
|
|
|
reader, err := node.LXCStats(c.Request.Context(), request.VMID, isWs)
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to get stats"))
|
|
return
|
|
}
|
|
defer reader.Close()
|
|
|
|
if !isWs {
|
|
var line [128]byte
|
|
n, err := reader.Read(line[:])
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to copy stats"))
|
|
return
|
|
}
|
|
c.Data(http.StatusOK, "text/plain; charset=utf-8", line[:n])
|
|
return
|
|
}
|
|
|
|
manager, err := websocket.NewManagerWithUpgrade(c)
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to upgrade to websocket"))
|
|
return
|
|
}
|
|
defer manager.Close()
|
|
|
|
writer := manager.NewWriter(websocket.TextMessage)
|
|
_, err = io.Copy(writer, reader)
|
|
if err != nil {
|
|
c.Error(apitypes.InternalServerError(err, "failed to copy stats"))
|
|
return
|
|
}
|
|
}
|