initial commit

logic/metrics.go

@@ -0,0 +1,65 @@
+package logic
+
+import (
+	"encoding/json"
+
+	"github.com/gravitl/netmaker/database"
+	"github.com/gravitl/netmaker/models"
+)
+
+// GetMetrics - gets the metrics
+func GetMetrics(nodeid string) (*models.Metrics, error) {
+	var metrics models.Metrics
+	record, err := database.FetchRecord(database.METRICS_TABLE_NAME, nodeid)
+	if err != nil {
+		if database.IsEmptyRecord(err) {
+			return &metrics, nil
+		}
+		return &metrics, err
+	}
+	err = json.Unmarshal([]byte(record), &metrics)
+	if err != nil {
+		return &metrics, err
+	}
+	return &metrics, nil
+}
+
+// UpdateMetrics - updates the metrics of a given client
+func UpdateMetrics(nodeid string, metrics *models.Metrics) error {
+	data, err := json.Marshal(metrics)
+	if err != nil {
+		return err
+	}
+	return database.Insert(nodeid, string(data), database.METRICS_TABLE_NAME)
+}
+
+// DeleteMetrics - deletes metrics of a given node
+func DeleteMetrics(nodeid string) error {
+	return database.DeleteRecord(database.METRICS_TABLE_NAME, nodeid)
+}
+
+// CollectServerMetrics - collects metrics for given server node
+func CollectServerMetrics(serverID string, networkNodes []models.Node) *models.Metrics {
+	newServerMetrics := models.Metrics{}
+	newServerMetrics.Connectivity = make(map[string]models.Metric)
+	for i := range networkNodes {
+		currNodeID := networkNodes[i].ID
+		if currNodeID == serverID {
+			continue
+		}
+		if currMetrics, err := GetMetrics(currNodeID); err == nil {
+			if currMetrics.Connectivity != nil && currMetrics.Connectivity[serverID].Connected {
+				metrics := currMetrics.Connectivity[serverID]
+				metrics.NodeName = networkNodes[i].Name
+				metrics.IsServer = "no"
+				newServerMetrics.Connectivity[currNodeID] = metrics
+			}
+		} else {
+			newServerMetrics.Connectivity[currNodeID] = models.Metric{
+				Connected: false,
+				Latency:   999,
+			}
+		}
+	}
+	return &newServerMetrics
+}