Merge pull request #31 from opsre:feat_custom_cron

✨ feat(cron): 新增定时任务配置功能

Author: eryajf

config.example.yaml

@@ -1,12 +1,18 @@
+cron_config:
+  # ↓↓↓ -------------------------- 1. 域名记录同步间隔（秒），默认30秒
+  domain_record_sync_interval: 30
+  # ↓↓↓ -------------------------- 2. 证书信息同步间隔（秒），默认3600秒（1小时）
+  cert_info_sync_interval: 3600
+# ↓↓↓ ---------------------------- 3. 自定义监控的域名记录列表，如果没有可留空
 custom_records:
   - "www.baidu.com"
   - "wiki.eryajf.net"
 cloud_providers:
-  # ↓↓↓ -------------------------- 1. DNS提供商Tencent，请勿更改此行，如无需腾讯云的配置，可删除此段配置至 aliyun，该字段会作为标签注入到指标中
+  # ↓↓↓ -------------------------- 4. DNS提供商Tencent，请勿更改此行，如无需腾讯云的配置，可删除此段配置至 aliyun，该字段会作为标签注入到指标中
   tencent:
-    # ↓↓↓ ------------------------ 2. 固定配置，无需更改
+    # ↓↓↓ ------------------------ 5. 固定配置，无需更改
     accounts:
-      # ↓↓↓ ---------------------- 3. 腾讯云账号配置，可配置多个，给每个账号取个名字，如 t1、t2，该字段会作为标签注入到指标中
+      # ↓↓↓ ---------------------- 6. 腾讯云账号配置，可配置多个，给每个账号取个名字，如 t1、t2，该字段会作为标签注入到指标中
       - name: t1
         secretId: "xxxxx"
         secretKey: "xxxxx"

docker-compose.yml

@@ -35,6 +35,10 @@ services:
 
 # 示例配置
 configs:
+  # 定时任务配置（可选，不配置则使用默认值）
+  cron_config:
+    domain_record_sync_interval: 30   # 域名记录同步间隔（秒），默认30秒
+    cert_info_sync_interval: 3600    # 证书信息同步间隔（秒），默认3600秒（1小时）
   # 云解析配置,以阿里云DNS为例
   cloud_dns_exporter_config:
     content: |

go.mod

@@ -79,5 +79,4 @@ require (
 	golang.org/x/sys v0.23.0 // indirect
 	google.golang.org/protobuf v1.33.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0
-
 )

pkg/export/cron_cache.go

@@ -15,22 +15,51 @@ import (
 
 // InitCron 初始化定时任务
 func InitCron() {
+	cronConfig := public.GetCronConfig()
 	c := cron.New(cron.WithSeconds())
-	_, _ = c.AddFunc("*/30 * * * * *", func() {
-		loading()
+
+	// 域名记录同步任务
+	domainRecordSyncSpec := convertSecondsToCron(cronConfig.DomainRecordSyncInterval)
+	_, _ = c.AddFunc(domainRecordSyncSpec, func() {
+		loadingDomainRecord()
 	})
-	loading()
-	_, _ = c.AddFunc("03 03 03 * * *", func() {
+	loadingDomainRecord() // 启动时立即执行一次
+
+	// 证书信息同步任务
+	certInfoSyncSpec := convertSecondsToCron(cronConfig.CertInfoSyncInterval)
+	_, _ = c.AddFunc(certInfoSyncSpec, func() {
 		loadingCert()
 		loadingCustomRecordCert()
 	})
-	loadingCert()
+	loadingCert() // 启动时立即执行一次
 	loadingCustomRecordCert()
 
 	c.Start()
+
+	logger.Info(fmt.Sprintf("Cron initialized - domain record sync: every %d seconds, cert info sync: every %d seconds",
+		cronConfig.DomainRecordSyncInterval, cronConfig.CertInfoSyncInterval))
+}
+
+// convertSecondsToCron 将秒数转换为 cron 表达式
+func convertSecondsToCron(seconds int) string {
+	if seconds < 60 {
+		// 小于60秒：每N秒执行一次
+		return fmt.Sprintf("*/%d * * * * *", seconds)
+	} else if seconds < 3600 {
+		// 小于1小时：每N分钟执行一次
+		minutes := seconds / 60
+		return fmt.Sprintf("0 */%d * * * *", minutes)
+	} else if seconds < 86400 {
+		// 小于1天：每N小时执行一次
+		hours := seconds / 3600
+		return fmt.Sprintf("0 0 */%d * * *", hours)
+	} else {
+		// 大于等于1天：每天凌晨3点执行
+		return "0 0 3 * * *"
+	}
 }
 
-func loading() {
+func loadingDomainRecord() {
 	var wg sync.WaitGroup
 	var mu sync.Mutex
 

public/public.go

@@ -33,6 +33,9 @@ const (
 	DomainList     string = "domain_list"
 	RecordList     string = "record_list"
 	RecordCertInfo string = "record_cert_info"
+	// Cron Config Defaults
+	DefaultDomainRecordSyncInterval = 30   // 默认域名记录同步间隔（秒）
+	DefaultCertInfoSyncInterval     = 3600 // 默认证书信息同步间隔（秒），24小时
 )
 
 var (
@@ -49,9 +52,16 @@ type Account struct {
 	SecretKey     string `yaml:"secretKey"`
 }
 
+// CronConfig 定时任务配置
+type CronConfig struct {
+	DomainRecordSyncInterval int `yaml:"domain_record_sync_interval"` // 域名记录同步间隔（秒）
+	CertInfoSyncInterval     int `yaml:"cert_info_sync_interval"`     // 证书信息同步间隔（秒）
+}
+
 // Config 表示配置文件的结构
 type Configuration struct {
-	CustomRecords  []string `yaml:"custom_records"`
+	CronConfig     *CronConfig `yaml:"cron_config"`
+	CustomRecords  []string    `yaml:"custom_records"`
 	CloudProviders map[string]struct {
 		Accounts []map[string]string `yaml:"accounts"`
 	} `yaml:"cloud_providers"`
@@ -90,3 +100,24 @@ func InitCache() {
 func GetID() string {
 	return xid.New().String()
 }
+
+// GetCronConfig 获取定时任务配置，如果未配置则返回默认值
+func GetCronConfig() CronConfig {
+	if Config.CronConfig == nil {
+		return CronConfig{
+			DomainRecordSyncInterval: DefaultDomainRecordSyncInterval,
+			CertInfoSyncInterval:     DefaultCertInfoSyncInterval,
+		}
+	}
+
+	config := *Config.CronConfig
+	// 如果配置了但值为0或负数，使用默认值
+	if config.DomainRecordSyncInterval <= 0 {
+		config.DomainRecordSyncInterval = DefaultDomainRecordSyncInterval
+	}
+	if config.CertInfoSyncInterval <= 0 {
+		config.CertInfoSyncInterval = DefaultCertInfoSyncInterval
+	}
+
+	return config
+}