place_name_search/src/main/java/com/skyversation/poiaddr/service/AreaService.java

package com.skyversation.poiaddr.service;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.skyversation.poiaddr.addquery.AddressQueryEngine;
import com.skyversation.poiaddr.addquery.Constant;
import com.skyversation.poiaddr.bean.AddressResult;
import com.skyversation.poiaddr.bean.GeoJsonBean;
import com.skyversation.poiaddr.entity.AmapAddressV3;
import com.skyversation.poiaddr.service.impl.AmapAddressV3Service;
import com.skyversation.poiaddr.service.impl.YyskAddressStandardizationServiceImpl;
import com.skyversation.poiaddr.util.AddressMatcher;
import org.geotools.geojson.geom.GeometryJSON;
import org.geotools.geometry.jts.JTSFactoryFinder;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.geom.MultiPolygon;
import org.locationtech.jts.io.ParseException;
import org.locationtech.jts.io.WKTReader;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.ClassPathResource;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.io.*;
import java.util.ArrayList;
import java.util.List;

@Service
public class AreaService {

    @Resource
    private YyskAddressStandardizationServiceImpl yyskAddressStandardizationService;

    @Autowired
    private AmapAddressV3Service amapAddressV3Service;

    private static AreaService instance;

    public static AreaService getInstance() {
        if (instance == null) {
            return new AreaService();
        }
        return instance;
    }

    @Value("${app.area}")
    private String area;

    @Value("${app.town}")
    private String town;

    WKTReader reader = new WKTReader(JTSFactoryFinder.getGeometryFactory());
    public List<GeoJsonBean> shAllAdministrativeDivisionPolygonList = new ArrayList<>();
    public List<GeoJsonBean> shAllToensPolygonList = new ArrayList<>();
    public List<GeoJsonBean> shAllResidentialCommitteeList = new ArrayList<>();

    @PostConstruct
    public void initGeoData() {
        System.out.println("<<<<<<<<------AreaService>initGeoData------>>>>>>>>>");
        instance = this;
        Thread thread3 = new Thread(this::initPolygons);
        thread3.start();

        try {
            thread3.join();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
//      TODO 初始化配置文件变量
        Constant.setTowns(town.split(","));
        Constant.setArea(area);
        System.out.println("<<<<<<<<------根据配置文件为AREA赋值:" + Constant.getArea());
        System.out.println("<<<<<<<<------当前AMAP_CITY_CODE:" + Constant.getAMAP_CITY_CODE());
//      TODO 网络连通性测试，可以请求一下测试的市中心地址
        try {
            System.out.println("电脑最大线程数：" + Runtime.getRuntime().availableProcessors());
            System.out.println("<<<<<<<<------开始网络连通性测试");
            long startTime = System.currentTimeMillis();
            List<String> addrs = new ArrayList<>();
            addrs.add("上海市松江区乐都路339号");
            AddressResult addressResult = AddressQueryEngine.getInstance().commonSearchByName(addrs);
            long endTime = System.currentTimeMillis();
            System.out.println(">>>>>>>>------网络连通性测试完成！用时" + (endTime - startTime) / 1000 + "秒！结果:" + addressResult);
        } catch (Exception e) {
            System.err.println(">>>>>>>>------网络连通性测试结果:" + e);
        }
//      TODO 数据库连通性测试
        try {
            long startTime = System.currentTimeMillis();
            System.out.println("<<<<<<<<------开始数据库连通性测试");
            Long amapAddressV3List = amapAddressV3Service.countAllUsersNative();
            long endTime = System.currentTimeMillis();
            System.out.println(">>>>>>>>------数据库连通性测试完成！用时" + (endTime - startTime) / 1000 + "秒！当前数据库数据条数:" + amapAddressV3List);
//            System.out.println("<<<<<<<<------开始执行任务！！！");
//            yyskAddressStandardizationService.searchLoadFileData("D:\\1.项目\\5. 松江项目\\4. 地名地址\\地名地址库治理数据\\原始数据\\",
//                    0, 10000, 1);
        } catch (Exception e) {
            System.err.println(">>>>>>>>------数据库连通性测试结果:" + e);
        }
    }

    public void initPolygons() {
        List<List<GeoJsonBean>> lis = new ArrayList<>();
        lis.add(shAllAdministrativeDivisionPolygonList);
        lis.add(shAllToensPolygonList);
        lis.add(shAllResidentialCommitteeList);
        String[] path = new String[]{"geojson/上海市_区县边界.geojson", "geojson/上海市_乡镇边界.geojson", "geojson/上海市_村居边界.geojson"};

        for (int j = 0; j < path.length; j++) {
            BufferedReader reader = null;
            try {
                // 创建 ClassPathResource 对象，指定静态文件的路径
                ClassPathResource resource = new ClassPathResource(path[j]);
                // 获取文件的输入流
                reader = new BufferedReader(new InputStreamReader(resource.getInputStream()));
                String line;
                StringBuilder sb = new StringBuilder();
                while ((line = reader.readLine()) != null) {
                    sb.append(line).append("\r\n");
                }
                reader.close();
                JSONObject json = JSONObject.parseObject(sb.toString());

                JSONArray featrues = json.getJSONArray("features");

                for (int i = 0; i < featrues.size(); i++) {
                    JSONObject properties = featrues.getJSONObject(i).getJSONObject("properties");
                    JSONObject geoJson = new JSONObject();
                    geoJson.put("type", "MultiPolygon");
                    geoJson.put("coordinates",
                            featrues.getJSONObject(i).getJSONObject("geometry").getJSONArray("coordinates"));
                    GeometryJSON geometryJSON = new GeometryJSON();
                    MultiPolygon multiPolygon = geometryJSON.readMultiPolygon(geoJson.toJSONString());
                    GeoJsonBean geoJsonBean = new GeoJsonBean();
                    geoJsonBean.setProperties(properties);
                    geoJsonBean.setMultiPolygon(multiPolygon);
                    lis.get(j).add(geoJsonBean);
                }
            } catch (IOException e) {
                e.printStackTrace();
                System.out.println("<<<<<<<<------run geo data error------>>>>>>>>>");
            }
        }

        System.out.println("<<<<<<<<------run geo data complete------>>>>>>>>>");
    }

    public void saveAmapAddressV3(List<AmapAddressV3> amapAddressV3List){
        amapAddressV3Service.saveAmapAddressV3(amapAddressV3List);
    }

    public List<AmapAddressV3> getAddressPoisByAddr(String addr){
        String[] strs = AddressMatcher.matchProvinceAndCity(addr);

        for(String str : strs){
            if(StringUtils.hasText(str)){
                addr = addr.replace(str, "");
            }
        }

        if(addr.contains("号")){
            addr = addr.substring(0, addr.indexOf("号") + 1);
        } else if(addr.contains("弄")){
            addr = addr.substring(0, addr.indexOf("弄") + 1);
        }
        System.out.println("<<<<<<<----- 开始数据库查询，数据查询地址为：" + addr);
        return amapAddressV3Service.getAddressPoisByAddr(addr);
    }

    public GeoJsonBean isInadPolygon(double lon, double lat) {
        try {
            Geometry point = reader.read("POINT (" + lat + " " + lon + ")");
            for (GeoJsonBean bean : shAllAdministrativeDivisionPolygonList) {
                if (bean.getMultiPolygon().contains(point)) {
                    return bean;
                }
            }
            return null;
        } catch (ParseException e) {
            throw new RuntimeException(e);
        }
    }

    public GeoJsonBean isInTownPolygon(double lon, double lat) {
        try {
            Geometry point = reader.read("POINT (" + lat + " " + lon + ")");
            for (GeoJsonBean bean : shAllToensPolygonList) {
                if (bean.getMultiPolygon().contains(point)) {
                    return bean;
                }
            }
            return null;
        } catch (ParseException e) {
            throw new RuntimeException(e);

        }
    }

    public GeoJsonBean isInResidentialCommitteePolygon(double lon, double lat) {
        try {
            Geometry point = reader.read("POINT (" + lat + " " + lon + ")");
            for (GeoJsonBean bean : shAllResidentialCommitteeList) {
                if (bean.getMultiPolygon().contains(point)) {
                    return bean;
                }
            }
            return null;
        } catch (ParseException e) {
            throw new RuntimeException(e);

        }
    }
}