KJZH001/nezha-2024
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve: Handle case when host is not found in LookupGeoIP

Browse Source
This commit is contained in:
naiba 2024-07-28 15:11:50 +08:00
parent df9487b5fc
commit fe234f4f38
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
service/rpc/nezha.go
View File

@ -243,9 +243,10 @@ func (s *NezhaHandler) LookupGeoIP(c context.Context, r *pb.GeoIP) (*pb.GeoIP, e
// 将地区码写入到 Host // 将地区码写入到 Host
singleton.ServerLock.RLock() singleton.ServerLock.RLock()
defer singleton.ServerLock.RUnlock() defer singleton.ServerLock.RUnlock()
if singleton.ServerList[clientID].Host != nil { if singleton.ServerList[clientID].Host == nil {
singleton.ServerList[clientID].Host.CountryCode = location return nil, fmt.Errorf("host not found")
} }
singleton.ServerList[clientID].Host.CountryCode = location
return &pb.GeoIP{Ip: ip, CountryCode: location}, nil return &pb.GeoIP{Ip: ip, CountryCode: location}, nil
} }