mirror of
https://gitee.com/mirrors/AllinSSL.git
synced 2026-03-14 02:20:53 +08:00
【调整】暗色主题样式
This commit is contained in:
cai
194
frontend/allinssl/backend/scheduler/monitor.go
Normal file
194
frontend/allinssl/backend/scheduler/monitor.go
Normal file
@@ -0,0 +1,194 @@
|
||||
package scheduler
|
||||
|
||||
import (
|
||||
"ALLinSSL/backend/internal/monitor"
|
||||
"ALLinSSL/backend/internal/report"
|
||||
"ALLinSSL/backend/public"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
MonitorErrCount = make(map[int64]int)
|
||||
monitorErrCountMu sync.Mutex
|
||||
)
|
||||
|
||||
// 通知模板，先写死，后续做成配置
|
||||
var MonitorErrTemplate = "监控名称：%v\n类型：%v\n域名：%v\n错误信息：%v\n请及时处理！\n检测时间：%v"
|
||||
|
||||
func Monitor() {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
fmt.Printf("Monitor 主流程捕获 panic: %v\n", r)
|
||||
}
|
||||
}()
|
||||
|
||||
s, err := monitor.GetSqlite()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
defer s.Close()
|
||||
s1, err := report.GetSqlite()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
defer s1.Close()
|
||||
s1.TableName = "report"
|
||||
|
||||
data, err := s.Select()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
now := time.Now()
|
||||
loc := now.Location()
|
||||
var wg sync.WaitGroup
|
||||
for _, v := range data {
|
||||
if v["active"].(int64) == 1 {
|
||||
lastTimeStr, ok := v["last_time"].(string)
|
||||
if !ok || lastTimeStr == "" {
|
||||
lastTimeStr = "1970-01-01 00:00:00"
|
||||
}
|
||||
lastTime, err := time.ParseInLocation("2006-01-02 15:04:05", lastTimeStr, loc)
|
||||
if err != nil {
|
||||
// fmt.Println(err)
|
||||
continue
|
||||
}
|
||||
monitorType, ok := v["monitor_type"].(string)
|
||||
if !ok {
|
||||
fmt.Println("监控类型错误")
|
||||
return
|
||||
}
|
||||
target, ok := v["target"].(string)
|
||||
if !ok {
|
||||
fmt.Println("监控目标错误")
|
||||
return
|
||||
}
|
||||
advanceDay, ok := v["advance_day"].(int64)
|
||||
if !ok {
|
||||
advanceDay = 30 // 默认提前30天
|
||||
}
|
||||
if now.Sub(lastTime).Minutes() >= float64(v["cycle"].(int64)) {
|
||||
wg.Add(1)
|
||||
go func(v map[string]any) {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
fmt.Printf("监控任务发生错误: %v\n", r)
|
||||
// 打印堆栈
|
||||
buf := make([]byte, 1<<16) // 64KB
|
||||
n := runtime.Stack(buf, false)
|
||||
fmt.Println("堆栈信息:\n", string(buf[:n]))
|
||||
}
|
||||
}()
|
||||
defer wg.Done()
|
||||
gs := *s
|
||||
var (
|
||||
certInfo *monitor.CertInfo
|
||||
certJson string
|
||||
Err error
|
||||
checkErr string
|
||||
)
|
||||
switch monitorType {
|
||||
case "https":
|
||||
certInfo, Err = monitor.CheckHttps(target, int(advanceDay))
|
||||
case "smtp":
|
||||
certInfo, Err = monitor.CheckSmtp(target, int(advanceDay))
|
||||
default:
|
||||
Err = fmt.Errorf("不支持的监控类型:%s", monitorType)
|
||||
}
|
||||
|
||||
if Err != nil {
|
||||
checkErr = Err.Error()
|
||||
} else {
|
||||
if certInfo.VerifyError != "" && (!certInfo.Valid || certInfo.DaysLeft <= int(advanceDay)) {
|
||||
checkErr = certInfo.VerifyError
|
||||
}
|
||||
certBytes, err := json.Marshal(certInfo)
|
||||
if err == nil {
|
||||
certJson = string(certBytes)
|
||||
}
|
||||
}
|
||||
id := v["id"].(int64)
|
||||
|
||||
// 此处应该发送错误邮件
|
||||
if checkErr != "" {
|
||||
// 更新监控记录
|
||||
valid := -1 // 状态为异常
|
||||
if certInfo != nil && certInfo.Valid {
|
||||
valid = 1 // 状态为正常
|
||||
}
|
||||
gs.Where("id=?", []interface{}{id}).Update(map[string]any{
|
||||
"last_time": now.Format("2006-01-02 15:04:05"),
|
||||
"except_end_time": now.Format("2006-01-02 15:04:05"),
|
||||
"info": certJson,
|
||||
"valid": valid, // 状态为异常
|
||||
})
|
||||
// 新增错误记录
|
||||
if certInfo == nil || !certInfo.Valid {
|
||||
gs.TableName = "err_record"
|
||||
gs.Insert(map[string]any{
|
||||
"id": public.GenerateUUID(),
|
||||
"monitor_id": id,
|
||||
"create_time": now.Format("2006-01-02 15:04:05"),
|
||||
"msg": checkErr,
|
||||
"info": certJson,
|
||||
})
|
||||
}
|
||||
monitorErrCountMu.Lock()
|
||||
MonitorErrCount[id] += 1
|
||||
errCount := MonitorErrCount[id]
|
||||
monitorErrCountMu.Unlock()
|
||||
|
||||
repeatSendGap, ok := v["repeat_send_gap"].(int64)
|
||||
if !ok {
|
||||
repeatSendGap = 10
|
||||
}
|
||||
reportTypes, ok := v["report_types"].(string)
|
||||
if ok && errCount == 1 {
|
||||
reportTypeArr := strings.Split(reportTypes, ",")
|
||||
for _, reportType := range reportTypeArr {
|
||||
if reportType == "" {
|
||||
continue
|
||||
}
|
||||
rdata, err := s1.Where("type=?", []interface{}{reportType}).Select()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if len(rdata) <= 0 {
|
||||
return
|
||||
}
|
||||
report.Notify(map[string]any{
|
||||
"provider": reportType,
|
||||
"provider_id": strconv.FormatInt(rdata[0]["id"].(int64), 10),
|
||||
"subject": "ALLinSSL 监控通知",
|
||||
"body": fmt.Sprintf(MonitorErrTemplate, v["name"], monitorType, v["target"], strings.Split(checkErr, "：")[0], now.Format("2006-01-02 15:04:05")),
|
||||
})
|
||||
}
|
||||
}
|
||||
monitorErrCountMu.Lock()
|
||||
if MonitorErrCount[id] >= int(repeatSendGap) {
|
||||
MonitorErrCount[id] = 0
|
||||
}
|
||||
monitorErrCountMu.Unlock()
|
||||
} else {
|
||||
// 更新监控记录
|
||||
gs.Where("id=?", []interface{}{id}).Update(map[string]any{
|
||||
"last_time": now.Format("2006-01-02 15:04:05"),
|
||||
"info": certJson,
|
||||
"valid": 1, // 状态为正常
|
||||
})
|
||||
monitorErrCountMu.Lock()
|
||||
MonitorErrCount[id] = 0
|
||||
monitorErrCountMu.Unlock()
|
||||
}
|
||||
}(v)
|
||||
}
|
||||
}
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
95
frontend/allinssl/backend/scheduler/scheduler.go
Normal file
95
frontend/allinssl/backend/scheduler/scheduler.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package scheduler
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// 你的任务列表
|
||||
var funcs = []func(){
|
||||
Monitor,
|
||||
RunWorkflows,
|
||||
}
|
||||
|
||||
// Scheduler 控制器
|
||||
type Scheduler struct {
|
||||
mu sync.Mutex
|
||||
ctx context.Context
|
||||
cancelFunc context.CancelFunc
|
||||
running bool
|
||||
wg sync.WaitGroup
|
||||
}
|
||||
|
||||
// 启动调度器（在 goroutine 中运行）
|
||||
func (s *Scheduler) Start() {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
|
||||
if s.running {
|
||||
return
|
||||
}
|
||||
|
||||
s.ctx, s.cancelFunc = context.WithCancel(context.Background())
|
||||
s.running = true
|
||||
s.wg.Add(1)
|
||||
|
||||
go s.loop() // goroutine 中运行任务调度
|
||||
}
|
||||
|
||||
// 停止调度器
|
||||
func (s *Scheduler) Stop() {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
|
||||
if !s.running {
|
||||
return
|
||||
}
|
||||
|
||||
s.cancelFunc() // 取消上下文
|
||||
s.wg.Wait() // 等待 goroutine 完成退出
|
||||
s.running = false // 标记为未运行
|
||||
}
|
||||
|
||||
// 重启调度器
|
||||
func (s *Scheduler) Restart() {
|
||||
s.Stop()
|
||||
time.Sleep(500 * time.Millisecond) // 可选，避免 race
|
||||
s.Start()
|
||||
}
|
||||
|
||||
// 调度主循环（内部）
|
||||
func (s *Scheduler) loop() {
|
||||
defer s.wg.Done()
|
||||
|
||||
for {
|
||||
// fmt.Println("Scheduler loop")
|
||||
select {
|
||||
case <-s.ctx.Done():
|
||||
return // 外部关闭信号，退出
|
||||
default:
|
||||
start := time.Now()
|
||||
|
||||
var taskWg sync.WaitGroup
|
||||
taskWg.Add(len(funcs))
|
||||
|
||||
for _, f := range funcs {
|
||||
go func(fn func()) {
|
||||
defer taskWg.Done()
|
||||
fn()
|
||||
}(f)
|
||||
}
|
||||
taskWg.Wait()
|
||||
|
||||
// 间隔控制
|
||||
elapsed := time.Since(start)
|
||||
if elapsed < 10*time.Second {
|
||||
select {
|
||||
case <-time.After(10*time.Second - elapsed):
|
||||
case <-s.ctx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
104
frontend/allinssl/backend/scheduler/workflow.go
Normal file
104
frontend/allinssl/backend/scheduler/workflow.go
Normal file
@@ -0,0 +1,104 @@
|
||||
package scheduler
|
||||
|
||||
import (
|
||||
wf "ALLinSSL/backend/internal/workflow"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ExecTime struct {
|
||||
Type string `json:"type"` // "day", "week", "month"
|
||||
Month int `json:"month,omitempty"` // 每月几号 type="month"时必填
|
||||
Week int `json:"week,omitempty"` // 星期几 type="week"时必填
|
||||
Hour int `json:"hour"` // 几点 必填
|
||||
Minute int `json:"minute"` // 几分 必填
|
||||
}
|
||||
|
||||
func RunWorkflows() {
|
||||
s, err := wf.GetSqlite()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
defer s.Close()
|
||||
data, err := s.Select()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
// 遍历工作流
|
||||
var wg sync.WaitGroup
|
||||
for _, workflow := range data {
|
||||
if workflow["exec_type"].(string) != "auto" {
|
||||
// fmt.Println("不是自动执行的工作流")
|
||||
continue
|
||||
}
|
||||
if workflow["active"].(int64) == 0 {
|
||||
// 1: 启用
|
||||
// 0: 禁用
|
||||
// fmt.Println("工作流未启用")
|
||||
continue
|
||||
}
|
||||
if workflow["last_run_status"] != nil && workflow["last_run_status"].(string) == "running" {
|
||||
// fmt.Println("工作流正在运行")
|
||||
continue
|
||||
}
|
||||
// if workflow["last"]
|
||||
if workflow["last_run_time"] != nil && now.Format("2006-01-02 15:04") == workflow["last_run_time"].(string)[0:16] {
|
||||
// fmt.Println("工作流已执行过")
|
||||
continue
|
||||
}
|
||||
// 判断是否到执行时间
|
||||
var execTime ExecTime
|
||||
execTimeStr := ""
|
||||
if et, ok := workflow["exec_time"].(string); ok {
|
||||
execTimeStr = et
|
||||
}
|
||||
err := json.Unmarshal([]byte(execTimeStr), &execTime)
|
||||
if err != nil {
|
||||
// fmt.Println("解析执行时间失败:", err)
|
||||
continue
|
||||
}
|
||||
if execTime.Minute != now.Minute() || execTime.Hour != now.Hour() {
|
||||
// fmt.Println("不在执行时间内1")
|
||||
continue
|
||||
}
|
||||
|
||||
if execTime.Type == "week" && execTime.Week != int(now.Weekday()) {
|
||||
// fmt.Println("不在执行时间内2")
|
||||
continue
|
||||
}
|
||||
if execTime.Type == "month" && execTime.Month != now.Day() {
|
||||
// fmt.Println("不在执行时间内3")
|
||||
continue
|
||||
}
|
||||
if content, ok := workflow["content"].(string); !ok {
|
||||
// fmt.Println("工作流内容为空")
|
||||
continue
|
||||
} else {
|
||||
wg.Add(1)
|
||||
go func(id int64, c string) {
|
||||
defer wg.Done()
|
||||
WorkflowID := strconv.FormatInt(id, 10)
|
||||
RunID, err := wf.AddWorkflowHistory(WorkflowID, "auto")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
ctx := wf.NewExecutionContext(RunID)
|
||||
defer ctx.Logger.Close()
|
||||
err = wf.RunWorkflow(c, ctx)
|
||||
if err != nil {
|
||||
fmt.Println("执行工作流失败:", err)
|
||||
wf.SetWorkflowStatus(WorkflowID, RunID, "fail")
|
||||
} else {
|
||||
wf.SetWorkflowStatus(WorkflowID, RunID, "success")
|
||||
}
|
||||
}(workflow["id"].(int64), content)
|
||||
}
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
Reference in New Issue
Block a user