mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
f0e234df1f
Agent 仅上报 host_metrics/edge_health/access_logs；业务流量与 UV 由 Server 侧访问日志聚合。新增 of_node_edge_health 与 of_access_log_hourly， 删除 request_reports/openresty 吞吐路径；API 不再暴露 traffic_reports 与 openresty_rx|tx。心跳/离线默认阈值与回填迁移一并入库。
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
package observability
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/agent/config"
|
|
)
|
|
|
|
func TestCollectEdgeHealth(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/openflare/observability" {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
_, _ = w.Write([]byte(`{"ok":true,"captured_at_unix":1710403200,"connections":{"active":12,"reading":1,"writing":2,"waiting":9}}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
health := CollectEdgeHealth(context.Background(), &config.Config{
|
|
OpenrestyObservabilityPort: mustPort(server.URL),
|
|
})
|
|
if health == nil {
|
|
t.Fatal("expected edge health")
|
|
}
|
|
if !health.OK || health.Connections != 12 {
|
|
t.Fatalf("unexpected health: %+v", health)
|
|
}
|
|
}
|
|
|
|
func mustPort(rawURL string) int {
|
|
u := rawURL
|
|
idx := stringsLastColon(u)
|
|
if idx < 0 {
|
|
return 0
|
|
}
|
|
var port int
|
|
for _, ch := range u[idx+1:] {
|
|
if ch < '0' || ch > '9' {
|
|
break
|
|
}
|
|
port = port*10 + int(ch-'0')
|
|
}
|
|
return port
|
|
}
|
|
|
|
func stringsLastColon(s string) int {
|
|
for i := len(s) - 1; i >= 0; i-- {
|
|
if s[i] == ':' {
|
|
return i
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
func TestCollectEdgeHealthHandlesUnavailableEndpoint(t *testing.T) {
|
|
if health := CollectEdgeHealth(context.Background(), &config.Config{
|
|
OpenrestyObservabilityPort: 1,
|
|
}); health != nil {
|
|
t.Fatalf("expected nil health, got %+v", health)
|
|
}
|
|
}
|