| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195 |
- package handler
- import (
- "time"
- "github.com/Wei-Shaw/sub2api/internal/domain"
- "github.com/Wei-Shaw/sub2api/internal/handler/admin"
- "github.com/Wei-Shaw/sub2api/internal/handler/dto"
- "github.com/Wei-Shaw/sub2api/internal/pkg/response"
- "github.com/Wei-Shaw/sub2api/internal/service"
- "github.com/gin-gonic/gin"
- )
- // ChannelMonitorUserHandler 渠道监控用户只读 handler。
- type ChannelMonitorUserHandler struct {
- monitorService *service.ChannelMonitorService
- settingService *service.SettingService
- }
- // NewChannelMonitorUserHandler 创建 handler。
- // settingService 用于每次请求前读取功能开关;关闭时 List/GetStatus 直接返回空/404。
- func NewChannelMonitorUserHandler(
- monitorService *service.ChannelMonitorService,
- settingService *service.SettingService,
- ) *ChannelMonitorUserHandler {
- return &ChannelMonitorUserHandler{
- monitorService: monitorService,
- settingService: settingService,
- }
- }
- // featureEnabled 返回当前渠道监控功能是否开启。
- // settingService 为 nil(测试场景)视为启用。
- func (h *ChannelMonitorUserHandler) featureEnabled(c *gin.Context) bool {
- if h.settingService == nil {
- return true
- }
- runtime := h.settingService.GetChannelMonitorRuntime(c.Request.Context())
- return runtime.Enabled && runtime.Mode == service.ChannelMonitorModeV1
- }
- // quotaVisible 返回用户端是否展示配额/余额快照(channel_monitor_show_quota,
- // fail-closed:未配置/非 "true" 一律视为关闭)。settingService 为 nil 时 fail-closed。
- func (h *ChannelMonitorUserHandler) quotaVisible(c *gin.Context) bool {
- if h.settingService == nil {
- return false
- }
- return h.settingService.GetChannelMonitorRuntime(c.Request.Context()).ShowQuota
- }
- // --- Response ---
- type channelMonitorUserListItem struct {
- ID int64 `json:"id"`
- Name string `json:"name"`
- Provider string `json:"provider"`
- GroupName string `json:"group_name"`
- PrimaryModel string `json:"primary_model"`
- PrimaryStatus string `json:"primary_status"`
- PrimaryLatencyMs *int `json:"primary_latency_ms"`
- PrimaryPingLatencyMs *int `json:"primary_ping_latency_ms"`
- Availability7d float64 `json:"availability_7d"`
- ExtraModels []dto.ChannelMonitorExtraModelStatus `json:"extra_models"`
- Timeline []channelMonitorUserTimelinePoint `json:"timeline"`
- // LatestQuota 主模型最近配额快照;channel_monitor_show_quota=false 时
- // 由 userMonitorViewToItem 的调用方传入 false 剥离(服务端脱敏,非仅前端隐藏)。
- LatestQuota *domain.MonitorQuotaSnapshot `json:"latest_quota,omitempty"`
- }
- // channelMonitorUserTimelinePoint 主模型最近一次检测的 timeline 点。
- // 仅用于用户视图 list 响应,admin 视图不使用。
- type channelMonitorUserTimelinePoint struct {
- Status string `json:"status"`
- LatencyMs *int `json:"latency_ms"`
- PingLatencyMs *int `json:"ping_latency_ms"`
- CheckedAt string `json:"checked_at"`
- }
- type channelMonitorUserDetailResponse struct {
- ID int64 `json:"id"`
- Name string `json:"name"`
- Provider string `json:"provider"`
- GroupName string `json:"group_name"`
- Models []channelMonitorUserModelStat `json:"models"`
- }
- type channelMonitorUserModelStat struct {
- Model string `json:"model"`
- LatestStatus string `json:"latest_status"`
- LatestLatencyMs *int `json:"latest_latency_ms"`
- Availability7d float64 `json:"availability_7d"`
- Availability15d float64 `json:"availability_15d"`
- Availability30d float64 `json:"availability_30d"`
- AvgLatency7dMs *int `json:"avg_latency_7d_ms"`
- }
- func userMonitorViewToItem(v *service.UserMonitorView, includeQuota bool) channelMonitorUserListItem {
- extras := make([]dto.ChannelMonitorExtraModelStatus, 0, len(v.ExtraModels))
- for _, e := range v.ExtraModels {
- extras = append(extras, dto.ChannelMonitorExtraModelStatus{
- Model: e.Model,
- Status: e.Status,
- LatencyMs: e.LatencyMs,
- })
- }
- timeline := make([]channelMonitorUserTimelinePoint, 0, len(v.Timeline))
- for _, p := range v.Timeline {
- timeline = append(timeline, channelMonitorUserTimelinePoint{
- Status: p.Status,
- LatencyMs: p.LatencyMs,
- PingLatencyMs: p.PingLatencyMs,
- CheckedAt: p.CheckedAt.UTC().Format(time.RFC3339),
- })
- }
- item := channelMonitorUserListItem{
- ID: v.ID,
- Name: v.Name,
- Provider: v.Provider,
- GroupName: v.GroupName,
- PrimaryModel: v.PrimaryModel,
- PrimaryStatus: v.PrimaryStatus,
- PrimaryLatencyMs: v.PrimaryLatencyMs,
- PrimaryPingLatencyMs: v.PrimaryPingLatencyMs,
- Availability7d: v.Availability7d,
- ExtraModels: extras,
- Timeline: timeline,
- }
- if includeQuota {
- item.LatestQuota = v.LatestQuota
- }
- return item
- }
- func userMonitorDetailToResponse(d *service.UserMonitorDetail) *channelMonitorUserDetailResponse {
- models := make([]channelMonitorUserModelStat, 0, len(d.Models))
- for _, m := range d.Models {
- models = append(models, channelMonitorUserModelStat{
- Model: m.Model,
- LatestStatus: m.LatestStatus,
- LatestLatencyMs: m.LatestLatencyMs,
- Availability7d: m.Availability7d,
- Availability15d: m.Availability15d,
- Availability30d: m.Availability30d,
- AvgLatency7dMs: m.AvgLatency7dMs,
- })
- }
- return &channelMonitorUserDetailResponse{
- ID: d.ID,
- Name: d.Name,
- Provider: d.Provider,
- GroupName: d.GroupName,
- Models: models,
- }
- }
- // --- Handlers ---
- // List GET /api/v1/channel-monitors
- func (h *ChannelMonitorUserHandler) List(c *gin.Context) {
- if !h.featureEnabled(c) {
- response.Success(c, gin.H{"items": []channelMonitorUserListItem{}})
- return
- }
- views, err := h.monitorService.ListUserView(c.Request.Context())
- if err != nil {
- response.ErrorFrom(c, err)
- return
- }
- includeQuota := h.quotaVisible(c)
- items := make([]channelMonitorUserListItem, 0, len(views))
- for _, v := range views {
- items = append(items, userMonitorViewToItem(v, includeQuota))
- }
- response.Success(c, gin.H{"items": items})
- }
- // GetStatus GET /api/v1/channel-monitors/:id/status
- func (h *ChannelMonitorUserHandler) GetStatus(c *gin.Context) {
- if !h.featureEnabled(c) {
- response.ErrorFrom(c, service.ErrChannelMonitorNotFound)
- return
- }
- // 复用 admin.ParseChannelMonitorID 保持错误码与日志一致。
- id, ok := admin.ParseChannelMonitorID(c)
- if !ok {
- return
- }
- detail, err := h.monitorService.GetUserDetail(c.Request.Context(), id)
- if err != nil {
- response.ErrorFrom(c, err)
- return
- }
- response.Success(c, userMonitorDetailToResponse(detail))
- }
|