aded CidrGeoMap asnd GeoIpData

ppl-spark-integration/src/main/java/org/opensearch/sql/ast/expression/GeoIp.java

@@ -1,5 +1,6 @@
 package org.opensearch.sql.ast.expression;
 
+import lombok.AllArgsConstructor;
 import lombok.EqualsAndHashCode;
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;

ppl-spark-integration/src/main/java/org/opensearch/sql/common/geospatial/CidrGeoMap.java

@@ -0,0 +1,41 @@
+package org.opensearch.sql.common.geospatial;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class CidrGeoMap extends HashMap<String, GeoIpData> {
+
+    private String toBinaryString(String ip) {
+        StringBuilder binary = new StringBuilder();
+        for (String octet : ip.split("\\.")) {
+            binary.append(String.format("%8s", Integer.toBinaryString(Integer.parseInt(octet)))
+                    .replace(' ', '0'));
+        }
+        return binary.toString();
+    }
+
+    @Override
+    public GeoIpData put(String cidr, GeoIpData data) {
+        String[] parts = cidr.split("/");
+        String cidrKey = toBinaryString(parts[0]);
+        int prefixLength = Integer.parseInt(parts[1]);
+        cidrKey = cidrKey.substring(0, prefixLength - 1);
+
+        super.put(cidrKey, data);
+
+        return data;
+    }
+
+    @Override
+    public GeoIpData get(Object ipAddress) {
+        String binaryIP = toBinaryString(ipAddress.toString());
+
+        GeoIpData res = null;
+
+        while (binaryIP.length() > 0 && res == null) {
+            res = super.get(binaryIP);
+            binaryIP = binaryIP.substring(0, binaryIP.length() - 2);
+        }
+    }
+}
+

ppl-spark-integration/src/main/java/org/opensearch/sql/common/geospatial/GeoIpData.java

@@ -0,0 +1,18 @@
+package org.opensearch.sql.common.geospatial;
+
+import lombok.*;
+
+@Getter
+@ToString
+@AllArgsConstructor
+public class GeoIpData {
+    private final String country_is_code;
+    private final String country_name;
+    private final String continent_name;
+    private final String region_iso_code;
+    private final String region_name;
+    private final String city_name;
+    private final String time_zone;
+    private final String lat;
+    private final String lon;
+}