feat:新增单机版本 (#163)

* refactor:

* fix:新增CLS链接信息

* feat:单机版本

* feat:单机版本

Author: chuntaojun

.gitignore

@@ -17,4 +17,5 @@ vendor
 
 polaris_console_package
 polaris-frontend.tar.gz
-.vscode
+.vscode
+polaris.bolt

common/model/alarm/types.go

@@ -87,7 +87,6 @@ type AlarmRule struct {
 	CreateTime  time.Time
 	ModifyTime  time.Time
 	EnableTime  time.Time
-	Valid       bool
 }
 
 type AlterExpr struct {

handlers/alarm_rule.go

@@ -37,6 +37,7 @@ import (
 var (
 	_alarmRuleSearchFilter = map[string]struct{}{
 		"name": {},
+		"id":   {},
 	}
 )
 

plugins.go

@@ -18,5 +18,6 @@
 package main
 
 import (
+	_ "github.com/polarismesh/polaris-console/store/boltdb"
 	_ "github.com/polarismesh/polaris-console/store/mysql"
 )

polaris-console.yaml

@@ -31,19 +31,19 @@ monitorServer:
 # 存储配置
 store:
   # 单机文件存储插件
-  # name: boltdbStore
-  # option:
-  #   path: ./polaris.bolt
-  # 数据库存储插件
-  name: defaultStore
+  name: boltdbStore
   option:
-    master:
-      dbType: mysql
-      dbName: polaris_console
-      dbUser: root ##DB_USER##
-      dbPwd: 528LcT528 ##DB_PWD##
-      dbAddr: 9.135.223.132:3306 ##DB_ADDR##
-      maxOpenConns: 300
-      maxIdleConns: 50
-      connMaxLifetime: 300 # 单位秒
-      txIsolationLevel: 2 #LevelReadCommitted
+    path: ./polaris.bolt
+  # 数据库存储插件
+  # name: defaultStore
+  # option:
+  #   master:
+  #     dbType: mysql
+  #     dbName: polaris_console
+  #     dbUser: ##DB_USER##
+  #     dbPwd: ##DB_PWD##
+  #     dbAddr: ##DB_ADDR##
+  #     maxOpenConns: 300
+  #     maxIdleConns: 50
+  #     connMaxLifetime: 300 # 单位秒
+  #     txIsolationLevel: 2 #LevelReadCommitted

store/boltdb/alarm_rule.go

@@ -17,42 +17,268 @@
 
 package boltdb
 
-import "github.com/polarismesh/polaris-console/common/model/alarm"
+import (
+	"encoding/json"
+	"errors"
+	"sort"
+	"strings"
+	"time"
+
+	commonhttp "github.com/polarismesh/polaris-console/common/http"
+	"github.com/polarismesh/polaris-console/common/model/alarm"
+)
+
+var (
+	ErrorMultiAlarmFound = errors.New("found multi alarm rule")
+)
+
+const (
+	tblAlarm = "alarm_rule"
+
+	alarmFieldID          = "ID"
+	alarmFieldName        = "Name"
+	alarmFieldEnable      = "Enable"
+	alarmFieldMonitorType = "MonitorType"
+	alarmFieldAlterExpr   = "AlterExpr"
+	alarmFieldInterval    = "Interval"
+	alarmFieldTopic       = "Topic"
+	alarmFieldMessage     = "Message"
+	alarmFieldCallback    = "Callback"
+	alarmFieldRevision    = "Revision"
+	alarmFieldCreateTime  = "CreateTime"
+	alarmFieldModifyTime  = "ModifyTime"
+	alarmFieldEnableTime  = "EnableTime"
+)
+
+type alarmRuleForStore struct {
+	ID          string
+	Name        string
+	Enable      bool
+	MonitorType string
+	AlterExpr   string
+	Interval    string
+	Topic       string
+	Message     string
+	Callback    string
+	Revision    string
+	CreateTime  time.Time
+	ModifyTime  time.Time
+	EnableTime  time.Time
+}
 
 type alarmRuleStore struct {
 	handle BoltHandler
 }
 
 // AddAlarmRule add alarm rule
 func (a *alarmRuleStore) AddAlarmRule(rule *alarm.AlarmRule) error {
+	if len(rule.ID) == 0 {
+		return errors.New("invalid alarm rule id")
+	}
+
+	tn := time.Now()
+	rule.CreateTime = tn
+	rule.ModifyTime = tn
 
-	return nil
+	if rule.Enable {
+		rule.EnableTime = tn
+	} else {
+		rule.EnableTime = time.Time{}
+	}
+
+	return a.handle.SaveValue(tblAlarm, rule.ID, a.parseToStore(rule))
 }
 
 // UpdateAlarmRule update alarm rule
 func (a *alarmRuleStore) UpdateAlarmRule(rule *alarm.AlarmRule) error {
-	return nil
+	if len(rule.ID) == 0 {
+		return errors.New("invalid alarm rule id")
+	}
+
+	alterExpr, _ := json.Marshal(rule.AlterExpr)
+	callback, _ := json.Marshal(rule.Callback)
+
+	properties := map[string]interface{}{
+		alarmFieldName:        rule.Name,
+		alarmFieldAlterExpr:   alterExpr,
+		alarmFieldCallback:    callback,
+		alarmFieldInterval:    rule.Interval,
+		alarmFieldTopic:       rule.Topic,
+		alarmFieldMessage:     rule.Message,
+		alarmFieldRevision:    rule.Revision,
+		alarmFieldMonitorType: rule.MonitorType,
+		alarmFieldModifyTime:  time.Now(),
+	}
+
+	return a.handle.UpdateValue(tblAlarm, rule.ID, properties)
 }
 
 // DeleteAlarmRule delete alarm rule
 func (a *alarmRuleStore) DeleteAlarmRule(rule *alarm.AlarmRule) error {
-	return nil
+	if len(rule.ID) == 0 {
+		return errors.New("invalid alarm rule id")
+	}
+	return a.handle.DeleteValues(tblAlarm, []string{rule.ID})
 }
 
 // EnableAlarmRule enable alarm rule
 func (a *alarmRuleStore) EnableAlarmRule(rule *alarm.AlarmRule) error {
-	return nil
+	if len(rule.ID) == 0 {
+		return errors.New("invalid alarm rule id")
+	}
+
+	if rule.Enable {
+		rule.EnableTime = time.Now()
+	} else {
+		rule.EnableTime = time.Time{}
+	}
+
+	properties := map[string]interface{}{
+		alarmFieldEnable:     rule.Enable,
+		alarmFieldRevision:   rule.Revision,
+		alarmFieldEnableTime: rule.EnableTime,
+		alarmFieldModifyTime: time.Now(),
+	}
+
+	return a.handle.UpdateValue(tblAlarm, rule.ID, properties)
 }
 
 func (a *alarmRuleStore) GetOneAlarmRule(id string) (*alarm.AlarmRule, error) {
-	return nil, nil
+	if len(id) == 0 {
+		return nil, errors.New("invalid alarm rule id")
+	}
+
+	val, err := a.handle.LoadValues(tblAlarm, []string{id}, &alarmRuleForStore{})
+	if err != nil {
+		return nil, err
+	}
+
+	if len(val) == 0 {
+		return nil, nil
+	}
+
+	if len(val) > 1 {
+		return nil, ErrorMultiAlarmFound
+	}
+
+	return a.parseToModel(val[id].(*alarmRuleForStore)), nil
 }
 
 // GetAlarmRules get alarm rules
 func (a *alarmRuleStore) GetAlarmRules(query map[string]string, offset, limit uint32) (uint32, []*alarm.AlarmRule, error) {
-	return 0, nil, nil
+	fields := []string{alarmFieldID, alarmFieldName}
+	searchId, hasId := query["id"]
+	searchName, hashName := query["name"]
+
+	ret, err := a.handle.LoadValuesByFilter(tblAlarm, fields, &alarmRuleForStore{}, func(m map[string]interface{}) bool {
+		saveId, _ := m[alarmFieldID].(string)
+		saveName, _ := m[alarmFieldName].(string)
+
+		if hasId && strings.Compare(saveId, searchId) != 0 {
+			return false
+		}
+
+		if hashName {
+			searchName, wild := commonhttp.ParseWildName(searchName)
+			if wild && !strings.Contains(saveName, searchName) {
+				return false
+			}
+			return strings.Compare(searchName, saveName) == 0
+		}
+
+		return true
+	})
+	if err != nil {
+		return 0, nil, err
+	}
+
+	return uint32(len(ret)), a.doAlarmRulePage(ret, offset, limit), nil
+}
+
+// doAlarmRulePage 进行分页
+func (a *alarmRuleStore) doAlarmRulePage(ret map[string]interface{}, offset, limit uint32) []*alarm.AlarmRule {
+	rules := make([]*alarm.AlarmRule, 0, len(ret))
+	beginIndex := offset
+	endIndex := beginIndex + limit
+	totalCount := uint32(len(ret))
+
+	if totalCount == 0 {
+		return rules
+	}
+	if beginIndex >= endIndex {
+		return rules
+	}
+	if beginIndex >= totalCount {
+		return rules
+	}
+	if endIndex > totalCount {
+		endIndex = totalCount
+	}
+	for k := range ret {
+		rules = append(rules, a.parseToModel(ret[k].(*alarmRuleForStore)))
+	}
+
+	sort.Slice(rules, func(i, j int) bool {
+		return rules[i].ModifyTime.After(rules[j].ModifyTime)
+	})
+
+	return rules[beginIndex:endIndex]
 }
 
 func (a *alarmRuleStore) GetAllAlarmRules() ([]*alarm.AlarmRule, error) {
-	return nil, nil
+	ret, err := a.handle.LoadValuesAll(tblAlarm, &alarmRuleForStore{})
+	if err != nil {
+		return nil, err
+	}
+
+	values := make([]*alarm.AlarmRule, 0, len(ret))
+	for k := range ret {
+		values = append(values, a.parseToModel(ret[k].(*alarmRuleForStore)))
+	}
+
+	return values, nil
+}
+
+func (a *alarmRuleStore) parseToStore(r *alarm.AlarmRule) *alarmRuleForStore {
+	alterExpr, _ := json.Marshal(r.AlterExpr)
+	callback, _ := json.Marshal(r.Callback)
+
+	return &alarmRuleForStore{
+		ID:          r.ID,
+		Name:        r.Name,
+		Enable:      r.Enable,
+		MonitorType: string(r.MonitorType),
+		AlterExpr:   string(alterExpr),
+		Interval:    r.Interval,
+		Topic:       r.Topic,
+		Message:     r.Message,
+		Callback:    string(callback),
+		Revision:    r.Revision,
+		CreateTime:  r.CreateTime,
+		ModifyTime:  r.ModifyTime,
+		EnableTime:  r.EnableTime,
+	}
+}
+
+func (a *alarmRuleStore) parseToModel(r *alarmRuleForStore) *alarm.AlarmRule {
+	var alterExpr alarm.AlterExpr
+	_ = json.Unmarshal([]byte(r.AlterExpr), &alterExpr)
+	var callback alarm.Callback
+	_ = json.Unmarshal([]byte(r.Callback), &callback)
+
+	return &alarm.AlarmRule{
+		ID:          r.ID,
+		Name:        r.Name,
+		Enable:      r.Enable,
+		MonitorType: alarm.MonitorType(r.MonitorType),
+		AlterExpr:   alterExpr,
+		Interval:    r.Interval,
+		Topic:       r.Topic,
+		Message:     r.Message,
+		Callback:    callback,
+		Revision:    r.Revision,
+		CreateTime:  r.CreateTime,
+		ModifyTime:  r.ModifyTime,
+		EnableTime:  r.EnableTime,
+	}
 }