63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
package watch
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/codfrm/cago/pkg/logger"
|
|
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
|
dnspod "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/dnspod/v20210323"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func (w *watch) queryRecord(domain, name, value, line string) (*dnspod.RecordListItem, error) {
|
|
// 实例化一个请求对象,每个接口都会对应一个request对象
|
|
request := dnspod.NewDescribeRecordListRequest()
|
|
|
|
request.Domain = common.StringPtr(domain)
|
|
|
|
response, err := w.client.DescribeRecordList(request)
|
|
if err != nil {
|
|
logger.Default().Error("DescribeRecordList error", zap.Error(err))
|
|
return nil, err
|
|
}
|
|
for _, v := range response.Response.RecordList {
|
|
if *v.Name == name && *v.Value == value {
|
|
if line != "" && *v.Line != line {
|
|
continue
|
|
}
|
|
logger.Default().Info("record found", zap.Any("record", v))
|
|
return v, nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("[%s] %s %s %s record not found", domain, name, value, line)
|
|
}
|
|
|
|
func (w *watch) enable(ctx context.Context, domain string, recordId uint64) error {
|
|
// 开启记录
|
|
request := dnspod.NewModifyRecordStatusRequest()
|
|
request.SetContext(ctx)
|
|
request.Domain = common.StringPtr(domain)
|
|
request.RecordId = common.Uint64Ptr(recordId)
|
|
request.Status = common.StringPtr("ENABLE")
|
|
_, err := w.client.ModifyRecordStatus(request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *watch) disable(ctx context.Context, domain string, recordId uint64) error {
|
|
// 开启记录
|
|
request := dnspod.NewModifyRecordStatusRequest()
|
|
request.SetContext(ctx)
|
|
request.Domain = common.StringPtr(domain)
|
|
request.RecordId = common.Uint64Ptr(recordId)
|
|
request.Status = common.StringPtr("DISABLE")
|
|
_, err := w.client.ModifyRecordStatus(request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|