country playon

src/main/java/app/entities/server/PlayOn.java

@@ -11,6 +11,7 @@ public class PlayOn {
     int player_id;
     String ip;
     String name;
+    String country;
 
     public String getIp() {
         if (ip.contains(":")){

src/main/java/app/services/StatsService.java

@@ -5,6 +5,7 @@ import app.entities.other.SteamID;
 import app.entities.server.PlayOn;
 import app.entities.server.Server;
 import app.entities.server.players.RCONPlayer;
+import app.services.io.GeoIP;
 import app.utils.CryptedCookie;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.ApplicationContext;
@@ -17,11 +18,13 @@ public class StatsService {
     Stats stats;
 
     private CryptedCookie cryptedCookie;
+    private GeoIP geoIP;
 
     @Autowired
-    public StatsService (Stats stats, CryptedCookie cryptedCookie) {
+    public StatsService (Stats stats, CryptedCookie cryptedCookie, GeoIP geoIP) {
         this.stats = stats;
         this.cryptedCookie = cryptedCookie;
+        this.geoIP = geoIP;
     }
 
     public PlayOn searchPlayer(SteamID steamID){
@@ -31,7 +34,8 @@ public class StatsService {
                     stringServerEntry.getKey(),
                     player.getId(),
                     cryptedCookie.Hashed(player.getIp()),
-                    player.getName());
+                    player.getName(),
+                    geoIP.GetCountry(player.getIp().split(":", 2)[0], "Unknown"));
         }
         return null;
     }

src/main/java/app/services/io/GeoIP.java

@@ -24,4 +24,12 @@ public class GeoIP {
     public String GetCountry(String ip) throws UnknownHostException, GeoIp2Exception, IOException {
         return databaseReader.country(InetAddress.getByName(ip)).getCountry().getName();
     }
+
+    public String GetCountry(String ip, String replace) {
+        try {
+            return this.GetCountry(ip);
+        } catch (GeoIp2Exception | IOException e) {
+            return replace;
+        }
+    }
 }