starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-25 02:21:05 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

feat(proxmox): add node-level stats endpoint with streaming support

Browse Source 
Add new `/proxmox/stats/{node}` API endpoint for retrieving Proxmox node
statistics in JSON format. The endpoint returns kernel version, CPU
usage/model, memory usage, rootfs usage, uptime, and load averages.

The existing `/proxmox/stats/{node}/{vmid}` endpoint has been corrected `VMStats` to return`text/plain` instead of `application/json`.

Both endpoints support WebSocket streaming for real-time stats updates
with a 1-second poll interval.
This commit is contained in:
yusing
2026-01-25 13:50:37 +08:00
parent 51704829c6
commit cb8f405e76
6 changed files with 424 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
internal/api/handler.go
View File

@@ -145,7 +145,8 @@ func NewHandler(requireAuth bool) *gin.Engine {
proxmox.GET("/journalctl/:node", proxmoxApi.Journalctl)
proxmox.GET("/journalctl/:node/:vmid", proxmoxApi.Journalctl)
proxmox.GET("/journalctl/:node/:vmid/:service", proxmoxApi.Journalctl)
proxmox.GET("/stats/:node/:vmid", proxmoxApi.Stats)
proxmox.GET("/stats/:node", proxmoxApi.NodeStats)
proxmox.GET("/stats/:node/:vmid", proxmoxApi.VMStats)
proxmox.POST("/lxc/:node/:vmid/start", proxmoxApi.Start)
proxmox.POST("/lxc/:node/:vmid/stop", proxmoxApi.Stop)
proxmox.POST("/lxc/:node/:vmid/restart", proxmoxApi.Restart)

146
internal/api/v1/docs/swagger.json
View File

@@ -2452,12 +2452,9 @@
"operationId": "lxcStop"
}
},
"/proxmox/stats/{node}/{vmid}": {
"/proxmox/stats/{node}": {
"get": {
"description": "Get proxmox stats in format of \"STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET I/O|BLOCK I/O\"",
"consumes": [
"application/json"
],
"description": "Get proxmox node stats in json",
"produces": [
"application/json"
],
@@ -2465,7 +2462,63 @@
"proxmox",
"websocket"
],
"summary": "Get proxmox stats",
"summary": "Get proxmox node stats",
"parameters": [
{
"type": "string",
"description": "Node name",
"name": "node",
"in": "path",
"required": true
}
],
"responses": {
"200": {
"description": "Stats output",
"schema": {
"$ref": "#/definitions/proxmox.NodeStats"
}
},
"400": {
"description": "Invalid request",
"schema": {
"$ref": "#/definitions/ErrorResponse"
}
},
"403": {
"description": "Unauthorized",
"schema": {
"$ref": "#/definitions/ErrorResponse"
}
},
"404": {
"description": "Node not found",
"schema": {
"$ref": "#/definitions/ErrorResponse"
}
},
"500": {
"description": "Internal server error",
"schema": {
"$ref": "#/definitions/ErrorResponse"
}
}
},
"x-id": "nodeStats",
"operationId": "nodeStats"
}
},
"/proxmox/stats/{node}/{vmid}": {
"get": {
"description": "Get proxmox VM stats in format of \"STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET I/O|BLOCK I/O\"",
"produces": [
"text/plain"
],
"tags": [
"proxmox",
"websocket"
],
"summary": "Get proxmox VM stats",
"parameters": [
{
"type": "string",
@@ -2512,8 +2565,8 @@
}
}
},
"x-id": "stats",
"operationId": "stats"
"x-id": "vmStats",
"operationId": "vmStats"
}
},
"/reload": {
@@ -6331,6 +6384,83 @@
"x-nullable": false,
"x-omitempty": false
},
"proxmox.NodeStats": {
"type": "object",
"properties": {
"cpu_model": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"cpu_usage": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"kernel_version": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"load_avg_15m": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"load_avg_1m": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"load_avg_5m": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"mem_pct": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"mem_total": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"mem_usage": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"pve_version": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"rootfs_pct": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"rootfs_total": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"rootfs_usage": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"uptime": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
}
},
"x-nullable": false,
"x-omitempty": false
},
"route.Route": {
"type": "object",
"properties": {

78
internal/api/v1/docs/swagger.yaml
View File

@@ -1784,6 +1784,37 @@ definitions:
type: object
netip.Addr:
type: object
proxmox.NodeStats:
properties:
cpu_model:
type: string
cpu_usage:
type: string
kernel_version:
type: string
load_avg_15m:
type: string
load_avg_1m:
type: string
load_avg_5m:
type: string
mem_pct:
type: string
mem_total:
type: string
mem_usage:
type: string
pve_version:
type: string
rootfs_pct:
type: string
rootfs_total:
type: string
rootfs_usage:
type: string
uptime:
type: string
type: object
route.Route:
properties:
access_log:
@@ -3611,11 +3642,46 @@ paths:
tags:
- proxmox
x-id: lxcStop
/proxmox/stats/{node}:
get:
description: Get proxmox node stats in json
parameters:
- description: Node name
in: path
name: node
required: true
type: string
produces:
- application/json
responses:
"200":
description: Stats output
schema:
$ref: '#/definitions/proxmox.NodeStats'
"400":
description: Invalid request
schema:
$ref: '#/definitions/ErrorResponse'
"403":
description: Unauthorized
schema:
$ref: '#/definitions/ErrorResponse'
"404":
description: Node not found
schema:
$ref: '#/definitions/ErrorResponse'
"500":
description: Internal server error
schema:
$ref: '#/definitions/ErrorResponse'
summary: Get proxmox node stats
tags:
- proxmox
- websocket
x-id: nodeStats
/proxmox/stats/{node}/{vmid}:
get:
consumes:
- application/json
description: Get proxmox stats in format of "STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET
description: Get proxmox VM stats in format of "STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET
I/O|BLOCK I/O"
parameters:
- in: path
@@ -3627,7 +3693,7 @@ paths:
required: true
type: integer
produces:
- application/json
- text/plain
responses:
"200":
description: Stats output
@@ -3649,11 +3715,11 @@ paths:
description: Internal server error
schema:
$ref: '#/definitions/ErrorResponse'
summary: Get proxmox stats
summary: Get proxmox VM stats
tags:
- proxmox
- websocket
x-id: stats
x-id: vmStats
/reload:
post:
consumes:

72
internal/api/v1/proxmox/stats.go
View File

@@ -16,13 +16,73 @@ type StatsRequest struct {
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"
// @x-id "nodeStats"
// @BasePath /api/v1
// @Summary Get proxmox node stats
// @Description Get proxmox node stats in json
// @Tags proxmox,websocket
// @Accept json
// @Produce application/json
// @Param node path string true "Node name"
// @Success 200 {object} proxmox.NodeStats "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 /proxmox/stats/{node} [get]
func NodeStats(c *gin.Context) {
nodeName := c.Param("node")
if nodeName == "" {
c.JSON(http.StatusBadRequest, apitypes.Error("node name is required"))
return
}
node, ok := proxmox.Nodes.Get(nodeName)
if !ok {
c.JSON(http.StatusNotFound, apitypes.Error("node not found"))
return
}
isWs := httpheaders.IsWebsocket(c.Request.Header)
reader, err := node.NodeStats(c.Request.Context(), isWs)
if err != nil {
c.Error(apitypes.InternalServerError(err, "failed to get stats"))
return
}
defer reader.Close()
if !isWs {
var line [512]byte
n, err := reader.Read(line[:])
if err != nil {
c.Error(apitypes.InternalServerError(err, "failed to copy stats"))
return
}
c.Data(http.StatusOK, "application/json", 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
}
}
// @x-id "vmStats"
// @BasePath /api/v1
// @Summary Get proxmox VM stats
// @Description Get proxmox VM stats in format of "STATUS|CPU%%|MEM USAGE/LIMIT|MEM%%|NET I/O|BLOCK I/O"
// @Tags proxmox,websocket
// @Produce text/plain
// @Param path path StatsRequest true "Request"
// @Success 200 string plain "Stats output"
// @Failure 400 {object} apitypes.ErrorResponse "Invalid request"
@@ -30,7 +90,7 @@ type StatsRequest struct {
// @Failure 404 {object} apitypes.ErrorResponse "Node not found"
// @Failure 500 {object} apitypes.ErrorResponse "Internal server error"
// @Router /proxmox/stats/{node}/{vmid} [get]
func Stats(c *gin.Context) {
func VMStats(c *gin.Context) {
var request StatsRequest
if err := c.ShouldBindUri(&request); err != nil {
c.JSON(http.StatusBadRequest, apitypes.Error("invalid request", err))