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.*;
import com.skyversation.poiaddr.service.impl.*;
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.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@Service
public class AreaService {

    @Resource
    private YyskAddressStandardizationServiceImpl yyskAddressStandardizationService;

    @Autowired
    private AmapAddressV3Service amapAddressV3Service;

    @Autowired
    private YyskAddressStandardizationSerice yyskAddressStandardizationSerice;

    @Autowired
    private FusionKjdlTydzWfSerice fusionKjdlTydzWfSerice;

    @Autowired
    private YyszAddressQpService yyszAddressQpService;

    //  暂存数据库中的所有数据
    private Map<String, AmapAddressV3> allAddressData = new ConcurrentHashMap<>();

    private static AreaService instance;

    public static AreaService getInstance() {
        if (instance == null) {
            instance = 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 {
            long startTime = System.currentTimeMillis();
            System.out.println("<<<<<<<<------开始数据库连通性测试");
            List<AmapAddressV3> amapAddressV3List = amapAddressV3Service.getAllData();
            long endTime = System.currentTimeMillis();
            System.out.println(">>>>>>>>------数据库连通性测试完成！用时" + (endTime - startTime) / 1000 + "秒！当前数据库数据条数:" + amapAddressV3List.size());
            System.out.println("<<<<<<<<------开始缓存数据库数据");
            for (AmapAddressV3 amapAddressV3 : amapAddressV3List) {
                if(amapAddressV3 != null && amapAddressV3.getAddress() != null){
                    allAddressData.put(amapAddressV3.getAddress(), amapAddressV3);
                }
            }
            amapAddressV3List.clear();
            System.out.println(">>>>>>>>------缓存数据库数据完成！allAddressData.size()=" + allAddressData.size());
        } catch (Exception e) {
            e.printStackTrace();
            System.err.println(">>>>>>>>------数据库连通性测试结果:" + e);
        }*/

//      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) {
            e.printStackTrace();
            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) {
        List<AmapAddressV3> amapAddressV3SaveList = new ArrayList<>();
        for (AmapAddressV3 amapAddressV3 : amapAddressV3List) {
            if (!allAddressData.containsKey(amapAddressV3.getAddress())) {
                allAddressData.put(amapAddressV3.getAddress(), amapAddressV3);
                amapAddressV3SaveList.add(amapAddressV3);
            }
        }
        amapAddressV3List.clear();
        if (amapAddressV3SaveList.size() > 0) {
            amapAddressV3Service.saveAmapAddressV3(amapAddressV3SaveList);
            amapAddressV3List.clear();
        }
    }

    public void saveYyskAddress(List<YyskDmdzAddressStandardization> datas) {
        yyskAddressStandardizationSerice.saveDatas(datas);
        datas.clear();
    }

    public void saveFktws(List<FusionKjdlTydzWf> datas){
        fusionKjdlTydzWfSerice.saveDatas(datas);
        datas.clear();
    }

    public List<YyszAddressQp> getAddressPoisByAddr(String addr) {
        /*List<YyszAddressQp> resultData = new ArrayList<>();
        String[] strs = AddressMatcher.matchProvinceAndCity(addr);
        if (strs != null) {
            for (String str : strs) {
                if (StringUtils.hasText(str)) {
                    addr = addr.replace(str, "");
                }
            }
        }

        if (addr.contains("号")) {
            addr = addr.substring(0, addr.lastIndexOf("号"));// 红宝石50号   红宝石50    返回 红宝石503弄
        }
        if (addr.contains("弄")) {
            addr = addr.substring(0, addr.lastIndexOf("弄"));
        }
        if (allAddressData.containsKey(addr)) {
            resultData.add(allAddressData.get(addr));
            return resultData;
        } else {
            int sumSize = 100;
            for (String address : allAddressData.keySet()) {
                if (address != null && address.contains(addr) && sumSize > 0) {
                    sumSize--;
                    resultData.add(allAddressData.get(address));
                }
            }
        }
        return resultData;*/
//        TODO 使用缓存替换数据库查询
//        System.out.println("<<<<<<<----- 开始数据库查询，数据查询地址为：" + addr);
        return yyszAddressQpService.getAddressPoisByAddr(addr);
    }

    public GeoJsonBean isInadPolygon(double lon, double lat) {
        try {
            Geometry point = reader.read("POINT (" + lon + " " + lat + ")");
            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 (" + lon + " " + lat + ")");
            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 (" + lon + " " + lat + ")");
            for (GeoJsonBean bean : shAllResidentialCommitteeList) {
                if (bean.getMultiPolygon().contains(point)) {
                    return bean;
                }
            }
            return null;
        } catch (ParseException e) {
            throw new RuntimeException(e);

        }
    }
}