54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
|
package monitor
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/wanghuiyt/ding"
|
|||
|
|
"samba/pkg/ksync"
|
|||
|
|
"samba/pkg/log"
|
|||
|
|
"samba/pkg/task"
|
|||
|
|
"samba/server/other/handler"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Ding struct {
|
|||
|
|
*ding.Webhook
|
|||
|
|
sendDing bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *Ding) SendMessageText(text string, at ...string) error {
|
|||
|
|
log.Debug(fmt.Sprintf("发送钉钉消息：%s", text))
|
|||
|
|
if !d.sendDing {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
return d.Webhook.SendMessageText(text, at...)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *Ding) Apply(t *task.Task) {
|
|||
|
|
branch, _ := handler.GetEnv[string]("GitBranch")
|
|||
|
|
if strings.Contains(strings.ToLower(branch), "release") {
|
|||
|
|
d.sendDing = true
|
|||
|
|
}
|
|||
|
|
if !d.sendDing {
|
|||
|
|
log.Debug(fmt.Sprintf("Task:%s 非release版本，不发送至钉钉", t.Name()))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
d.Webhook = &ding.Webhook{
|
|||
|
|
AccessToken: ksync.DingAccessToken,
|
|||
|
|
Secret: ksync.DingSecret,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type TestTicker struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (*TestTicker) Apply(t *task.Task) {
|
|||
|
|
// FIXME for test
|
|||
|
|
ksync.DingAccessToken = "e20c2217ae056af5d574d7a5c01340ce5d72eb77ae4ff05974e8c72f3a6b1ca2"
|
|||
|
|
ksync.DingSecret = "SECfc662dd07db57dea77bd2e44798cb88d89fec9c6ced8207408fd9d7c1f377dfb"
|
|||
|
|
|
|||
|
|
log.Debug(fmt.Sprintf("任务 %s 测试ticker为10秒", t.Name()))
|
|||
|
|
t.Ticker(10 * time.Second).Fixed()
|
|||
|
|
}
|