MM-28733 : Admin Advisor v2 (#15515)

store/sqlstore/system_store.go

@@ -6,9 +6,13 @@ package sqlstore
 import (
 	"context"
 	"database/sql"
+	"strconv"
+	"strings"
+	"time"
 
 	"github.com/mattermost/mattermost-server/v5/model"
 	"github.com/mattermost/mattermost-server/v5/store"
+	"github.com/mattermost/mattermost-server/v5/utils"
 
 	"github.com/pkg/errors"
 )
@@ -52,6 +56,26 @@ func (s SqlSystemStore) SaveOrUpdate(system *model.System) error {
 	return nil
 }
 
+func (s SqlSystemStore) SaveOrUpdateWithWarnMetricHandling(system *model.System) error {
+	if err := s.GetMaster().SelectOne(&model.System{}, "SELECT * FROM Systems WHERE Name = :Name", map[string]interface{}{"Name": system.Name}); err == nil {
+		if _, err := s.GetMaster().Update(system); err != nil {
+			return errors.Wrapf(err, "failed to update system property with name=%s", system.Name)
+		}
+	} else {
+		if err := s.GetMaster().Insert(system); err != nil {
+			return errors.Wrapf(err, "failed to save system property with name=%s", system.Name)
+		}
+	}
+
+	if strings.HasPrefix(system.Name, model.WARN_METRIC_STATUS_STORE_PREFIX) && (system.Value == model.WARN_METRIC_STATUS_RUNONCE || system.Value == model.WARN_METRIC_STATUS_LIMIT_REACHED) {
+		if err := s.SaveOrUpdate(&model.System{Name: model.SYSTEM_WARN_METRIC_LAST_RUN_TIMESTAMP_KEY, Value: strconv.FormatInt(utils.MillisFromTime(time.Now()), 10)}); err != nil {
+			return errors.Wrapf(err, "failed to save system property with name=%s", model.SYSTEM_WARN_METRIC_LAST_RUN_TIMESTAMP_KEY)
+		}
+	}
+
+	return nil
+}
+
 func (s SqlSystemStore) Update(system *model.System) error {
 	if _, err := s.GetMaster().Update(system); err != nil {
 		return errors.Wrapf(err, "failed to update system property with name=%s", system.Name)