🐛fix a bug

cron_task/cron.go

@@ -0,0 +1,17 @@
+package cron_task
+
+import (
+	"gitee.ltd/lxh/logger/log"
+	"github.com/go-co-op/gocron/v2"
+	"time"
+)
+
+func StartCronTask() {
+	s, err := gocron.NewScheduler()
+	if err != nil {
+		log.Errorf("初始化定时任务失败:  %v", err.Error())
+		return
+	}
+	_, _ = s.NewJob(gocron.DurationJob(time.Hour), gocron.NewTask(offlineMonitoring)) // 每小时执行一次
+	s.Start()
+}

cron_task/offline_monitoring.go

@@ -0,0 +1,63 @@
+package cron_task
+
+import (
+	"fmt"
+	"gitee.ltd/lxh/logger/log"
+	"strings"
+	"time"
+	"wireguard-dashboard/client"
+	"wireguard-dashboard/repository"
+	"wireguard-dashboard/utils"
+)
+
+// offlineMonitoring
+// @description: 离线监听任务
+func offlineMonitoring() {
+	devices, err := client.WireguardClient.Devices()
+	if err != nil {
+		time.Sleep(5 * time.Minute) // 休眠五分钟再执行
+		offlineMonitoring()
+		return
+	}
+
+	// 遍历客户端数据，并渲染数据信息
+	for _, d := range devices {
+		for _, p := range d.Peers {
+			clientInfo, err := repository.Client().GetByPublicKey(p.PublicKey.String())
+			if err != nil {
+				continue
+			}
+
+			// 没有启用离线监听时，即使客户端已经离线则也不执行
+			if clientInfo.OfflineMonitoring == nil || *clientInfo.OfflineMonitoring != 1 || clientInfo.Email == "" {
+				continue
+			}
+
+			var ipAllocation string
+			for _, iaip := range p.AllowedIPs {
+				ipAllocation += iaip.String() + ","
+			}
+			ipAllocation = strings.TrimRight(ipAllocation, ",")
+			isOnline := time.Since(p.LastHandshakeTime).Minutes() < 3
+
+			// 未离线
+			if isOnline {
+				continue
+			}
+
+			content := fmt.Sprintf("客户端：%s\r\n", clientInfo.Name)
+			content += fmt.Sprintf("客户端IP：%s\r\n", ipAllocation)
+			content += fmt.Sprintf("端点IP：%s\r\n", p.Endpoint.String())
+			content += fmt.Sprintf("最后握手时间：%s\r\n", p.LastHandshakeTime.Format("2006-01-02 15:04:05"))
+
+			// 离线并且配置了邮箱，准备发送邮件
+			err = utils.Mail().SendMail(clientInfo.Email, fmt.Sprintf("客户端[%s]离线通知", clientInfo.Name), content, "")
+			if err != nil {
+				log.Errorf("发送离线通知邮件失败: %v", err.Error())
+				continue
+			}
+
+		}
+	}
+
+}

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/dustin/go-humanize v1.0.1
 	github.com/gin-gonic/gin v1.9.1
 	github.com/glebarez/sqlite v1.11.0
+	github.com/go-co-op/gocron/v2 v2.5.0
 	github.com/go-resty/resty/v2 v2.11.0
 	github.com/golang-jwt/jwt/v5 v5.2.1
 	github.com/google/uuid v1.6.0
@@ -57,6 +58,7 @@ require (
 	github.com/jackc/puddle/v2 v2.2.1 // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
+	github.com/jonboulle/clockwork v0.4.0 // indirect
 	github.com/josharian/native v1.1.0 // indirect
 	github.com/jpillora/backoff v1.0.0 // indirect
 	github.com/klauspost/cpuid/v2 v2.2.4 // indirect
@@ -80,6 +82,7 @@ require (
 	github.com/prometheus/procfs v0.8.0 // indirect
 	github.com/prometheus/prometheus v1.8.2-0.20201028100903-3245b3267b24 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+	github.com/robfig/cron/v3 v3.0.1 // indirect
 	github.com/rogpeppe/go-internal v1.12.0 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.2.11 // indirect
@@ -87,10 +90,11 @@ require (
 	go.uber.org/multierr v1.10.0 // indirect
 	go.uber.org/zap v1.27.0 // indirect
 	golang.org/x/arch v0.3.0 // indirect
+	golang.org/x/exp v0.0.0-20240416160154-fe59bbe5cc7f // indirect
 	golang.org/x/image v0.15.0 // indirect
 	golang.org/x/net v0.21.0 // indirect
 	golang.org/x/oauth2 v0.1.0 // indirect
-	golang.org/x/sync v0.1.0 // indirect
+	golang.org/x/sync v0.7.0 // indirect
 	golang.org/x/sys v0.18.0 // indirect
 	golang.org/x/text v0.14.0 // indirect
 	golang.org/x/time v0.3.0 // indirect

go.sum

@@ -217,6 +217,8 @@ github.com/globalsign/mgo v0.0.0-20180905125535-1ca0a4f7cbcb/go.mod h1:xkRDCp4j0
 github.com/globalsign/mgo v0.0.0-20181015135952-eeefdecb41b8/go.mod h1:xkRDCp4j0OGD1HRkm4kmhM+pmpv3AKq5SU7GMg4oO/Q=
 github.com/glycerine/go-unsnap-stream v0.0.0-20180323001048-9f0cb55181dd/go.mod h1:/20jfyN9Y5QPEAprSgKAUr+glWDY39ZiUEAYOEv5dsE=
 github.com/glycerine/goconvey v0.0.0-20190410193231-58a59202ab31/go.mod h1:Ogl1Tioa0aV7gstGFO7KhffUsb9M4ydbEbbxpcEDc24=
+github.com/go-co-op/gocron/v2 v2.5.0 h1:ff/TJX9GdTJBDL1il9cyd/Sj3WnS+BB7ZzwHKSNL5p8=
+github.com/go-co-op/gocron/v2 v2.5.0/go.mod h1:ckPQw96ZuZLRUGu88vVpd9a6d9HakI14KWahFZtGvNw=
 github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9AVAgeJqvqgH9Q5CA+iKCZ2gyEVpxRU=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20191125211704-12ad95a8df72/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=
 github.com/go-gl/glfw/v3.3/glfw v0.0.0-20200222043503-6f7a984d4dc4/go.mod h1:tQ2UAYgL5IevRw8kRxooKSPJfGvJ9fJQFa0TUsXzTg8=
@@ -498,6 +500,8 @@ github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHW
 github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=
 github.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg=
 github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=
+github.com/jonboulle/clockwork v0.4.0 h1:p4Cf1aMWXnXAUh8lVfewRBx1zaTSYKrKMF2g3ST4RZ4=
+github.com/jonboulle/clockwork v0.4.0/go.mod h1:xgRqUGwRcjKCO1vbZUEtSLrqKoPSsUpK7fnezOII0kc=
 github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible h1:jdpOPRN1zP63Td1hDQbZW73xKmzDvZHzVdNYxhnTMDA=
 github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible/go.mod h1:1c7szIrayyPPB/987hsnvNzLushdWf4o/79s3P08L8A=
 github.com/josharian/native v1.1.0 h1:uuaP0hAbW7Y4l0ZRQ6C9zfb7Mg1mbFKry/xzDAfmtLA=
@@ -737,6 +741,8 @@ github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0/go.mod h1:qq
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
 github.com/retailnext/hllpp v1.0.1-0.20180308014038-101a6d2f8b52/go.mod h1:RDpi1RftBQPUCDRw6SmxeaREsAaRKnOclghuzp/WRzc=
+github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
+github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
 github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
 github.com/rogpeppe/fastuuid v1.2.0/go.mod h1:jVj6XXZzXRy/MSR5jhDC/2q6DgLz+nrA6LYCDYWNEvQ=
 github.com/rogpeppe/go-internal v1.1.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=
@@ -799,8 +805,9 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
-github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
 github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
 github.com/tinylib/msgp v1.0.2/go.mod h1:+d+yLhGm8mzTaHzB+wgMYrodPfmZrzkirds8fDWklFE=
 github.com/tmc/grpc-websocket-proxy v0.0.0-20170815181823-89b8d40f7ca8/go.mod h1:ncp9v5uamzpCO7NfCPTXjqaC+bZgJeR0sMTm6dMHP7U=
@@ -894,6 +901,8 @@ golang.org/x/exp v0.0.0-20191227195350-da58074b4299/go.mod h1:2RIsYlXP63K8oxa1u0
 golang.org/x/exp v0.0.0-20200119233911-0405dc783f0a/go.mod h1:2RIsYlXP63K8oxa1u096TMicItID8zy7Y6sNkU49FU4=
 golang.org/x/exp v0.0.0-20200207192155-f17229e696bd/go.mod h1:J/WKrq2StrnmMY6+EHIKF9dgMWnmCNThgcyBT1FY9mM=
 golang.org/x/exp v0.0.0-20200224162631-6cc2880d07d6/go.mod h1:3jZMyOhIsHpP37uCMkUooju7aAi5cS1Q23tOzKc+0MU=
+golang.org/x/exp v0.0.0-20240416160154-fe59bbe5cc7f h1:99ci1mjWVBWwJiEKYY6jWa4d2nTQVIEhZIptnrVb1XY=
+golang.org/x/exp v0.0.0-20240416160154-fe59bbe5cc7f/go.mod h1:/lliqkxwWAhPjf5oSOIJup2XcqJaw8RGS6k3TGEc7GI=
 golang.org/x/image v0.0.0-20180708004352-c73c2afc3b81/go.mod h1:ux5Hcp/YLpHSI86hEcLt0YII63i6oz57MZXIpbrjZUs=
 golang.org/x/image v0.0.0-20190227222117-0694c2d4d067/go.mod h1:kZ7UVZpmo3dzQBMxlp+ypCbDeSB+sBbTgSJuh5dn5js=
 golang.org/x/image v0.0.0-20190802002840-cff245a6509b/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
@@ -994,8 +1003,9 @@ golang.org/x/sync v0.0.0-20200930132711-30421366ff76/go.mod h1:RxMgew5VJxzue5/jJ
 golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.7.0 h1:YsImfSBoP9QPYL0xyKJPq0gcaJdG3rInoqxTWbfQu9M=
+golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20180823144017-11551d06cbcc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=

initialize/init.go

@@ -12,8 +12,10 @@ import (
 	"gorm.io/driver/mysql"
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
+	gl "gorm.io/gorm/logger"
 	"log"
 	"os"
+	"time"
 	"wireguard-dashboard/client"
 	"wireguard-dashboard/config"
 )
@@ -69,8 +71,18 @@ func initDatabase() {
 		dbDialector = sqlite.Open(config.Config.Database.GetDSN())
 	}
 
+	logLevel := gl.Info
+	if os.Getenv("GIN_MODE") == "release" {
+		logLevel = gl.Error
+	}
+
 	db, err := gorm.Open(dbDialector, &gorm.Config{
-		Logger: logger.DefaultGormLogger(),
+		Logger: logger.NewGormLoggerWithConfig(gl.Config{
+			SlowThreshold:             time.Second, // Slow SQL threshold
+			IgnoreRecordNotFoundError: false,       // 忽略没找到结果的错误
+			LogLevel:                  logLevel,    // Log level
+			Colorful:                  false,       // Disable color
+		}),
 	})
 
 	if err != nil {

main.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"time"
 	"wireguard-dashboard/config"
+	"wireguard-dashboard/cron_task"
 	"wireguard-dashboard/initialize"
 	"wireguard-dashboard/queues"
 	"wireguard-dashboard/route"
@@ -18,7 +19,8 @@ func init() {
 	if err := script.NewScript().Do(); err != nil {
 		log.Errorf("执行脚本失败: %v", err.Error())
 	}
-	go queues.StartConsumer() // 启动队列
+	go queues.StartConsumer()    // 启动队列
+	go cron_task.StartCronTask() // 启动定时任务
 }
 
 func main() {

queues/consumer.go

@@ -6,5 +6,5 @@ func StartConsumer() {
 	// 同步配置文件
 	go asyncWireguardConfigFile()
 	// 离线监听
-	go offlineMonitoring()
+	//go offlineMonitoring()
 }

queues/offline_monitoring.go

@@ -1,65 +0,0 @@
-package queues
-
-import (
-	"fmt"
-	"gitee.ltd/lxh/logger/log"
-	"strings"
-	"time"
-	"wireguard-dashboard/client"
-	"wireguard-dashboard/repository"
-	"wireguard-dashboard/utils"
-)
-
-// offlineMonitoring
-// @description: 离线监听任务
-func offlineMonitoring() {
-	for {
-		devices, err := client.WireguardClient.Devices()
-		if err != nil {
-			time.Sleep(5 * time.Minute) // 休眠五分钟再执行
-			offlineMonitoring()
-			continue
-		}
-
-		// 遍历客户端数据，并渲染数据信息
-		for _, d := range devices {
-			for _, p := range d.Peers {
-				clientInfo, err := repository.Client().GetByPublicKey(p.PublicKey.String())
-				if err != nil {
-					continue
-				}
-
-				// 没有启用离线监听时，即使客户端已经离线则也不执行
-				if clientInfo.OfflineMonitoring == nil || *clientInfo.OfflineMonitoring != 1 || clientInfo.Email == "" {
-					continue
-				}
-
-				var ipAllocation string
-				for _, iaip := range p.AllowedIPs {
-					ipAllocation += iaip.String() + ","
-				}
-				ipAllocation = strings.TrimRight(ipAllocation, ",")
-				isOnline := time.Since(p.LastHandshakeTime).Minutes() < 3
-
-				// 未离线
-				if !isOnline {
-					continue
-				}
-
-				content := fmt.Sprintf("客户端：%s\r\n", clientInfo.Name)
-				content += fmt.Sprintf("客户端IP：%s\r\n", ipAllocation)
-				content += fmt.Sprintf("端点IP：%s", p.Endpoint.String())
-				content += fmt.Sprintf("最后握手时间：%s\r\n", p.LastHandshakeTime.Format("2006-01-02 15:04:05"))
-				content += fmt.Sprintf("离线时间：%s\r\n", time.Now().Format("2006-01-02 15:04:05"))
-
-				// 离线并且配置了邮箱，准备发送邮件
-				err = utils.Mail().SendMail(clientInfo.Email, fmt.Sprintf("客户端离线通知"), content, "")
-				if err != nil {
-					log.Errorf("发送离线通知邮件失败: %v", err.Error())
-					continue
-				}
-
-			}
-		}
-	}
-}