Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-04 04:30:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: clean code

Browse Source
This commit is contained in:
naiba
2025-03-31 19:52:23 +08:00
parent 67c129635e
commit b69eac4310
3 changed files with 32 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
service/singleton/server.go
View File

@@ -18,19 +18,14 @@ type ServerClass struct {
uuidToID map[string]uint64 uuidToID map[string]uint64
sortedListForGuest []*model.Server sortedListForGuest []*model.Server
conf *ConfigClass
dc *DDNSClass
} }
func NewServerClass(conf *ConfigClass, dc *DDNSClass) *ServerClass { func NewServerClass() *ServerClass {
sc := &ServerClass{ sc := &ServerClass{
class: class[uint64, *model.Server]{ class: class[uint64, *model.Server]{
list: make(map[uint64]*model.Server), list: make(map[uint64]*model.Server),
}, },
uuidToID: make(map[string]uint64), uuidToID: make(map[string]uint64),
conf: conf,
dc: dc,
} }
var servers []model.Server var servers []model.Server
@@ -95,10 +90,10 @@ func (c *ServerClass) UUIDToID(uuid string) (id uint64, ok bool) {
} }
func (c *ServerClass) UpdateDDNS(server *model.Server, ip *model.IP) error { func (c *ServerClass) UpdateDDNS(server *model.Server, ip *model.IP) error {
confServers := strings.Split(c.conf.DNSServers, ",") confServers := strings.Split(Conf.DNSServers, ",")
ctx := context.WithValue(context.Background(), ddns.DNSServerKey{}, utils.IfOr(confServers[0] != "", confServers, utils.DNSServers)) ctx := context.WithValue(context.Background(), ddns.DNSServerKey{}, utils.IfOr(confServers[0] != "", confServers, utils.DNSServers))
providers, err := c.dc.GetDDNSProvidersFromProfiles(server.DDNSProfiles, utils.IfOr(ip != nil, ip, &server.GeoIP.IP)) providers, err := DDNSShared.GetDDNSProvidersFromProfiles(server.DDNSProfiles, utils.IfOr(ip != nil, ip, &server.GeoIP.IP))
if err != nil { if err != nil {
return err return err
} }

55
service/singleton/servicesentinel.go
View File

@@ -83,15 +83,10 @@ type ServiceSentinel struct {
// 30天数据缓存 // 30天数据缓存
monthlyStatusLock sync.Mutex monthlyStatusLock sync.Mutex
monthlyStatus map[uint64]*serviceResponseItem monthlyStatus map[uint64]*serviceResponseItem
// references
serverc *ServerClass
notificationc *NotificationClass
crc *CronClass
} }
// NewServiceSentinel 创建服务监控器 // NewServiceSentinel 创建服务监控器
func NewServiceSentinel(serviceSentinelDispatchBus chan<- *model.Service, sc *ServerClass, nc *NotificationClass, crc *CronClass) (*ServiceSentinel, error) { func NewServiceSentinel(serviceSentinelDispatchBus chan<- *model.Service) (*ServiceSentinel, error) {
ss := &ServiceSentinel{ ss := &ServiceSentinel{
serviceReportChannel: make(chan ReportData, 200), serviceReportChannel: make(chan ReportData, 200),
serviceStatusToday: make(map[uint64]*_TodayStatsOfService), serviceStatusToday: make(map[uint64]*_TodayStatsOfService),
@@ -103,10 +98,6 @@ func NewServiceSentinel(serviceSentinelDispatchBus chan<- *model.Service, sc *Se
// 30天数据缓存 // 30天数据缓存
monthlyStatus: make(map[uint64]*serviceResponseItem), monthlyStatus: make(map[uint64]*serviceResponseItem),
dispatchBus: serviceSentinelDispatchBus, dispatchBus: serviceSentinelDispatchBus,
serverc: sc,
notificationc: nc,
crc: crc,
} }
// 加载历史记录 // 加载历史记录
@@ -139,7 +130,7 @@ func NewServiceSentinel(serviceSentinelDispatchBus chan<- *model.Service, sc *Se
go ss.worker() go ss.worker()
// 每日将游标往后推一天 // 每日将游标往后推一天
_, err = crc.AddFunc("0 0 0 * * *", ss.refreshMonthlyServiceStatus) _, err = CronShared.AddFunc("0 0 0 * * *", ss.refreshMonthlyServiceStatus)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@@ -204,7 +195,7 @@ func (ss *ServiceSentinel) loadServiceHistory() error {
for _, service := range services { for _, service := range services {
task := service task := service
// 通过cron定时将服务监控任务传递给任务调度管道 // 通过cron定时将服务监控任务传递给任务调度管道
service.CronJobID, err = ss.crc.AddFunc(task.CronSpec(), func() { service.CronJobID, err = CronShared.AddFunc(task.CronSpec(), func() {
ss.dispatchBus <- task ss.dispatchBus <- task
}) })
if err != nil { if err != nil {
@@ -261,7 +252,7 @@ func (ss *ServiceSentinel) Update(m *model.Service) error {
var err error var err error
// 写入新任务 // 写入新任务
m.CronJobID, err = ss.crc.AddFunc(m.CronSpec(), func() { m.CronJobID, err = CronShared.AddFunc(m.CronSpec(), func() {
ss.dispatchBus <- m ss.dispatchBus <- m
}) })
if err != nil { if err != nil {
@@ -269,7 +260,7 @@ func (ss *ServiceSentinel) Update(m *model.Service) error {
} }
if ss.services[m.ID] != nil { if ss.services[m.ID] != nil {
// 停掉旧任务 // 停掉旧任务
ss.crc.Remove(ss.services[m.ID].CronJobID) CronShared.Remove(ss.services[m.ID].CronJobID)
} else { } else {
// 新任务初始化数据 // 新任务初始化数据
ss.monthlyStatus[m.ID] = &serviceResponseItem{ ss.monthlyStatus[m.ID] = &serviceResponseItem{
@@ -306,7 +297,7 @@ func (ss *ServiceSentinel) Delete(ids []uint64) {
delete(ss.serviceStatusToday, id) delete(ss.serviceStatusToday, id)
// 停掉定时任务 // 停掉定时任务
ss.crc.Remove(ss.services[id].CronJobID) CronShared.Remove(ss.services[id].CronJobID)
delete(ss.services, id) delete(ss.services, id)
delete(ss.monthlyStatus, id) delete(ss.monthlyStatus, id)
@@ -510,10 +501,10 @@ func (ss *ServiceSentinel) worker() {
} }
cs, _ := ss.Get(mh.GetId()) cs, _ := ss.Get(mh.GetId())
m := ss.serverc.GetList() m := ServerShared.GetList()
// 延迟报警 // 延迟报警
if mh.Delay > 0 { if mh.Delay > 0 {
delayCheck(&r, ss.notificationc, m, cs, mh) delayCheck(&r, m, cs, mh)
} }
// 状态变更报警+触发任务执行 // 状态变更报警+触发任务执行
@@ -522,7 +513,7 @@ func (ss *ServiceSentinel) worker() {
// 存储新的状态值 // 存储新的状态值
ss.serviceCurrentStatusData[mh.GetId()].lastStatus = stateCode ss.serviceCurrentStatusData[mh.GetId()].lastStatus = stateCode
notifyCheck(&r, ss.notificationc, ss.crc, m, cs, mh, lastStatus, stateCode) notifyCheck(&r, m, cs, mh, lastStatus, stateCode)
} }
ss.serviceResponseDataStoreLock.Unlock() ss.serviceResponseDataStoreLock.Unlock()
@@ -536,12 +527,12 @@ func (ss *ServiceSentinel) worker() {
errMsg = mh.Data errMsg = mh.Data
if cs.Notify { if cs.Notify {
muteLabel := NotificationMuteLabel.ServiceTLS(mh.GetId(), "network") muteLabel := NotificationMuteLabel.ServiceTLS(mh.GetId(), "network")
go ss.notificationc.SendNotification(cs.NotificationGroupID, Localizer.Tf("[TLS] Fetch cert info failed, Reporter: %s, Error: %s", cs.Name, errMsg), muteLabel) go NotificationShared.SendNotification(cs.NotificationGroupID, Localizer.Tf("[TLS] Fetch cert info failed, Reporter: %s, Error: %s", cs.Name, errMsg), muteLabel)
} }
} }
} else { } else {
// 清除网络错误静音缓存 // 清除网络错误静音缓存
ss.notificationc.UnMuteNotification(cs.NotificationGroupID, NotificationMuteLabel.ServiceTLS(mh.GetId(), "network")) NotificationShared.UnMuteNotification(cs.NotificationGroupID, NotificationMuteLabel.ServiceTLS(mh.GetId(), "network"))
var newCert = strings.Split(mh.Data, "|") var newCert = strings.Split(mh.Data, "|")
if len(newCert) > 1 { if len(newCert) > 1 {
@@ -579,7 +570,7 @@ func (ss *ServiceSentinel) worker() {
// 静音规则： 服务id+证书过期时间 // 静音规则： 服务id+证书过期时间
// 用于避免多个监测点对相同证书同时报警 // 用于避免多个监测点对相同证书同时报警
muteLabel := NotificationMuteLabel.ServiceTLS(mh.GetId(), fmt.Sprintf("expire_%s", expiresTimeStr)) muteLabel := NotificationMuteLabel.ServiceTLS(mh.GetId(), fmt.Sprintf("expire_%s", expiresTimeStr))
go ss.notificationc.SendNotification(notificationGroupID, fmt.Sprintf("[TLS] %s %s", serviceName, errMsg), muteLabel) go NotificationShared.SendNotification(notificationGroupID, fmt.Sprintf("[TLS] %s %s", serviceName, errMsg), muteLabel)
} }
// 证书变更提醒 // 证书变更提醒
@@ -589,7 +580,7 @@ func (ss *ServiceSentinel) worker() {
oldCert[0], expiresOld.Format("2006-01-02 15:04:05"), newCert[0], expiresNew.Format("2006-01-02 15:04:05")) oldCert[0], expiresOld.Format("2006-01-02 15:04:05"), newCert[0], expiresNew.Format("2006-01-02 15:04:05"))
// 证书变更后会自动更新缓存，所以不需要静音 // 证书变更后会自动更新缓存，所以不需要静音
go ss.notificationc.SendNotification(notificationGroupID, fmt.Sprintf("[TLS] %s %s", serviceName, errMsg), "") go NotificationShared.SendNotification(notificationGroupID, fmt.Sprintf("[TLS] %s %s", serviceName, errMsg), "")
} }
} }
} }
@@ -597,7 +588,7 @@ func (ss *ServiceSentinel) worker() {
} }
} }
func delayCheck(r *ReportData, nc *NotificationClass, m map[uint64]*model.Server, ss *model.Service, mh *pb.TaskResult) { func delayCheck(r *ReportData, m map[uint64]*model.Server, ss *model.Service, mh *pb.TaskResult) {
if !ss.LatencyNotify { if !ss.LatencyNotify {
return return
} }
@@ -609,20 +600,20 @@ func delayCheck(r *ReportData, nc *NotificationClass, m map[uint64]*model.Server
// 延迟超过最大值 // 延迟超过最大值
reporterServer := m[r.Reporter] reporterServer := m[r.Reporter]
msg := Localizer.Tf("[Latency] %s %2f > %2f, Reporter: %s", ss.Name, mh.Delay, ss.MaxLatency, reporterServer.Name) msg := Localizer.Tf("[Latency] %s %2f > %2f, Reporter: %s", ss.Name, mh.Delay, ss.MaxLatency, reporterServer.Name)
go nc.SendNotification(notificationGroupID, msg, minMuteLabel) go NotificationShared.SendNotification(notificationGroupID, msg, minMuteLabel)
} else if mh.Delay < ss.MinLatency { } else if mh.Delay < ss.MinLatency {
// 延迟低于最小值 // 延迟低于最小值
reporterServer := m[r.Reporter] reporterServer := m[r.Reporter]
msg := Localizer.Tf("[Latency] %s %2f < %2f, Reporter: %s", ss.Name, mh.Delay, ss.MinLatency, reporterServer.Name) msg := Localizer.Tf("[Latency] %s %2f < %2f, Reporter: %s", ss.Name, mh.Delay, ss.MinLatency, reporterServer.Name)
go nc.SendNotification(notificationGroupID, msg, maxMuteLabel) go NotificationShared.SendNotification(notificationGroupID, msg, maxMuteLabel)
} else { } else {
// 正常延迟， 清除静音缓存 // 正常延迟， 清除静音缓存
nc.UnMuteNotification(notificationGroupID, minMuteLabel) NotificationShared.UnMuteNotification(notificationGroupID, minMuteLabel)
nc.UnMuteNotification(notificationGroupID, maxMuteLabel) NotificationShared.UnMuteNotification(notificationGroupID, maxMuteLabel)
} }
} }
func notifyCheck(r *ReportData, nc *NotificationClass, crc *CronClass, m map[uint64]*model.Server, func notifyCheck(r *ReportData, m map[uint64]*model.Server,
ss *model.Service, mh *pb.TaskResult, lastStatus, stateCode uint8) { ss *model.Service, mh *pb.TaskResult, lastStatus, stateCode uint8) {
// 判断是否需要发送通知 // 判断是否需要发送通知
isNeedSendNotification := ss.Notify && (lastStatus != 0 || stateCode == StatusDown) isNeedSendNotification := ss.Notify && (lastStatus != 0 || stateCode == StatusDown)
@@ -634,10 +625,10 @@ func notifyCheck(r *ReportData, nc *NotificationClass, crc *CronClass, m map[uin
// 状态变更时，清除静音缓存 // 状态变更时，清除静音缓存
if stateCode != lastStatus { if stateCode != lastStatus {
nc.UnMuteNotification(notificationGroupID, muteLabel) NotificationShared.UnMuteNotification(notificationGroupID, muteLabel)
} }
go nc.SendNotification(notificationGroupID, notificationMsg, muteLabel) go NotificationShared.SendNotification(notificationGroupID, notificationMsg, muteLabel)
} }
// 判断是否需要触发任务 // 判断是否需要触发任务
@@ -646,10 +637,10 @@ func notifyCheck(r *ReportData, nc *NotificationClass, crc *CronClass, m map[uin
reporterServer := m[r.Reporter] reporterServer := m[r.Reporter]
if stateCode == StatusGood && lastStatus != stateCode { if stateCode == StatusGood && lastStatus != stateCode {
// 当前状态正常 前序状态非正常时 触发恢复任务 // 当前状态正常 前序状态非正常时 触发恢复任务
go crc.SendTriggerTasks(ss.RecoverTriggerTasks, reporterServer.ID) go CronShared.SendTriggerTasks(ss.RecoverTriggerTasks, reporterServer.ID)
} else if lastStatus == StatusGood && lastStatus != stateCode { } else if lastStatus == StatusGood && lastStatus != stateCode {
// 前序状态正常 当前状态非正常时 触发失败任务 // 前序状态正常 当前状态非正常时 触发失败任务
go crc.SendTriggerTasks(ss.FailTriggerTasks, reporterServer.ID) go CronShared.SendTriggerTasks(ss.FailTriggerTasks, reporterServer.ID)
} }
} }
} }

11
service/singleton/singleton.go
View File

@@ -52,14 +52,15 @@ func InitTimezoneAndCache() error {
// LoadSingleton 加载子服务并执行 // LoadSingleton 加载子服务并执行
func LoadSingleton(bus chan<- *model.Service) (err error) { func LoadSingleton(bus chan<- *model.Service) (err error) {
initUser() // 加载用户ID绑定表
initI18n() // 加载本地化服务 initI18n() // 加载本地化服务
initUser() // 加载用户ID绑定表
NATShared = NewNATClass() NATShared = NewNATClass()
DDNSShared = NewDDNSClass() DDNSShared = NewDDNSClass()
NotificationShared = NewNotificationClass() // 加载通知服务 NotificationShared = NewNotificationClass()
ServerShared = NewServerClass(Conf, DDNSShared) // 加载服务器列表 ServerShared = NewServerClass()
CronShared = NewCronClass() // 加载定时任务 CronShared = NewCronClass()
ServiceSentinelShared, err = NewServiceSentinel(bus, ServerShared, NotificationShared, CronShared) // 最后初始化 ServiceSentinel
ServiceSentinelShared, err = NewServiceSentinel(bus)
return return
} }