mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
324 lines
11 KiB
Go
324 lines
11 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package node
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/openflare/apiutil"
|
|
"github.com/Rain-kl/Wavelet/internal/common/response"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func handleLogicError(c *gin.Context, err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
return apiutil.AbortNotFoundIfMissing(c, err, errNodeNotFound)
|
|
}
|
|
|
|
// ListNodesHandler lists all nodes.
|
|
// @Summary 获取节点列表
|
|
// @Description 返回所有节点及最新配置下发记录，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Success 200 {object} response.Any{data=[]node.View} "节点列表"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或不存在"
|
|
// @Router /api/v1/d/nodes [get]
|
|
func ListNodesHandler(c *gin.Context) {
|
|
nodes, err := ListNodes(c.Request.Context())
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(nodes))
|
|
}
|
|
|
|
// CreateNodeHandler creates a node.
|
|
// @Summary 创建节点
|
|
// @Description 创建新的边缘节点记录，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param body body node.Input true "节点参数"
|
|
// @Success 200 {object} response.Any{data=node.View} "创建成功"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或不存在"
|
|
// @Router /api/v1/d/nodes [post]
|
|
func CreateNodeHandler(c *gin.Context) {
|
|
var input Input
|
|
if !apiutil.BindJSON(c, &input) {
|
|
return
|
|
}
|
|
view, err := CreateNode(c.Request.Context(), input)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// UpdateNodeHandler updates a node.
|
|
// @Summary 更新节点
|
|
// @Description 更新指定节点的配置信息，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Param body body node.Input true "节点参数"
|
|
// @Success 200 {object} response.Any{data=node.View} "更新成功"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/update [post]
|
|
func UpdateNodeHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var input Input
|
|
if !apiutil.BindJSON(c, &input) {
|
|
return
|
|
}
|
|
view, err := UpdateNode(c.Request.Context(), id, input)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// DeleteNodeHandler deletes a node.
|
|
// @Summary 删除节点
|
|
// @Description 删除指定节点记录，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Success 200 {object} response.Any{data=string} "删除成功"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/delete [post]
|
|
func DeleteNodeHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
if err := DeleteNode(c.Request.Context(), id); handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OKNil())
|
|
}
|
|
|
|
// GetBootstrapTokenHandler returns the global discovery token.
|
|
// @Summary 获取引导令牌
|
|
// @Description 返回全局节点发现引导令牌，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Success 200 {object} response.Any{data=node.BootstrapView} "引导令牌"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或不存在"
|
|
// @Router /api/v1/d/nodes/bootstrap-token [get]
|
|
func GetBootstrapTokenHandler(c *gin.Context) {
|
|
view, err := GetBootstrapToken(c.Request.Context())
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// RotateBootstrapTokenHandler rotates the global discovery token.
|
|
// @Summary 轮换引导令牌
|
|
// @Description 重新生成全局节点发现引导令牌，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Success 200 {object} response.Any{data=node.BootstrapView} "新引导令牌"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或不存在"
|
|
// @Router /api/v1/d/nodes/bootstrap-token/rotate [post]
|
|
func RotateBootstrapTokenHandler(c *gin.Context) {
|
|
view, err := RotateBootstrapToken(c.Request.Context())
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// GetAgentReleaseHandler returns the latest agent release for a node.
|
|
// @Summary 获取 Agent 发布信息
|
|
// @Description 返回指定节点可用的最新 Agent 版本信息，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Param channel query string false "发布渠道"
|
|
// @Success 200 {object} response.Any{data=node.AgentReleaseInfo} "Agent 发布信息"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/agent-release [get]
|
|
func GetAgentReleaseHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
release, err := GetAgentRelease(c.Request.Context(), id, c.Query("channel"))
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(release))
|
|
}
|
|
|
|
// RequestAgentUpdateHandler requests agent self-update on a node.
|
|
// @Summary 请求 Agent 更新
|
|
// @Description 向指定节点下发 Agent 自更新指令，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Param body body node.AgentUpdateInput false "更新参数（可选）"
|
|
// @Success 200 {object} response.Any{data=node.View} "更新请求已下发"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/agent-update [post]
|
|
func RequestAgentUpdateHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var request AgentUpdateInput
|
|
if c.Request.ContentLength > 0 {
|
|
if err := bindOptionalJSON(c.Request.Body, &request); err != nil {
|
|
response.AbortBadRequest(c, "参数错误")
|
|
return
|
|
}
|
|
}
|
|
view, err := RequestAgentUpdate(c.Request.Context(), id, request)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// RequestOpenrestyRestartHandler requests openresty restart on a node.
|
|
// @Summary 请求重启 OpenResty
|
|
// @Description 向指定节点下发 OpenResty 重启指令，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Success 200 {object} response.Any{data=node.View} "重启请求已下发"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/openresty-restart [post]
|
|
func RequestOpenrestyRestartHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
view, err := RequestOpenrestyRestart(c.Request.Context(), id)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// RequestForceSyncHandler requests force sync on a node.
|
|
// @Summary 请求强制同步配置
|
|
// @Description 向指定节点下发强制同步当前活跃配置的指令，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Success 200 {object} response.Any{data=node.View} "同步请求已下发"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/force-sync [post]
|
|
func RequestForceSyncHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
view, err := RequestForceSync(c.Request.Context(), id)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// GetObservabilityHandler returns node observability details.
|
|
// @Summary 获取节点可观测性数据
|
|
// @Description 返回指定节点的指标、健康事件与流量分析数据，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Param hours query int false "统计时间范围（小时）"
|
|
// @Param limit query int false "返回记录数量上限"
|
|
// @Success 200 {object} response.Any{data=node.ObservabilityView} "可观测性数据"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/observability [get]
|
|
func GetObservabilityHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var query ObservabilityQuery
|
|
if err := c.ShouldBindQuery(&query); err != nil {
|
|
response.AbortBadRequest(c, "参数错误")
|
|
return
|
|
}
|
|
view, err := GetObservability(c.Request.Context(), id, query)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(view))
|
|
}
|
|
|
|
// CleanupHealthEventsHandler cleans up node health events.
|
|
// @Summary 清理节点健康事件
|
|
// @Description 清理指定节点的历史健康事件记录，需要管理员权限
|
|
// @Tags openflare-node
|
|
// @Produce json
|
|
// @Security SessionCookie
|
|
// @Param id path int true "节点 ID"
|
|
// @Success 200 {object} response.Any{data=node.HealthEventCleanupResult} "清理结果"
|
|
// @Failure 400 {object} response.Any "参数错误"
|
|
// @Failure 401 {object} response.Any "未登录"
|
|
// @Failure 404 {object} response.Any "无权限或节点不存在"
|
|
// @Router /api/v1/d/nodes/{id}/observability/cleanup [post]
|
|
func CleanupHealthEventsHandler(c *gin.Context) {
|
|
id, ok := apiutil.IDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
result, err := CleanupHealthEvents(c.Request.Context(), id)
|
|
if handleLogicError(c, err) {
|
|
return
|
|
}
|
|
c.JSON(http.StatusOK, response.OK(result))
|
|
}
|
|
|
|
func bindOptionalJSON(body io.Reader, target any) error {
|
|
if err := json.NewDecoder(body).Decode(target); err != nil && !errors.Is(err, io.EOF) {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|