mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-16 05:41:13 +08:00
[NET-546] Move ee code to ee package, unify ee status and terminology (#2538)
* Move ee code to ee package and unify ee status to IsPro * Consolidate naming for paid/professional/enterprise version as "pro". Notes: - Changes image tags - Changes build tags - Changes package names - Doesn't change links to docs that mention "ee" - Doesn't change parameters sent to PostHog that mention "ee" * Revert docker image tag being -pro, back to -ee * Revert go build tag being pro, back to ee * Add build tags for some ee content * [2] Revert go build tag being pro, back to ee * Fix test workflow * Add a json tag to be backwards compatible with frontend "IsEE" check * Add a json tag for the serverconfig struct for IsEE * Ammend json tag to Is_EE * fix ee tags --------- Co-authored-by: Abhishek Kondur <abhi281342@gmail.com>
This commit is contained in:
Gabriel de Souza Seibel
committed by
GitHub
GitHub
parent
31fcc8cd6d
commit
1a1ba1ccf4
164
pro/controllers/metrics.go
Normal file
164
pro/controllers/metrics.go
Normal file
@@ -0,0 +1,164 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
proLogic "github.com/gravitl/netmaker/pro/logic"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
"github.com/gravitl/netmaker/database"
|
||||
"github.com/gravitl/netmaker/logger"
|
||||
"github.com/gravitl/netmaker/logic"
|
||||
"github.com/gravitl/netmaker/models"
|
||||
)
|
||||
|
||||
// MetricHandlers - How we handle Pro Metrics
|
||||
func MetricHandlers(r *mux.Router) {
|
||||
r.HandleFunc("/api/metrics/{network}/{nodeid}", logic.SecurityCheck(true, http.HandlerFunc(getNodeMetrics))).Methods(http.MethodGet)
|
||||
r.HandleFunc("/api/metrics/{network}", logic.SecurityCheck(true, http.HandlerFunc(getNetworkNodesMetrics))).Methods(http.MethodGet)
|
||||
r.HandleFunc("/api/metrics", logic.SecurityCheck(true, http.HandlerFunc(getAllMetrics))).Methods(http.MethodGet)
|
||||
r.HandleFunc("/api/metrics-ext/{network}", logic.SecurityCheck(true, http.HandlerFunc(getNetworkExtMetrics))).Methods(http.MethodGet)
|
||||
}
|
||||
|
||||
// get the metrics of a given node
|
||||
func getNodeMetrics(w http.ResponseWriter, r *http.Request) {
|
||||
// set header.
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
|
||||
var params = mux.Vars(r)
|
||||
nodeID := params["nodeid"]
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "requested fetching metrics for node", nodeID, "on network", params["network"])
|
||||
metrics, err := logic.GetMetrics(nodeID)
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to fetch metrics of node", nodeID, err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "fetched metrics for node", params["nodeid"])
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(metrics)
|
||||
}
|
||||
|
||||
// get the metrics of all nodes in given network
|
||||
func getNetworkNodesMetrics(w http.ResponseWriter, r *http.Request) {
|
||||
// set header.
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
|
||||
var params = mux.Vars(r)
|
||||
network := params["network"]
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "requested fetching network node metrics on network", network)
|
||||
networkNodes, err := logic.GetNetworkNodes(network)
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to fetch metrics of all nodes in network", network, err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
networkMetrics := models.NetworkMetrics{}
|
||||
networkMetrics.Nodes = make(models.MetricsMap)
|
||||
|
||||
for i := range networkNodes {
|
||||
id := networkNodes[i].ID
|
||||
metrics, err := proLogic.GetMetrics(id.String())
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to append metrics of node", id.String(), "during network metrics fetch", err.Error())
|
||||
continue
|
||||
}
|
||||
networkMetrics.Nodes[id.String()] = *metrics
|
||||
}
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "fetched metrics for network", network)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(networkMetrics)
|
||||
}
|
||||
|
||||
// get the metrics for ext clients on a given network
|
||||
func getNetworkExtMetrics(w http.ResponseWriter, r *http.Request) {
|
||||
// set header.
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
|
||||
var params = mux.Vars(r)
|
||||
network := params["network"]
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "requested fetching external client metrics on network", network)
|
||||
ingresses, err := proLogic.GetNetworkIngresses(network) // grab all the ingress gateways
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to fetch metrics of ext clients in network", network, err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
clients, err := logic.GetNetworkExtClients(network) // grab all the network ext clients
|
||||
if err != nil {
|
||||
if database.IsEmptyRecord(err) {
|
||||
var metrics struct{}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(metrics)
|
||||
return
|
||||
}
|
||||
logger.Log(1, r.Header.Get("user"), "failed to fetch metrics of ext clients in network", network, err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
networkMetrics := models.Metrics{}
|
||||
networkMetrics.Connectivity = make(map[string]models.Metric)
|
||||
|
||||
for i := range ingresses {
|
||||
id := ingresses[i].ID
|
||||
ingressMetrics, err := proLogic.GetMetrics(id.String())
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to append external client metrics from ingress node", id.String(), err.Error())
|
||||
continue
|
||||
}
|
||||
if ingressMetrics.Connectivity == nil {
|
||||
continue
|
||||
}
|
||||
for j := range clients {
|
||||
if clients[j].Network != network {
|
||||
continue
|
||||
}
|
||||
// if metrics for that client have been reported, append them
|
||||
if len(ingressMetrics.Connectivity[clients[j].ClientID].NodeName) > 0 {
|
||||
networkMetrics.Connectivity[clients[j].ClientID] = ingressMetrics.Connectivity[clients[j].ClientID]
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "fetched ext client metrics for network", network)
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(networkMetrics.Connectivity)
|
||||
}
|
||||
|
||||
// get Metrics of all nodes on server, lots of data
|
||||
func getAllMetrics(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
logger.Log(1, r.Header.Get("user"), "requested fetching all metrics")
|
||||
|
||||
allNodes, err := logic.GetAllNodes()
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to fetch metrics of all nodes on server", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
networkMetrics := models.NetworkMetrics{}
|
||||
networkMetrics.Nodes = make(models.MetricsMap)
|
||||
|
||||
for i := range allNodes {
|
||||
id := allNodes[i].ID
|
||||
metrics, err := proLogic.GetMetrics(id.String())
|
||||
if err != nil {
|
||||
logger.Log(1, r.Header.Get("user"), "failed to append metrics of node", id.String(), "during all nodes metrics fetch", err.Error())
|
||||
continue
|
||||
}
|
||||
networkMetrics.Nodes[id.String()] = *metrics
|
||||
}
|
||||
|
||||
logger.Log(1, r.Header.Get("user"), "fetched metrics for all nodes on server")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
json.NewEncoder(w).Encode(networkMetrics)
|
||||
}
|
||||
Reference in New Issue
Block a user