-
Notifications
You must be signed in to change notification settings - Fork 18
/
metrics.go
90 lines (76 loc) · 2.88 KB
/
metrics.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// Copyright 2023 The ClusterLink Authors.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
/**********************************************************/
/* Package Metrics provides an exporter of gateway's connection-level metrics
/**********************************************************/
package metrics
import (
"encoding/json"
"net/http"
"github.com/go-chi/chi"
"github.com/sirupsen/logrus"
event "github.com/clusterlink-net/clusterlink/pkg/controlplane/eventmanager"
)
var (
mlog = logrus.WithField("component", "Metrics")
MyMetricsManager Metrics
)
type Metrics struct {
ConnectionFlow map[string]*event.ConnectionStatusAttr
}
func (m *Metrics) Routes(r *chi.Mux) chi.Router {
r.Route("/"+event.ConnectionStatus, func(r chi.Router) {
r.Get("/", m.GetConnectionMetrics) // Get Metrics from the metrics manager
r.Post("/", m.PostConnectionMetrics) // Post Metrics to the metrics manager
})
// TODO : Add more endpoints to support query
return r
}
func (m *Metrics) init(router *chi.Mux) {
m.ConnectionFlow = make(map[string]*event.ConnectionStatusAttr)
routes := m.Routes(router)
router.Mount("/metrics", routes)
}
func (m *Metrics) GetConnectionMetrics(w http.ResponseWriter, _ *http.Request) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusOK)
if err := json.NewEncoder(w).Encode(m.ConnectionFlow); err != nil {
mlog.Errorf("Error happened in JSON encode. Err: %s", err)
}
}
func (m *Metrics) PostConnectionMetrics(w http.ResponseWriter, r *http.Request) {
var connectionStatus event.ConnectionStatusAttr
err := json.NewDecoder(r.Body).Decode(&connectionStatus)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
// Aggregate Metrics
m.aggregateMetrics(&connectionStatus)
}
func (m *Metrics) aggregateMetrics(connectionStatus *event.ConnectionStatusAttr) {
if _, exists := m.ConnectionFlow[connectionStatus.ConnectionID]; exists {
// Update existing metrics
flow := m.ConnectionFlow[connectionStatus.ConnectionID]
flow.IncomingBytes += connectionStatus.IncomingBytes
flow.OutgoingBytes += connectionStatus.OutgoingBytes
flow.LastTstamp = connectionStatus.LastTstamp
flow.State = connectionStatus.State
} else {
m.ConnectionFlow[connectionStatus.ConnectionID] = connectionStatus
}
}
func StartMetricsManager(router *chi.Mux) {
mlog.Infof("Metrics Manager started")
MyMetricsManager.init(router)
}