MM-27199 - Add metrics to MM server for total enabled user count (#15116)

* Add metrics to MM server for total enabled user count and include installation ID from a new env var

* diagnostics context

Co-authored-by: Mattermod <mattermod@users.noreply.github.com>

app/app.go

@@ -120,7 +120,9 @@ func (a *App) initJobs() {
 	if jobsExpiryNotifyInterface != nil {
 		a.srv.Jobs.ExpiryNotify = jobsExpiryNotifyInterface(a)
 	}
-
+	if jobsActiveUsersInterface != nil {
+		a.srv.Jobs.ActiveUsers = jobsActiveUsersInterface(a)
+	}
 	a.srv.Jobs.Workers = a.srv.Jobs.InitWorkers()
 	a.srv.Jobs.Schedulers = a.srv.Jobs.InitSchedulers()
 }

app/diagnostics.go

@@ -112,10 +112,16 @@ func (s *Server) sendDailyDiagnostics(override bool) {
 
 func (s *Server) SendDiagnostic(event string, properties map[string]interface{}) {
 	if s.rudderClient != nil {
+		var context *rudder.Context
+		// if we are part of a cloud installation, add it's ID to the tracked event's context
+		if installationId := os.Getenv("MM_CLOUD_INSTALLATION_ID"); installationId != "" {
+			context = &rudder.Context{Traits: map[string]interface{}{"installationId": installationId}}
+		}
 		s.rudderClient.Enqueue(rudder.Track{
 			Event:      event,
 			UserId:     s.diagnosticId,
 			Properties: properties,
+			Context:    context,
 		})
 	}
 }

app/enterprise.go

@@ -90,6 +90,12 @@ func RegisterJobsBleveIndexerInterface(f func(*Server) tjobs.IndexerJobInterface
 	jobsBleveIndexerInterface = f
 }
 
+var jobsActiveUsersInterface func(*App) tjobs.ActiveUsersJobInterface
+
+func RegisterJobsActiveUsersInterface(f func(*App) tjobs.ActiveUsersJobInterface) {
+	jobsActiveUsersInterface = f
+}
+
 var jobsExpiryNotifyInterface func(*App) tjobs.ExpiryNotifyJobInterface
 
 func RegisterJobsExpiryNotifyJobInterface(f func(*App) tjobs.ExpiryNotifyJobInterface) {