Cloudpods/backend/pkg/mcclient/options/monitor/alertrecord.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 monitor

import (
	"time"

	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/errors"

	"yunion.io/x/onecloud/pkg/apis"
	"yunion.io/x/onecloud/pkg/mcclient/options"
)

type AlertRecordListOptions struct {
	options.BaseListOptions

	AlertId   string   `help:"id of alert"`
	Level     string   `help:"alert level"`
	State     string   `help:"alert state"`
	ResTypes  []string `json:"res_types"`
	ResId     string   `json:"res_id"`
	ResName   string   `json:"res_name"`
	Alerting  bool     `json:"alerting"`
	AlertName string   `json:"alert_name"`
}

func (o *AlertRecordListOptions) Params() (jsonutils.JSONObject, error) {
	return options.ListStructToParams(o)
}

type AlertRecordShowOptions struct {
	ID string `help:"ID of Metric " json:"-"`
}

func (o *AlertRecordShowOptions) Params() (jsonutils.JSONObject, error) {
	return options.StructToParams(o)
}

func (o *AlertRecordShowOptions) GetId() string {
	return o.ID
}

type AlertRecordTotalOptions struct {
	options.BaseListOptions
}

func (o *AlertRecordTotalOptions) Params() (jsonutils.JSONObject, error) {
	return options.ListStructToParams(o)
}

func (o *AlertRecordTotalOptions) Property() string {
	return "total-alert"
}

type AlertRecordHistoryAlertOptions struct {
	options.BaseListOptions
}

func (o *AlertRecordHistoryAlertOptions) Params() (jsonutils.JSONObject, error) {
	return options.ListStructToParams(o)
}

func (o *AlertRecordHistoryAlertOptions) Property() string {
	return "history-alert"
}

type AlertRecordProjectAlertResourceCountOptions struct {
	StartTime time.Time `help:"start time (RFC3339 format)" json:"start_time" default:"2025-01-01 00:00:00"`
	EndTime   time.Time `help:"end time (RFC3339 format)" json:"end_time" default:"2025-01-01 00:00:00"`
	ResType   string    `help:"resource type" json:"res_type"`
	AlertId   string    `help:"alert id" json:"alert_id"`
	Scope     string    `help:"scope" json:"scope" choices:"system|domain|project"`
}

func (o *AlertRecordProjectAlertResourceCountOptions) Params() (jsonutils.JSONObject, error) {
	return options.StructToParams(o)
}

func (o *AlertRecordProjectAlertResourceCountOptions) Property() string {
	return "project-alert-resource-count"
}

type AlertRecordShieldListOptions struct {
	options.BaseListOptions

	AlertId   string   `help:"id of alert"`
	Alertname string   `json:"alertname"`
	ResName   string   `json:"res_name"`
	ResTypes  []string `json:"res_types"`
}

func (o *AlertRecordShieldListOptions) Params() (jsonutils.JSONObject, error) {
	return options.ListStructToParams(o)
}

type AlertRecordShieldShowOptions struct {
	ID string `help:"ID of Metric " json:"-"`
}

func (o *AlertRecordShieldShowOptions) Params() (jsonutils.JSONObject, error) {
	return options.StructToParams(o)
}

func (o *AlertRecordShieldShowOptions) GetId() string {
	return o.ID
}

type AlertRecordShieldDeleteOptions struct {
	ID string `json:"-"`
}

func (o *AlertRecordShieldDeleteOptions) GetId() string {
	return o.ID
}

func (o *AlertRecordShieldDeleteOptions) Params() (jsonutils.JSONObject, error) {
	return options.StructToParams(o)
}

type AlertRecordShieldCreateOptions struct {
	apis.ScopedResourceCreateInput

	AlertId string `json:"alert_id" help:"common alert Id" required:"true"`
	ResType string `json:"res_type" help:"resource tyge" choices:"host|guest|redis|oss|rds|cloudaccount"`
	ResName string `json:"res_name" help:"resource name" required:"true"`
	ResId   string `json:"res_id" help:"resource id" required:"true"`

	ShieldPeriod string `json:"shield_period" help:"shield time eg:'1m,2h'" required:"true"`
}

func (o *AlertRecordShieldCreateOptions) Params() (jsonutils.JSONObject, error) {
	params, err := options.StructToParams(o)
	if err != nil {
		return nil, err
	}
	duration, err := time.ParseDuration(o.ShieldPeriod)
	if err != nil {
		return nil, errors.Wrap(err, "parse shield_period err")
	}
	startTime := time.Now()
	endTime := startTime.Add(duration)
	params.Add(jsonutils.NewTimeString(startTime), "start_time")
	params.Add(jsonutils.NewTimeString(endTime), "end_time")
	return params, nil
}