@@ -246,6 +246,7 @@ private Map<Field,Object> retrieveCityGeoData(InetAddress ipAddress) throws GeoI
246
246
Postal postal = response .getPostal ();
247
247
Subdivision subdivision = response .getMostSpecificSubdivision ();
248
248
Map <Field , Object > geoData = new EnumMap <>(Field .class );
249
+ Network network = response .getTraits ().getNetwork ();
249
250
250
251
// if location is empty, there is no point populating geo data
251
252
// and most likely all other fields are empty as well
@@ -344,6 +345,11 @@ private Map<Field,Object> retrieveCityGeoData(InetAddress ipAddress) throws GeoI
344
345
geoData .put (Field .LONGITUDE , lon );
345
346
}
346
347
break ;
348
+ case NETWORK :
349
+ if (network != null ) {
350
+ geoData .put (Field .NETWORK , network .toString ());
351
+ }
352
+ break ;
347
353
}
348
354
}
349
355
@@ -397,6 +403,7 @@ private Map<Field, Object> retrieveIspGeoData(InetAddress ipAddress) throws GeoI
397
403
} catch (NullPointerException e ) {
398
404
throw new GeoIp2InvalidCustomFieldException (e );
399
405
}
406
+ Network network = response .getNetwork ();
400
407
401
408
Map <Field , Object > geoData = new EnumMap <>(Field .class );
402
409
for (Field desiredField : this .desiredFields ) {
@@ -405,7 +412,7 @@ private Map<Field, Object> retrieveIspGeoData(InetAddress ipAddress) throws GeoI
405
412
geoData .put (Field .IP , ipAddress .getHostAddress ());
406
413
break ;
407
414
case AUTONOMOUS_SYSTEM_NUMBER :
408
- Integer asn = response .getAutonomousSystemNumber ();
415
+ Long asn = response .getAutonomousSystemNumber ();
409
416
if (asn != null ) {
410
417
geoData .put (desiredField , asn );
411
418
}
@@ -428,6 +435,11 @@ private Map<Field, Object> retrieveIspGeoData(InetAddress ipAddress) throws GeoI
428
435
geoData .put (Field .ORGANIZATION , org );
429
436
}
430
437
break ;
438
+ case NETWORK :
439
+ if (network != null ) {
440
+ geoData .put (Field .NETWORK , network .toString ());
441
+ }
442
+ break ;
431
443
}
432
444
}
433
445
@@ -450,7 +462,7 @@ private Map<Field, Object> retrieveAsnGeoData(InetAddress ipAddress) throws GeoI
450
462
geoData .put (Field .IP , ipAddress .getHostAddress ());
451
463
break ;
452
464
case AUTONOMOUS_SYSTEM_NUMBER :
453
- Integer asn = response .getAutonomousSystemNumber ();
465
+ Long asn = response .getAutonomousSystemNumber ();
454
466
if (asn != null ) {
455
467
geoData .put (Field .AUTONOMOUS_SYSTEM_NUMBER , asn );
456
468
}
@@ -507,7 +519,7 @@ private Map<Field, Object> retrieveEnterpriseGeoData(InetAddress ipAddress) thro
507
519
Continent continent = response .getContinent ();
508
520
Subdivision subdivision = response .getMostSpecificSubdivision ();
509
521
510
- Integer asn = response .getTraits ().getAutonomousSystemNumber ();
522
+ Long asn = response .getTraits ().getAutonomousSystemNumber ();
511
523
String organizationName = response .getTraits ().getAutonomousSystemOrganization ();
512
524
Network network = response .getTraits ().getNetwork ();
513
525
@@ -620,6 +632,7 @@ private Map<Field, Object> retrieveAnonymousIpGeoData(final InetAddress ipAddres
620
632
boolean isAnonymous = response .isAnonymous ();
621
633
boolean isPublicProxy = response .isPublicProxy ();
622
634
boolean isResidentialProxy = response .isResidentialProxy ();
635
+ Network network = response .getNetwork ();
623
636
624
637
for (Field desiredField : this .desiredFields ) {
625
638
switch (desiredField ) {
@@ -644,6 +657,11 @@ private Map<Field, Object> retrieveAnonymousIpGeoData(final InetAddress ipAddres
644
657
case RESIDENTIAL_PROXY :
645
658
geoData .put (desiredField , isResidentialProxy );
646
659
break ;
660
+ case NETWORK :
661
+ if (network != null ) {
662
+ geoData .put (Field .NETWORK , network .toString ());
663
+ }
664
+ break ;
647
665
}
648
666
}
649
667
return geoData ;
0 commit comments