perf: tool support ipv6

This commit is contained in:
feng
2026-01-21 14:22:31 +08:00
committed by ZhaoJiSen
parent 462b299280
commit abe5b7b157
2 changed files with 9 additions and 3 deletions

View File

@@ -1,5 +1,3 @@
from django.conf import settings
from django.utils.translation import gettext_lazy as _
from django_filters import rest_framework as drf_filters
from rest_framework import status
from rest_framework.decorators import action

View File

@@ -113,6 +113,14 @@ def get_ip_city(ip):
def lookup_domain(domain):
try:
return socket.gethostbyname(domain), ''
addrinfos = socket.getaddrinfo(
domain, None, socket.AF_UNSPEC, socket.SOCK_DGRAM
)
family, _, _, _, sockaddr = addrinfos[0]
if family == socket.AF_INET:
return sockaddr[0], ''
if family == socket.AF_INET6:
return sockaddr[0], ''
return domain, ''
except Exception as e:
return None, f'Cannot resolve {domain}: Unknown host, {e}'