Prometheus X Vector

infrastructure/monitor/monitor.go

@@ -0,0 +1,75 @@
+package monitor
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"oc-datacenter/conf"
+	"sync"
+	"time"
+
+	oclib "cloud.o-forge.io/core/oc-lib"
+	"cloud.o-forge.io/core/oc-lib/dbs"
+	"cloud.o-forge.io/core/oc-lib/models/booking"
+	"cloud.o-forge.io/core/oc-lib/models/common/models"
+	"cloud.o-forge.io/core/oc-lib/models/live"
+	"cloud.o-forge.io/core/oc-lib/models/resources"
+	"github.com/gorilla/websocket"
+)
+
+type MonitorInterface interface {
+	Stream(ctx context.Context, bookingID string, interval time.Duration, ws *websocket.Conn)
+}
+
+var _monitorService = map[string]func() MonitorInterface{
+	"prometheus": func() MonitorInterface { return NewPrometheusService() },
+	"vector":     func() MonitorInterface { return NewVectorService() },
+}
+
+func NewMonitorService() (MonitorInterface, error) {
+	service, ok := _monitorService[conf.GetConfig().MonitorMode]
+	if !ok {
+		return nil, errors.New("monitor service not found")
+	}
+	return service(), nil
+}
+
+func Call(book *booking.Booking,
+	f func(*live.LiveDatacenter, *resources.ComputeResourceInstance,
+		map[string]models.MetricsSnapshot, *sync.WaitGroup, *sync.Mutex)) (*booking.Booking, map[string]models.MetricsSnapshot) {
+	logger := oclib.GetLogger()
+	metrics := map[string]models.MetricsSnapshot{}
+	var wg sync.WaitGroup
+	var mu sync.Mutex
+
+	cUAccess := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.LIVE_DATACENTER), nil)
+	cRAccess := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.COMPUTE_RESOURCE), nil)
+
+	rr := cRAccess.LoadOne(book.ResourceID)
+	if rr.Err != "" {
+		logger.Err(fmt.Errorf("can't proceed because of unfound resource %s : %s", book.ResourceID, rr.Err))
+		return book, metrics
+	}
+	computeRes := rr.ToComputeResource()
+	for _, instance := range computeRes.Instances {
+		res := cUAccess.Search(&dbs.Filters{
+			And: map[string][]dbs.Filter{
+				"source":                    {{Operator: dbs.EQUAL.String(), Value: instance.Source}},
+				"abstractlive.resources_id": {{Operator: dbs.EQUAL.String(), Value: computeRes.GetID()}},
+			},
+		}, "", false)
+		if res.Err != "" {
+			continue
+		}
+		for _, r := range res.Data {
+			dc := r.(*live.LiveDatacenter)
+			if dc.MonitorPath == "" {
+				continue
+			}
+			wg.Add(1)
+			go f(dc, instance, metrics, &wg, &mu)
+		}
+	}
+	wg.Wait()
+	return book, metrics
+}