mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-06-23 12:07:15 +08:00
89640938f0
1.suggestsys_const.go 中定义规则和资源列表中涉及到一些常量 2.suggestsysalart.go 1）新增一些列来匹配重用过滤;2) 进行删除相关操作时，通过task处理; 3.driverconfig.go 中包含目前driver中可以共用的函数 4.suggestsysruledrivers.go 调整driver interface 涉及的函数
257 lines
8.2 KiB
Go
257 lines
8.2 KiB
Go
// Copyright 2019 Yunion
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
|
||
package models
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"fmt"
|
||
"time"
|
||
|
||
"yunion.io/x/jsonutils"
|
||
"yunion.io/x/log"
|
||
"yunion.io/x/pkg/errors"
|
||
"yunion.io/x/pkg/tristate"
|
||
"yunion.io/x/sqlchemy"
|
||
|
||
"yunion.io/x/onecloud/pkg/apis/monitor"
|
||
"yunion.io/x/onecloud/pkg/cloudcommon/cronman"
|
||
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
||
"yunion.io/x/onecloud/pkg/httperrors"
|
||
"yunion.io/x/onecloud/pkg/mcclient"
|
||
"yunion.io/x/onecloud/pkg/util/stringutils2"
|
||
)
|
||
|
||
var (
|
||
SuggestSysRuleManager *SSuggestSysRuleManager
|
||
)
|
||
|
||
func init() {
|
||
SuggestSysRuleManager = &SSuggestSysRuleManager{
|
||
SVirtualResourceBaseManager: db.NewVirtualResourceBaseManager(
|
||
&SSuggestSysRule{},
|
||
"suggestsysrule_tbl",
|
||
"suggestsysrule",
|
||
"suggestsysrules",
|
||
),
|
||
}
|
||
SuggestSysRuleManager.SetVirtualObject(SuggestSysRuleManager)
|
||
}
|
||
|
||
type SSuggestSysRuleManager struct {
|
||
db.SVirtualResourceBaseManager
|
||
db.SEnabledResourceBaseManager
|
||
}
|
||
|
||
type SSuggestSysRule struct {
|
||
db.SVirtualResourceBase
|
||
db.SEnabledResourceBase
|
||
|
||
Type string `width:"256" charset:"ascii" list:"user" update:"user"`
|
||
Period string `width:"256" charset:"ascii" list:"user" update:"user"`
|
||
Setting jsonutils.JSONObject ` list:"user" update:"user"`
|
||
ExecTime time.Time `json:"exec_time"`
|
||
}
|
||
|
||
func (man *SSuggestSysRuleManager) FetchSuggestSysAlartSettings(ruleTypes ...string) (map[string]*monitor.SuggestSysRuleDetails, error) {
|
||
objs := make([]SSuggestSysRule, 0)
|
||
suggestSysAlerSettingMap := make(map[string]*monitor.SuggestSysRuleDetails, 0)
|
||
q := man.Query()
|
||
if q == nil {
|
||
fmt.Println(" query is nil")
|
||
}
|
||
if len(ruleTypes) != 0 {
|
||
q.Equals("type", ruleTypes)
|
||
}
|
||
err := db.FetchModelObjects(man, q, &objs)
|
||
if err != nil && err != sql.ErrNoRows {
|
||
return suggestSysAlerSettingMap, errors.Wrap(err, "FetchSuggestSysAlartSettings")
|
||
}
|
||
for _, config := range objs {
|
||
suggestSysRuleDetails := config.getMoreDetails(monitor.SuggestSysRuleDetails{})
|
||
if err != nil {
|
||
return suggestSysAlerSettingMap, errors.Wrap(err, "FetchSuggestSysAlartSettings")
|
||
}
|
||
suggestSysAlerSettingMap[config.Type] = &suggestSysRuleDetails
|
||
}
|
||
return suggestSysAlerSettingMap, nil
|
||
}
|
||
|
||
//根据数据库中查询得到的信息进行适配转换，同时更新drivers中的内容
|
||
func (dConfig *SSuggestSysRule) getSuggestSysAlertSetting() (*monitor.SSuggestSysAlertSetting, error) {
|
||
setting := new(monitor.SSuggestSysAlertSetting)
|
||
switch dConfig.Type {
|
||
case monitor.EIP_UN_USED:
|
||
setting.EIPUnused = new(monitor.EIPUnused)
|
||
err := dConfig.Setting.Unmarshal(setting.EIPUnused)
|
||
if err != nil {
|
||
return nil, errors.Wrap(err, "SSuggestSysRule getSuggestSysAlertSetting error")
|
||
}
|
||
}
|
||
return setting, nil
|
||
}
|
||
|
||
type DiskUnsed struct {
|
||
Status string
|
||
}
|
||
|
||
func (manager *SSuggestSysRuleManager) ListItemFilter(
|
||
ctx context.Context,
|
||
q *sqlchemy.SQuery,
|
||
userCred mcclient.TokenCredential,
|
||
query monitor.SuggestSysRuleListInput) (*sqlchemy.SQuery, error) {
|
||
var err error
|
||
q, err = manager.SVirtualResourceBaseManager.ListItemFilter(ctx, q, userCred, query.VirtualResourceListInput)
|
||
if err != nil {
|
||
return nil, errors.Wrap(err, "SVirtualResourceBaseManager.ListItemFilter")
|
||
}
|
||
q, err = manager.SEnabledResourceBaseManager.ListItemFilter(ctx, q, userCred, query.EnabledResourceBaseListInput)
|
||
if err != nil {
|
||
return nil, errors.Wrap(err, "SEnabledResourceBaseManager.ListItemFilter")
|
||
}
|
||
return q, nil
|
||
}
|
||
|
||
func (man *SSuggestSysRuleManager) ValidateCreateData(
|
||
ctx context.Context, userCred mcclient.TokenCredential,
|
||
ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject,
|
||
data monitor.SuggestSysRuleCreateInput) (monitor.SuggestSysRuleCreateInput, error) {
|
||
if data.Period == "" {
|
||
// default 30s
|
||
data.Period = "30s"
|
||
}
|
||
if _, err := time.ParseDuration(data.Period); err != nil {
|
||
return data, httperrors.NewInputParameterError("Invalid period format: %s", data.Period)
|
||
}
|
||
if dri, ok := suggestSysRuleDrivers[data.Type]; !ok {
|
||
return data, httperrors.NewInputParameterError("not support type %q", data.Type)
|
||
} else {
|
||
//Type is uniq
|
||
err := db.NewNameValidator(man, ownerId, data.Type, "")
|
||
if err != nil {
|
||
return data, err
|
||
}
|
||
if data.Setting != nil {
|
||
err = dri.ValidateSetting(data.Setting)
|
||
if err != nil {
|
||
return data, errors.Wrap(err, "validate setting error")
|
||
}
|
||
}
|
||
}
|
||
return data, nil
|
||
}
|
||
|
||
func (rule *SSuggestSysRule) ValidateUpdateData(
|
||
ctx context.Context, userCred mcclient.TokenCredential,
|
||
query jsonutils.JSONObject,
|
||
data monitor.SuggestSysRuleUpdateInput) (monitor.SuggestSysRuleUpdateInput, error) {
|
||
if data.Period == "" {
|
||
// default 30s
|
||
data.Period = "30s"
|
||
}
|
||
if data.Enabled != nil {
|
||
rule.SetEnabled(*data.Enabled)
|
||
}
|
||
if _, err := time.ParseDuration(data.Period); err != nil {
|
||
return data, httperrors.NewInputParameterError("Invalid period format: %s", data.Period)
|
||
}
|
||
if data.Setting != nil {
|
||
err := suggestSysRuleDrivers[rule.Type].ValidateSetting(data.Setting)
|
||
if err != nil {
|
||
return data, errors.Wrap(err, "validate setting error")
|
||
}
|
||
}
|
||
return data, nil
|
||
}
|
||
|
||
func (man *SSuggestSysRuleManager) FetchCustomizeColumns(
|
||
ctx context.Context,
|
||
userCred mcclient.TokenCredential,
|
||
query jsonutils.JSONObject,
|
||
objs []interface{},
|
||
fields stringutils2.SSortedStrings,
|
||
isList bool,
|
||
) []monitor.SuggestSysRuleDetails {
|
||
rows := make([]monitor.SuggestSysRuleDetails, len(objs))
|
||
virtRows := man.SVirtualResourceBaseManager.FetchCustomizeColumns(ctx, userCred, query, objs, fields, isList)
|
||
for i := range rows {
|
||
rows[i] = monitor.SuggestSysRuleDetails{
|
||
VirtualResourceDetails: virtRows[i],
|
||
}
|
||
}
|
||
return rows
|
||
}
|
||
|
||
func (self *SSuggestSysRule) getMoreDetails(out monitor.SuggestSysRuleDetails) monitor.SuggestSysRuleDetails {
|
||
var err error
|
||
out.Setting, err = self.getSuggestSysAlertSetting()
|
||
if err != nil {
|
||
log.Errorln("getMoreDetails err:", err)
|
||
}
|
||
out.ID = self.Id
|
||
out.Name = self.Name
|
||
out.Enabled = self.GetEnabled()
|
||
return out
|
||
}
|
||
|
||
func (self *SSuggestSysRule) GetExtraDetails(
|
||
ctx context.Context,
|
||
userCred mcclient.TokenCredential,
|
||
query jsonutils.JSONObject,
|
||
isList bool,
|
||
) (monitor.SuggestSysRuleDetails, error) {
|
||
return monitor.SuggestSysRuleDetails{}, nil
|
||
}
|
||
|
||
//after create, update Cronjob's info
|
||
func (self *SSuggestSysRule) PostCreate(ctx context.Context, userCred mcclient.TokenCredential, ownerId mcclient.IIdentityProvider, query jsonutils.JSONObject, data jsonutils.JSONObject) {
|
||
self.SVirtualResourceBase.PostCreate(ctx, userCred, ownerId, query, data)
|
||
cronman.GetCronJobManager().Remove(self.Type)
|
||
if self.Enabled.Bool() {
|
||
dur, _ := time.ParseDuration(self.Period)
|
||
cronman.GetCronJobManager().AddJobAtIntervalsWithStartRun(self.Type, dur,
|
||
suggestSysRuleDrivers[self.Type].DoSuggestSysRule, true)
|
||
}
|
||
}
|
||
|
||
//after update, update Cronjob's info
|
||
func (self *SSuggestSysRule) PostUpdate(
|
||
ctx context.Context, userCred mcclient.TokenCredential,
|
||
query jsonutils.JSONObject, data jsonutils.JSONObject) {
|
||
cronman.GetCronJobManager().Remove(self.Type)
|
||
if self.Enabled.Bool() {
|
||
dur, _ := time.ParseDuration(self.Period)
|
||
cronman.GetCronJobManager().AddJobAtIntervalsWithStartRun(self.Type, dur,
|
||
suggestSysRuleDrivers[self.Type].DoSuggestSysRule, true)
|
||
}
|
||
}
|
||
|
||
func (self *SSuggestSysRule) AllowPerformEnable(ctx context.Context, userCred mcclient.TokenCredential,
|
||
query jsonutils.JSONObject, data jsonutils.JSONObject) bool {
|
||
return db.IsAdminAllowPerform(userCred, self, "Enable")
|
||
}
|
||
|
||
func (self *SSuggestSysRule) PerformEnable(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
|
||
if !self.Enabled.Bool() {
|
||
db.Update(self, func() error {
|
||
self.Enabled = tristate.True
|
||
return nil
|
||
})
|
||
db.OpsLog.LogEvent(self, db.ACT_ENABLE, "", userCred)
|
||
self.PostUpdate(ctx, userCred, query, data)
|
||
}
|
||
return nil, nil
|
||
}
|