83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package watch
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/codfrm/cago"
|
|
"github.com/codfrm/cago/configs"
|
|
"github.com/codfrm/cago/pkg/logger"
|
|
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
|
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
|
|
dnspod "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/dnspod/v20210323"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type LoadBalance struct {
|
|
Value string // 记录值
|
|
Line string // 线路
|
|
}
|
|
|
|
type CheckDomain struct {
|
|
Domain string // 域名
|
|
Name string // 记录名
|
|
Value string // 记录值
|
|
Line string // 线路
|
|
LoadBalance *LoadBalance `yaml:"loadBalance"` // 负载均衡
|
|
}
|
|
|
|
type Config struct {
|
|
SecretID string `yaml:"secretID"` // 账号 secret id
|
|
SecretKey string `yaml:"secretKey"` // 账号 secret key
|
|
CheckDomain []*CheckDomain `yaml:"checkDomain"`
|
|
}
|
|
|
|
type watch struct {
|
|
config *Config
|
|
credential *common.Credential
|
|
client *dnspod.Client
|
|
isDisable bool
|
|
}
|
|
|
|
func Watch() cago.Component {
|
|
return &watch{
|
|
isDisable: false,
|
|
}
|
|
}
|
|
|
|
func (w *watch) Start(ctx context.Context, cfg *configs.Config) error {
|
|
config := &Config{}
|
|
if err := cfg.Scan("watch", config); err != nil {
|
|
return err
|
|
}
|
|
// 获取记录列表
|
|
w.config = config
|
|
w.credential = common.NewCredential(
|
|
w.config.SecretID,
|
|
w.config.SecretKey,
|
|
)
|
|
cpf := profile.NewClientProfile()
|
|
cpf.HttpProfile.Endpoint = "dnspod.tencentcloudapi.com"
|
|
var err error
|
|
w.client, err = dnspod.NewClient(w.credential, "", cpf)
|
|
if err != nil {
|
|
logger.Default().Error("NewClient error", zap.Error(err))
|
|
return err
|
|
}
|
|
for _, c := range w.config.CheckDomain {
|
|
r, err := w.queryRecord(c.Domain, c.Name, c.Value, c.Line)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
record, err := newRecord(w, r, c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go record.watch(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *watch) CloseHandle() {
|
|
|
|
}
|