ops_realtime_traffic.go 1.4 KB

123456789101112131415161718192021222324252627282930313233343536
  1. package service
  2. import (
  3. "context"
  4. "time"
  5. infraerrors "github.com/Wei-Shaw/sub2api/internal/pkg/errors"
  6. )
  7. // GetRealtimeTrafficSummary returns QPS/TPS current/peak/avg for the provided window.
  8. // This is used by the Ops dashboard "Realtime Traffic" card and is intentionally lightweight.
  9. func (s *OpsService) GetRealtimeTrafficSummary(ctx context.Context, filter *OpsDashboardFilter) (*OpsRealtimeTrafficSummary, error) {
  10. if err := s.RequireMonitoringEnabled(ctx); err != nil {
  11. return nil, err
  12. }
  13. if s.opsRepo == nil {
  14. return nil, infraerrors.ServiceUnavailable("OPS_REPO_UNAVAILABLE", "Ops repository not available")
  15. }
  16. if filter == nil {
  17. return nil, infraerrors.BadRequest("OPS_FILTER_REQUIRED", "filter is required")
  18. }
  19. if filter.StartTime.IsZero() || filter.EndTime.IsZero() {
  20. return nil, infraerrors.BadRequest("OPS_TIME_RANGE_REQUIRED", "start_time/end_time are required")
  21. }
  22. if filter.StartTime.After(filter.EndTime) {
  23. return nil, infraerrors.BadRequest("OPS_TIME_RANGE_INVALID", "start_time must be <= end_time")
  24. }
  25. if filter.EndTime.Sub(filter.StartTime) > time.Hour {
  26. return nil, infraerrors.BadRequest("OPS_TIME_RANGE_TOO_LARGE", "invalid time range: max window is 1 hour")
  27. }
  28. // Realtime traffic summary always uses raw logs (minute granularity peaks).
  29. filter.QueryMode = OpsQueryModeRaw
  30. return s.opsRepo.GetRealtimeTrafficSummary(ctx, filter)
  31. }