✨ 新增ip2region解析IP地址

eiam-common/src/main/java/cn/topiam/employee/common/enums/Country.java

@@ -17,10 +17,9 @@
  */
 package cn.topiam.employee.common.enums;
 
-import com.fasterxml.jackson.annotation.JsonValue;
-
 import cn.topiam.employee.support.enums.BaseEnum;
 import cn.topiam.employee.support.web.converter.EnumConvert;
+import com.fasterxml.jackson.annotation.JsonValue;
 
 /**
  * <p>
@@ -1031,6 +1030,16 @@ public enum Country implements BaseEnum {
         return null;
     }
 
+    public static Country getByCountryName(String name) {
+        Country[] values = values();
+        for (Country country : values) {
+            if (String.valueOf(country.getName()).equals(name)) {
+                return country;
+            }
+        }
+        return null;
+    }
+
     @Override
     public String getCode() {
         return code;

eiam-common/src/main/java/cn/topiam/employee/common/geo/ip2region/Ip2regionGeoLocationServiceImpl.java

@@ -0,0 +1,95 @@
+/*
+ * eiam-common - Employee Identity and Access Management
+ * Copyright © 2022-Present Jinan Yuanchuang Network Technology Co., Ltd. (support@topiam.cn)
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package cn.topiam.employee.common.geo.ip2region;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Objects;
+
+import org.lionsoul.ip2region.xdb.Searcher;
+import org.springframework.beans.factory.DisposableBean;
+import org.springframework.util.Assert;
+
+import cn.topiam.employee.common.enums.Country;
+import cn.topiam.employee.support.geo.GeoLocation;
+import cn.topiam.employee.support.geo.GeoLocationProvider;
+import cn.topiam.employee.support.geo.GeoLocationService;
+
+import lombok.extern.slf4j.Slf4j;
+import static cn.topiam.employee.common.geo.District.CITY_DISTRICT;
+import static cn.topiam.employee.common.geo.District.PROVINCE_DISTRICT;
+
+/**
+ * Ip2regionGeo
+ *
+ * @author SanLi
+ * Created by qinggang.zuo@gmail.com / 2689170096@qq.com on  2023/10/30 19:11
+ */
+@Slf4j
+public class Ip2regionGeoLocationServiceImpl implements GeoLocationService, DisposableBean {
+
+    public static GeoLocationProvider IP2REGION        = new GeoLocationProvider("default",
+        "ip2region");
+
+    private static final String       DEFAULT_XDB_PATH = "/ip2region/ip2region.xdb";
+
+    private final Searcher            searcher;
+
+    public Ip2regionGeoLocationServiceImpl() throws IOException {
+        // 加载整个xdb到内存。
+        try (InputStream in = this.getClass().getResourceAsStream(DEFAULT_XDB_PATH)) {
+            Assert.notNull(in, "XDB must not be null");
+            byte[] cBuff = in.readAllBytes();
+            log.info("create content cached searcher");
+            // 2、使用上述的 cBuff 创建一个完全基于内存的查询对象。
+            searcher = Searcher.newWithBuffer(cBuff);
+        }
+    }
+
+    @Override
+    public GeoLocation getGeoLocation(String remote) {
+        try {
+            String region = searcher.search(remote);
+            return buildGeoLocation(remote, region);
+        } catch (Exception e) {
+            log.error("failed to search({})", remote, e);
+        }
+        return GeoLocation.builder().provider(IP2REGION).ip(remote).build();
+    }
+
+    private GeoLocation buildGeoLocation(String remote, String region) {
+        String[] geo = region.split("\\|");
+        String countryName = geo[0];
+        String provinceName = geo[2];
+        String cityName = geo[3];
+        Country country = Country.getByCountryName(countryName);
+        return GeoLocation.builder().provider(IP2REGION).ip(remote)
+            .countryCode(Objects.nonNull(country) ? country.getCode() : null)
+            .countryName(countryName)
+            .provinceCode(Objects.toString(PROVINCE_DISTRICT.get(provinceName)))
+            .provinceName(provinceName).cityCode(Objects.toString(CITY_DISTRICT.get(cityName)))
+            .cityName(cityName).build();
+    }
+
+    @Override
+    public void destroy() throws Exception {
+        if (Objects.nonNull(searcher)) {
+            searcher.close();
+        }
+    }
+}