improve: Handle case when host is not found in LookupGeoIP

nezhahq · Jul 28, 2024 · fe234f4 · fe234f4
1 parent df9487b
commit fe234f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/service/rpc/nezha.go b/service/rpc/nezha.go
@@ -243,9 +243,10 @@ func (s *NezhaHandler) LookupGeoIP(c context.Context, r *pb.GeoIP) (*pb.GeoIP, e
 	// 将地区码写入到 Host
 	singleton.ServerLock.RLock()
 	defer singleton.ServerLock.RUnlock()
-	if singleton.ServerList[clientID].Host != nil {
-		singleton.ServerList[clientID].Host.CountryCode = location
+	if singleton.ServerList[clientID].Host == nil {
+		return nil, fmt.Errorf("host not found")
 	}
+	singleton.ServerList[clientID].Host.CountryCode = location
 
 	return &pb.GeoIP{Ip: ip, CountryCode: location}, nil
 }