refactor(api/health): simplify health info type

- Updated health-related functions to return simplified health information.
- Introduced HealthStatusString type for correct swagger and schema generation.
- Refactored HealthJSON structure to utilize the new HealthStatusString type.
This commit is contained in:
yusing
2026-01-02 18:02:49 +08:00
parent 65383c7061
commit ad2bfac275
5 changed files with 112 additions and 139 deletions

View File

@@ -2956,43 +2956,6 @@
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"HealthInfo": {
"type": "object",
"properties": {
"detail": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"latency": {
"description": "latency in microseconds",
"type": "number",
"x-nullable": false,
"x-omitempty": false
},
"status": {
"type": "string",
"enum": [
"healthy",
"unhealthy",
"napping",
"starting",
"error",
"unknown"
],
"x-nullable": false,
"x-omitempty": false
},
"uptime": {
"description": "uptime in milliseconds",
"type": "number",
"x-nullable": false,
"x-omitempty": false
}
},
"x-nullable": false,
"x-omitempty": false
},
"HealthInfoWithoutDetail": { "HealthInfoWithoutDetail": {
"type": "object", "type": "object",
"properties": { "properties": {
@@ -3047,22 +3010,14 @@
"x-nullable": true "x-nullable": true
}, },
"lastSeen": { "lastSeen": {
"description": "unix timestamp in seconds",
"type": "integer", "type": "integer",
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"lastSeenStr": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"latency": { "latency": {
"type": "number", "description": "latency in milliseconds",
"x-nullable": false, "type": "integer",
"x-omitempty": false
},
"latencyStr": {
"type": "string",
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
@@ -3072,30 +3027,22 @@
"x-omitempty": false "x-omitempty": false
}, },
"started": { "started": {
"description": "unix timestamp in seconds",
"type": "integer", "type": "integer",
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"startedStr": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"status": { "status": {
"type": "string", "$ref": "#/definitions/HealthStatusString",
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"uptime": { "uptime": {
"description": "uptime in seconds",
"type": "number", "type": "number",
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"uptimeStr": {
"type": "string",
"x-nullable": false,
"x-omitempty": false
},
"url": { "url": {
"type": "string", "type": "string",
"x-nullable": false, "x-nullable": false,
@@ -3108,11 +3055,32 @@
"HealthMap": { "HealthMap": {
"type": "object", "type": "object",
"additionalProperties": { "additionalProperties": {
"$ref": "#/definitions/HealthInfo" "$ref": "#/definitions/HealthStatusString"
}, },
"x-nullable": false, "x-nullable": false,
"x-omitempty": false "x-omitempty": false
}, },
"HealthStatusString": {
"type": "string",
"enum": [
"unknown",
"healthy",
"napping",
"starting",
"unhealthy",
"error"
],
"x-enum-varnames": [
"StatusUnknownStr",
"StatusHealthyStr",
"StatusNappingStr",
"StatusStartingStr",
"StatusUnhealthyStr",
"StatusErrorStr"
],
"x-nullable": false,
"x-omitempty": false
},
"HomepageCategory": { "HomepageCategory": {
"type": "object", "type": "object",
"properties": { "properties": {

View File

@@ -302,26 +302,6 @@ definitions:
additionalProperties: {} additionalProperties: {}
type: object type: object
type: object type: object
HealthInfo:
properties:
detail:
type: string
latency:
description: latency in microseconds
type: number
status:
enum:
- healthy
- unhealthy
- napping
- starting
- error
- unknown
type: string
uptime:
description: uptime in milliseconds
type: number
type: object
HealthInfoWithoutDetail: HealthInfoWithoutDetail:
properties: properties:
latency: latency:
@@ -351,32 +331,44 @@ definitions:
- $ref: '#/definitions/HealthExtra' - $ref: '#/definitions/HealthExtra'
x-nullable: true x-nullable: true
lastSeen: lastSeen:
description: unix timestamp in seconds
type: integer type: integer
lastSeenStr:
type: string
latency: latency:
type: number description: latency in milliseconds
latencyStr: type: integer
type: string
name: name:
type: string type: string
started: started:
description: unix timestamp in seconds
type: integer type: integer
startedStr:
type: string
status: status:
type: string $ref: '#/definitions/HealthStatusString'
uptime: uptime:
description: uptime in seconds
type: number type: number
uptimeStr:
type: string
url: url:
type: string type: string
type: object type: object
HealthMap: HealthMap:
additionalProperties: additionalProperties:
$ref: '#/definitions/HealthInfo' $ref: '#/definitions/HealthStatusString'
type: object type: object
HealthStatusString:
enum:
- unknown
- healthy
- napping
- starting
- unhealthy
- error
type: string
x-enum-varnames:
- StatusUnknownStr
- StatusHealthyStr
- StatusNappingStr
- StatusStartingStr
- StatusUnhealthyStr
- StatusErrorStr
HomepageCategory: HomepageCategory:
properties: properties:
items: items:

View File

@@ -12,8 +12,6 @@ import (
_ "github.com/yusing/goutils/apitypes" _ "github.com/yusing/goutils/apitypes"
) )
type HealthMap = map[string]routes.HealthInfo // @name HealthMap
// @x-id "health" // @x-id "health"
// @BasePath /api/v1 // @BasePath /api/v1
// @Summary Get routes health info // @Summary Get routes health info
@@ -21,16 +19,16 @@ type HealthMap = map[string]routes.HealthInfo // @name HealthMap
// @Tags v1,websocket // @Tags v1,websocket
// @Accept json // @Accept json
// @Produce json // @Produce json
// @Success 200 {object} HealthMap "Health info by route name" // @Success 200 {object} routes.HealthMap "Health info by route name"
// @Failure 403 {object} apitypes.ErrorResponse // @Failure 403 {object} apitypes.ErrorResponse
// @Failure 500 {object} apitypes.ErrorResponse // @Failure 500 {object} apitypes.ErrorResponse
// @Router /health [get] // @Router /health [get]
func Health(c *gin.Context) { func Health(c *gin.Context) {
if httpheaders.IsWebsocket(c.Request.Header) { if httpheaders.IsWebsocket(c.Request.Header) {
websocket.PeriodicWrite(c, 1*time.Second, func() (any, error) { websocket.PeriodicWrite(c, 1*time.Second, func() (any, error) {
return routes.GetHealthInfo(), nil return routes.GetHealthInfoSimple(), nil
}) })
} else { } else {
c.JSON(http.StatusOK, routes.GetHealthInfo()) c.JSON(http.StatusOK, routes.GetHealthInfoSimple())
} }
} }

View File

@@ -17,6 +17,8 @@ type HealthInfoWithoutDetail struct {
Latency time.Duration `json:"latency" swaggertype:"number"` // latency in microseconds Latency time.Duration `json:"latency" swaggertype:"number"` // latency in microseconds
} // @name HealthInfoWithoutDetail } // @name HealthInfoWithoutDetail
type HealthMap = map[string]types.HealthStatusString // @name HealthMap
// GetHealthInfo returns a map of route name to health info. // GetHealthInfo returns a map of route name to health info.
// //
// The health info is for all routes, including excluded routes. // The health info is for all routes, including excluded routes.
@@ -39,6 +41,14 @@ func GetHealthInfoWithoutDetail() map[string]HealthInfoWithoutDetail {
return healthMap return healthMap
} }
func GetHealthInfoSimple() map[string]types.HealthStatus {
healthMap := make(map[string]types.HealthStatus, NumAllRoutes())
for r := range IterAll {
healthMap[r.Name()] = getHealthInfoSimple(r)
}
return healthMap
}
func getHealthInfo(r types.Route) HealthInfo { func getHealthInfo(r types.Route) HealthInfo {
mon := r.HealthMonitor() mon := r.HealthMonitor()
if mon == nil { if mon == nil {
@@ -73,6 +83,14 @@ func getHealthInfoWithoutDetail(r types.Route) HealthInfoWithoutDetail {
} }
} }
func getHealthInfoSimple(r types.Route) types.HealthStatus {
mon := r.HealthMonitor()
if mon == nil {
return types.StatusUnknown
}
return mon.Status()
}
// ByProvider returns a map of provider name to routes. // ByProvider returns a map of provider name to routes.
// //
// The routes are all routes, including excluded routes. // The routes are all routes, including excluded routes.

View File

@@ -8,12 +8,12 @@ import (
"time" "time"
"github.com/bytedance/sonic" "github.com/bytedance/sonic"
strutils "github.com/yusing/goutils/strings"
"github.com/yusing/goutils/task" "github.com/yusing/goutils/task"
) )
type ( type (
HealthStatus uint8 HealthStatus uint8 // @name HealthStatus
HealthStatusString string // @name HealthStatusString
HealthCheckResult struct { HealthCheckResult struct {
Healthy bool `json:"healthy"` Healthy bool `json:"healthy"`
@@ -45,20 +45,16 @@ type (
HealthChecker HealthChecker
} }
HealthJSON struct { HealthJSON struct {
Name string `json:"name"` Name string `json:"name"`
Config *HealthCheckConfig `json:"config"` Config *HealthCheckConfig `json:"config"`
Started int64 `json:"started"` Started int64 `json:"started"` // unix timestamp in seconds
StartedStr string `json:"startedStr"` Status HealthStatusString `json:"status"`
Status string `json:"status"` Uptime float64 `json:"uptime"` // uptime in seconds
Uptime float64 `json:"uptime"` Latency int64 `json:"latency"` // latency in milliseconds
UptimeStr string `json:"uptimeStr"` LastSeen int64 `json:"lastSeen"` // unix timestamp in seconds
Latency float64 `json:"latency"` Detail string `json:"detail"`
LatencyStr string `json:"latencyStr"` URL string `json:"url"`
LastSeen int64 `json:"lastSeen"` Extra *HealthExtra `json:"extra,omitempty" extensions:"x-nullable"`
LastSeenStr string `json:"lastSeenStr"`
Detail string `json:"detail"`
URL string `json:"url"`
Extra *HealthExtra `json:"extra,omitempty" extensions:"x-nullable"`
} // @name HealthJSON } // @name HealthJSON
HealthJSONRepr struct { HealthJSONRepr struct {
@@ -88,12 +84,12 @@ const (
StatusUnhealthy StatusUnhealthy
StatusError StatusError
StatusUnknownStr = "unknown" StatusUnknownStr HealthStatusString = "unknown"
StatusHealthyStr = "healthy" StatusHealthyStr HealthStatusString = "healthy"
StatusNappingStr = "napping" StatusNappingStr HealthStatusString = "napping"
StatusStartingStr = "starting" StatusStartingStr HealthStatusString = "starting"
StatusUnhealthyStr = "unhealthy" StatusUnhealthyStr HealthStatusString = "unhealthy"
StatusErrorStr = "error" StatusErrorStr HealthStatusString = "error"
NumStatuses int = iota - 1 NumStatuses int = iota - 1
@@ -102,15 +98,15 @@ const (
) )
var ( var (
StatusHealthyStr2 = strconv.Itoa(int(StatusHealthy)) StatusHealthyStr2 HealthStatusString = HealthStatusString(strconv.Itoa(int(StatusHealthy)))
StatusNappingStr2 = strconv.Itoa(int(StatusNapping)) StatusNappingStr2 HealthStatusString = HealthStatusString(strconv.Itoa(int(StatusNapping)))
StatusStartingStr2 = strconv.Itoa(int(StatusStarting)) StatusStartingStr2 HealthStatusString = HealthStatusString(strconv.Itoa(int(StatusStarting)))
StatusUnhealthyStr2 = strconv.Itoa(int(StatusUnhealthy)) StatusUnhealthyStr2 HealthStatusString = HealthStatusString(strconv.Itoa(int(StatusUnhealthy)))
StatusErrorStr2 = strconv.Itoa(int(StatusError)) StatusErrorStr2 HealthStatusString = HealthStatusString(strconv.Itoa(int(StatusError)))
) )
func NewHealthStatusFromString(s string) HealthStatus { func NewHealthStatusFromString(s string) HealthStatus {
switch s { switch HealthStatusString(s) {
case StatusHealthyStr, StatusHealthyStr2: case StatusHealthyStr, StatusHealthyStr2:
return StatusHealthy return StatusHealthy
case StatusUnhealthyStr, StatusUnhealthyStr2: case StatusUnhealthyStr, StatusUnhealthyStr2:
@@ -126,7 +122,7 @@ func NewHealthStatusFromString(s string) HealthStatus {
} }
} }
func (s HealthStatus) String() string { func (s HealthStatus) StatusString() HealthStatusString {
switch s { switch s {
case StatusHealthy: case StatusHealthy:
return StatusHealthyStr return StatusHealthyStr
@@ -143,6 +139,11 @@ func (s HealthStatus) String() string {
} }
} }
// String implements fmt.Stringer.
func (s HealthStatus) String() string {
return string(s.StatusString())
}
func (s HealthStatus) Good() bool { func (s HealthStatus) Good() bool {
return s&HealthyMask != 0 return s&HealthyMask != 0
} }
@@ -178,19 +179,15 @@ func (jsonRepr *HealthJSONRepr) MarshalJSON() ([]byte, error) {
url = "" url = ""
} }
return sonic.Marshal(HealthJSON{ return sonic.Marshal(HealthJSON{
Name: jsonRepr.Name, Name: jsonRepr.Name,
Config: jsonRepr.Config, Config: jsonRepr.Config,
Started: jsonRepr.Started.Unix(), Started: jsonRepr.Started.Unix(),
StartedStr: strutils.FormatTime(jsonRepr.Started), Status: HealthStatusString(jsonRepr.Status.String()),
Status: jsonRepr.Status.String(), Uptime: jsonRepr.Uptime.Seconds(),
Uptime: jsonRepr.Uptime.Seconds(), Latency: jsonRepr.Latency.Milliseconds(),
UptimeStr: strutils.FormatDuration(jsonRepr.Uptime), LastSeen: jsonRepr.LastSeen.Unix(),
Latency: jsonRepr.Latency.Seconds(), Detail: jsonRepr.Detail,
LatencyStr: strconv.Itoa(int(jsonRepr.Latency.Milliseconds())) + " ms", URL: url,
LastSeen: jsonRepr.LastSeen.Unix(), Extra: jsonRepr.Extra,
LastSeenStr: strutils.FormatLastSeen(jsonRepr.LastSeen),
Detail: jsonRepr.Detail,
URL: url,
Extra: jsonRepr.Extra,
}) })
} }