Compare commits

..

4 Commits

Author SHA1 Message Date
cbe1188e51 🎨恢复日志,优化监听 2024-06-14 17:23:38 +08:00
1bcad97ed2 🐛fix a bug 2024-06-14 13:43:40 +08:00
c2a9618f27 🎨恢复定时任务 2024-06-14 10:25:53 +08:00
8dab976cba 🐛修复CPU负载过高 2024-06-14 08:45:16 +08:00
2 changed files with 46 additions and 48 deletions

View File

@@ -13,17 +13,28 @@ import (
// offlineMonitoring // offlineMonitoring
// @description: 离线监听任务 // @description: 离线监听任务
func offlineMonitoring() { func offlineMonitoring() {
for {
devices, err := client.WireguardClient.Devices() devices, err := client.WireguardClient.Devices()
if err != nil { if err != nil {
time.Sleep(5 * time.Minute) // 休眠五分钟再执行 time.Sleep(5 * time.Minute) // 休眠五分钟再执行
offlineMonitoring() offlineMonitoring()
continue return
} }
// 遍历客户端数据,并渲染数据信息 // 遍历客户端数据,并渲染数据信息
for _, d := range devices { for _, d := range devices {
for _, p := range d.Peers { for _, p := range d.Peers {
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
}
clientInfo, err := repository.Client().GetByPublicKey(p.PublicKey.String()) clientInfo, err := repository.Client().GetByPublicKey(p.PublicKey.String())
if err != nil { if err != nil {
continue continue
@@ -34,26 +45,13 @@ func offlineMonitoring() {
continue 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("客户端:%s\r\n", clientInfo.Name)
content += fmt.Sprintf("客户端IP%s\r\n", ipAllocation) content += fmt.Sprintf("客户端IP%s\r\n", ipAllocation)
content += fmt.Sprintf("端点IP%s", p.Endpoint.String()) 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")) 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, "") err = utils.Mail().SendMail(clientInfo.Email, fmt.Sprintf("客户端[%s]离线通知", clientInfo.Name), content, "")
if err != nil { if err != nil {
log.Errorf("发送离线通知邮件失败: %v", err.Error()) log.Errorf("发送离线通知邮件失败: %v", err.Error())
continue continue
@@ -61,5 +59,5 @@ func offlineMonitoring() {
} }
} }
}
} }

View File

@@ -72,9 +72,9 @@ func initDatabase() {
} }
logLevel := gl.Info logLevel := gl.Info
if os.Getenv("GIN_MODE") == "release" { //if os.Getenv("GIN_MODE") == "release" {
logLevel = gl.Error // logLevel = gl.Error
} //}
db, err := gorm.Open(dbDialector, &gorm.Config{ db, err := gorm.Open(dbDialector, &gorm.Config{
Logger: logger.NewGormLoggerWithConfig(gl.Config{ Logger: logger.NewGormLoggerWithConfig(gl.Config{