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

ddns: retreive dns servers from context (#1034)

Browse Source
This commit is contained in:
UUBulb
2025-03-17 23:11:40 +08:00
committed by GitHub
parent c3ec52e392
commit 38c2374bad
5 changed files with 17 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
service/rpc/nezha.go
View File

@@ -6,6 +6,7 @@ import (
"fmt"
"log"
"net"
"strings"
"sync"
"time"
@@ -13,6 +14,7 @@ import (
"github.com/nezhahq/nezha/pkg/ddns"
geoipx "github.com/nezhahq/nezha/pkg/geoip"
"github.com/nezhahq/nezha/pkg/grpcx"
"github.com/nezhahq/nezha/pkg/utils"
"github.com/nezhahq/nezha/model"
pb "github.com/nezhahq/nezha/proto"
@@ -236,12 +238,15 @@ func (s *NezhaHandler) ReportGeoIP(c context.Context, r *pb.GeoIP) (*pb.GeoIP, e
ipv4 := geoip.IP.IPv4Addr
ipv6 := geoip.IP.IPv6Addr
dnsServers := strings.Split(singleton.Conf.DNSServers, ",")
ctx := context.WithValue(context.Background(), ddns.DNSServerKey{}, utils.IfOr(len(dnsServers) > 0, dnsServers, utils.DNSServers))
providers, err := singleton.DDNSShared.GetDDNSProvidersFromProfiles(server.DDNSProfiles, &model.IP{IPv4Addr: ipv4, IPv6Addr: ipv6})
if err == nil {
for _, provider := range providers {
domains := server.OverrideDDNSDomains[provider.GetProfileID()]
go func(provider *ddns.Provider) {
provider.UpdateDomain(context.Background(), domains...)
provider.UpdateDomain(ctx, domains...)
}(provider)
}
} else {