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 = false
            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 < 219) {
            //                                 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 {
                            if (!noTownshipLoop) {
                                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)
}