Highways4Good/scripts/1.4.0/data/1_calc_village_distance_无需执行/index.js

try {
    const { Pool, Client } = require('pg')
    const request = require('superagent')
    const path = require('path')
    const fs = require("fs");

    const pool = new Pool({
        host: '10.8.30.32',
        port: 5432,
        user: 'postgres',
        password: '123',
        database: 'highways4good',
    })
    const godKey = '21c2d970e1646bb9a795900dd00093ce'

    function getDistance (lat1, lon1, lat2, lon2) {
        var R = 6371000; // 半径 of the earth in meters    
        var dLat = deg2rad(lat2 - lat1);  // deg2rad below    
        var dLon = deg2rad(lon2 - lon1);
        var a =
            Math.sin(dLat / 2) * Math.sin(dLat / 2) +
            Math.cos(deg2rad(lat1)) * Math.cos(deg2rad(lat2)) *
            Math.sin(dLon / 2) * Math.sin(dLon / 2)
            ;
        var c = 2 * Math.atan2(Math.sqrt(a), Math.sqrt(1 - a));
        var d = R * c; // Distance in meters    
        return d;
    }

    function deg2rad (deg) {
        return deg * (Math.PI / 180)
    }

    // let distance = getDistance(coord1.latitude, coord1.longitude, coord2.latitude, coord2.longitude);   

    const fun = async () => {
        // note: we don't try/catch this because if connecting throws an exception
        // we don't need to dispose of the client (it will be undefined)
        const client = await pool.connect()
        try {
            await client.query('BEGIN')
            console.log(`开始`);

            const noTownshipLoop = true

            // 查南昌市下的区县
            const nanchangRes =
                noTownshipLoop ? [] : await request.get(`https://restapi.amap.com/v3/config/district?key=${godKey}&keywords=360100&subdistrict=1`)

            let townshipList =
                noTownshipLoop ? [] : nanchangRes.body.districts[0].districts

            // 获取各村位置

            for (let t of (noTownshipLoop ? [{}] : townshipList)) {
                console.log(`${t.name}`);

                // 划分 20 个矩形查询
                let leftTop = [115.792048, 28.976526]
                let rightBottom = [116.324981, 28.262191]

                for (let i = 0; i < 5; i++) { // 纵向

                    let top = leftTop[1] + (rightBottom[1] - leftTop[1]) / 5 * i
                    let bottom = leftTop[1] + (rightBottom[1] - leftTop[1]) / 5 * (i + 1)

                    for (let j = 0; j < 5; j++) { // 横向
                        let left = leftTop[0] + (rightBottom[0] - leftTop[0]) / 5 * j
                        let right = leftTop[0] + (rightBottom[0] - leftTop[0]) / 5 * (j + 1)

                        console.log(`矩形：${left},${top},${right},${bottom}`);



                        let pageNum = 1
                        let continueSearch = true
                        for (; continueSearch;) {
                            console.log(`查询第${pageNum}页`);
                            let pageSize = 10
                            /** 关键词查询 */
                            // const res = await request.get(`https://restapi.amap.com/v5/place/text?key=${godKey}&types=130106&region=360000&citylimit=true&page_size=25&page_num=${pageNum}&extensions=all`)

                            /** 区县查询 */
                            // const res = await request.get(`https://restapi.amap.com/v3/place/text?key=${godKey}&keywords=%E6%9D%91%E5%A7%94%E4%BC%9A%7C%E6%9D%91%E6%B0%91%E5%A7%94%E5%91%98%E4%BC%9A&page_size=25&types=130106&citylimit=true&city=${t.adcode}&page=${pageNum}`)

                            /** 矩形查询 */
                            pageSize = 50
                            // const res = await request.get(`https://restapi.amap.com/v3/place/polygon?key=21c2d970e1646bb9a795900dd00093ce&keywords=%E6%9D%91%E5%A7%94%E4%BC%9A%7C%E6%9D%91%E6%B0%91%E5%A7%94%E5%91%98%E4%BC%9A&types=130106&offset=${pageSize}&page=${pageNum}&polygon=${left},${top}|${right},${bottom}`)

                            /** 矩形查询 V2 */
                            pageSize = 25
                            const res = await request.get(`https://restapi.amap.com/v5/place/polygon?key=21c2d970e1646bb9a795900dd00093ce&keywords=%E6%9D%91%E5%A7%94%E4%BC%9A%7C%E6%9D%91%E6%B0%91%E5%A7%94%E5%91%98%E4%BC%9A&types=130106&page_size=${pageSize}&page_num=${pageNum}&polygon=${left},${top}|${right},${bottom}`)

                            await new Promise(resolve => setTimeout(() => resolve(), 2000));

                            // console.log(res.body);
                            const data = res.body
                            // console.log(data);
                            console.log(`查得 ${data.pois.length} 条`);
                            if (data.count == 0) {
                                continueSearch = false
                                break;
                            } else {
                                pageNum++
                            }
                            for await (let p of data.pois) {
                                console.log(p.name);
                                if (p.name.includes('村委会') || p.name.includes('村民委员会')) {
                                    let vName = p.name.replace('村委会', '').replace('村民委员会', '').replace(/\(.*?\)/g, '')

                                    if (vName.includes('乡')) {
                                        let townKeyIndex = vName.indexOf('乡')
                                        if (townKeyIndex >= 0) {
                                            vName = vName.substring(townKeyIndex + 1)
                                        }
                                    }
                                    if (vName.includes('镇')) {
                                        let townKeyIndex = vName.indexOf('镇')
                                        if (townKeyIndex >= 0) {
                                            vName = vName.substring(townKeyIndex + 1)
                                        }
                                    }

                                    if (!vName.endsWith('村')) {
                                        vName = vName + '村'
                                    }

                                    // console.log(`${vName}`);
                                    let locationArr = p.location.split(',')

                                    // 查询镇名是否存在
                                    let existRes = await client.query(
                                        `SELECT * FROM village WHERE name=$1`,
                                        [vName]
                                    )
                                    if (existRes.rowCount == 0) {
                                        existRes = await client.query(
                                            `SELECT * FROM village WHERE name=$1`,
                                            [vName + '委会']
                                        )
                                    }

                                    if (existRes.rowCount == 0) {

                                    } else {
                                        let existV = existRes.rows[0]

                                        if (existV.id < 207) {
                                            continue
                                        }
                                        // console.log(existRes);
                                        await client.query(
                                            `UPDATE village SET longitude=$1, latitude=$2 WHERE id=$3`,
                                            [locationArr[0], locationArr[1], existV.id]
                                        )
                                    }
                                }
                            }

                            if (data.pois.length < pageSize) {
                                continueSearch = false
                            }
                        }
                    }

                }
            }

            const allVillagesRes = await client.query(
                `SELECT * FROM village`
            )
            const allVillages = allVillagesRes.rows
            for (let v of allVillages) {
                if (!v.longitude || !v.latitude) {
                    continue
                }
                for (let vv of allVillages) {
                    if (
                        v.id != vv.id
                        // 只计算一个乡镇内的村
                        && v.township_code == vv.township_code
                    ) {
                        if (!vv.longitude || !vv.latitude) {
                            continue
                        }
                        let distance = getDistance(v.latitude, v.longitude, vv.latitude, vv.longitude)

                        // console.log(`${v.name} 与 ${vv.name} 的距离为 ${distance} 米`);

                        distance = Math.round(distance)
                        const existRes = await client.query(
                            `SELECT * FROM village_distance WHERE origin_village=$1 AND calc_village=$2`,
                            [v.id, vv.id]
                        )
                        if (existRes.rowCount == 0) {
                            await client.query(
                                `INSERT INTO village_distance (origin_village, calc_village, distance) VALUES ($1, $2, $3)`,
                                [v.id, vv.id, distance]
                            )
                        } else {
                            let upSql = `UPDATE village_distance SET distance=$1 WHERE origin_village=$2 AND calc_village=$3`
                            await client.query(
                                upSql
                                , [distance, v.id, vv.id]
                            )
                        }
                    }
                }
            }

            // await client.query('ROLLBACK')
            await client.query('COMMIT')
            console.log('执行完毕~')
        } catch (e) {
            await client.query('ROLLBACK')
            console.log('执行错误~')
            throw e
        } finally {
            client.release();
        }
    }

    fun()
} catch (error) {
    console.error(error)
}