EdgeAPI/internal/db/models/dns/dns_domain_model_ext.go

package dns

import (
	"encoding/json"

	"github.com/TeaOSLab/EdgeAPI/internal/dnsclients/dnstypes"
)

// DecodeRoutes 获取所有的线路
func (this *DNSDomain) DecodeRoutes() ([]*dnstypes.Route, error) {
	if len(this.Routes) == 0 {
		return nil, nil
	}
	result := []*dnstypes.Route{}
	err := json.Unmarshal(this.Routes, &result)
	if err != nil {
		return nil, err
	}
	return result, nil
}

// ContainsRouteCode 检查是否包含某个线路
func (this *DNSDomain) ContainsRouteCode(route string) (bool, error) {
	routes, err := this.DecodeRoutes()
	if err != nil {
		return false, err
	}
	for _, r := range routes {
		if r.Code == route {
			return true, nil
		}
	}
	return false, nil
}

// DecodeRecords 获取所有的记录
func (this *DNSDomain) DecodeRecords() ([]*dnstypes.Record, error) {
	records := this.Records
	if len(records) == 0 {
		return nil, nil
	}
	result := []*dnstypes.Record{}
	err := json.Unmarshal(records, &result)
	if err != nil {
		return nil, err
	}
	return result, nil
}