Rainbond/grctl/cmd/notification_event.go

package cmd

import (
	"fmt"
	"strconv"
	"time"

	"github.com/apcera/termtables"
	"github.com/goodrain/rainbond/grctl/clients"
	"github.com/urfave/cli"
)

//NewCmdNode NewCmdNode
func NewCmdNotificationEvent() cli.Command {
	c := cli.Command{
		Name:  "msg",
		Usage: "应用异常通知事件。grctl msg",
		Subcommands: []cli.Command{
			{
				Name:  "get",
				Usage: "获取未处理事件，不指定起止时间默认72小时内",
				Flags: []cli.Flag{
					cli.StringFlag{
						Name:  "StartTime,st",
						Value: "",
						Usage: "StartTime timestamp",
					},
					cli.StringFlag{
						Name:  "EndTime,et",
						Value: "",
						Usage: "EndTime timestamp",
					},
				},
				Action: func(c *cli.Context) error {
					Common(c)
					startTime := c.String("StartTime")
					EndTme := c.String("EndTime")
					if startTime == "" && EndTme == "" {
						NowTime := time.Now()
						startTimeTimestamp := NowTime.AddDate(0, 0, -3).Unix()
						startTime = strconv.FormatInt(startTimeTimestamp, 10)
						EndTme = strconv.FormatInt(NowTime.Unix(), 10)
					} else if EndTme == "" && startTime != "" {
						NowTime := time.Now().Unix()
						EndTme = strconv.FormatInt(NowTime, 10)
					}
					val, err := clients.RegionClient.Notification().GetNotification(startTime, EndTme)
					handleErr(err)
					serviceTable := termtables.CreateTable()
					serviceTable.AddHeaders("TenantName(租户别名)/ServiceName(应用别名)", "Message(异常信息)", "Reason(异常原因)", "Count(出现次数)", "LastTime(最后一次异常时间)", "FirstTime(第一次异常时间)")
					for _, v := range val {
						if v.KindID == "" || v.ServiceName == "" || v.TenantName == "" {
							continue
						}
						serviceTable.AddRow(v.TenantName+"/"+v.ServiceName, v.Message, v.Reason, v.Count, v.LastTime, v.FirstTime)
					}
					fmt.Println(serviceTable.Render())
					return nil
				},
			},
			{
				Name:  "handle",
				Usage: "handle --help",
				Flags: []cli.Flag{
					cli.StringFlag{
						Name:  "ServiceName,n",
						Value: "",
						Usage: "ServiceName",
					},
					cli.StringFlag{
						Name:  "HandleMessage,m",
						Value: "",
						Usage: "HandleMessage",
					},
				},
				Action: func(c *cli.Context) error {
					Common(c)
					if !c.IsSet("ServiceName") {
						println("ServiceName must not null")
						return nil
					}
					serviceName := c.String("ServiceName")
					handleMessage := c.String("HandleMessage")
					_, err := clients.RegionClient.Notification().HandleNotification(serviceName, handleMessage)
					handleErr(err)
					fmt.Println("Handling successfully")
					return nil
				},
			},
		},
	}
	return c
}