MongoDB（28）什么是地理空间索引？

地理空间索引（Geospatial Index）是MongoDB中用于优化地理空间数据查询的索引类型。地理空间索引允许你对地理位置数据进行高效的存储和查询，如距离计算、位置查找和多边形内查找等。

MongoDB支持两种类型的地理空间索引：

1. 2D索引：用于平面坐标系，适用于简单的两维地理数据。
2. 2dsphere索引：用于球面坐标系，适用于地球表面的地理数据。

为什么要使用地理空间索引？

地理空间索引用于以下场景：

1. 距离查询：查找离某个点最近的N个位置。
2. 范围查询：查找在某个范围内的所有点，如查找某个多边形内的点。
3. 邻近查询：查找在某个点附近的所有点。

创建地理空间索引

你可以使用MongoDB Shell或编程语言（如Node.js、Python和Java）来创建地理空间索引。以下是详细的步骤和代码示例。

创建2D地理空间索引

在MongoDB Shell中创建地理空间索引

1. 启动MongoDB Shell

首先，打开终端或命令提示符，启动MongoDB Shell：

mongo

2. 选择数据库

选择你要创建索引的数据库：

use myDatabase

3. 创建2D地理空间索引

使用 createIndex 方法对集合中的字段创建2D地理空间索引：

db.myCollection.createIndex({ location: "2d" })

创建2dsphere地理空间索引

在MongoDB Shell中创建地理空间索引

1. 启动MongoDB Shell

首先，打开终端或命令提示符，启动MongoDB Shell：

mongo

2. 选择数据库

选择你要创建索引的数据库：

use myDatabase

3. 创建2dsphere地理空间索引

使用 createIndex 方法对集合中的字段创建2dsphere地理空间索引：

db.myCollection.createIndex({ location: "2dsphere" })

使用Node.js创建地理空间索引

1. 安装MongoDB Node.js驱动

在终端中运行以下命令来安装MongoDB的Node.js驱动：

npm install mongodb

2. 创建并运行Node.js脚本

创建一个新的Node.js脚本文件（如 createGeospatialIndex.js）并添加以下代码：

const { MongoClient } = require('mongodb');

async function main() {
    const uri = "mongodb://localhost:27017";
    const client = new MongoClient(uri, { useUnifiedTopology: true });

    try {
        // 连接到MongoDB服务器
        await client.connect();
        console.log("Connected to MongoDB");

        // 选择数据库
        const db = client.db('myDatabase');

        // 选择集合
        const collection = db.collection('myCollection');

        // 创建2D地理空间索引
        await collection.createIndex({ location: "2d" });
        console.log('Created 2D geospatial index on location');

        // 创建2dsphere地理空间索引
        await collection.createIndex({ location: "2dsphere" });
        console.log('Created 2dsphere geospatial index on location');

    } finally {
        // 关闭连接
        await client.close();
    }
}

main().catch(console.error);

运行这个脚本：

node createGeospatialIndex.js

使用Python创建地理空间索引

1. 安装PyMongo

在终端中运行以下命令来安装PyMongo：

pip install pymongo

2. 创建并运行Python脚本

创建一个新的Python脚本文件（如 create_geospatial_index.py）并添加以下代码：

from pymongo import MongoClient

def main():
    client = MongoClient('mongodb://localhost:27017/')

    # 选择数据库
    db = client['myDatabase']

    # 选择集合
    collection = db['myCollection']

    # 创建2D地理空间索引
    collection.create_index([('location', '2d')])
    print('Created 2D geospatial index on location')

    # 创建2dsphere地理空间索引
    collection.create_index([('location', '2dsphere')])
    print('Created 2dsphere geospatial index on location')

    # 关闭连接
    client.close()

if __name__ == '__main__':
    main()

运行这个脚本：

python create_geospatial_index.py

使用Java创建地理空间索引

1. 添加MongoDB Java驱动依赖

如果你使用的是Maven项目，添加以下依赖到你的 pom.xml 文件中：

<dependency>
    <groupId>org.mongodb</groupId>
    <artifactId>mongodb-driver-sync</artifactId>
    <version>4.4.0</version>
</dependency>

2. 创建Java类并添加代码

创建一个新的Java类文件（如 CreateGeospatialIndex.java）并添加以下代码：

import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

public class CreateGeospatialIndex {
    public static void main(String[] args) {
        // 连接到MongoDB服务器
        MongoClient mongoClient = new MongoClient("localhost", 27017);

        // 选择数据库
        MongoDatabase database = mongoClient.getDatabase("myDatabase");

        // 选择集合
        MongoCollection<Document> collection = database.getCollection("myCollection");

        // 创建2D地理空间索引
        collection.createIndex(new Document("location", "2d"));
        System.out.println("Created 2D geospatial index on location");

        // 创建2dsphere地理空间索引
        collection.createIndex(new Document("location", "2dsphere"));
        System.out.println("Created 2dsphere geospatial index on location");

        // 关闭连接
        mongoClient.close();
    }
}

编译并运行这个Java类：

javac -cp .:path/to/mongodb-driver-sync-4.4.0.jar CreateGeospatialIndex.java
java -cp .:path/to/mongodb-driver-sync-4.4.0.jar CreateGeospatialIndex

使用地理空间索引的查询示例

创建地理空间索引后，你可以通过以下查询来利用这个索引：

查找邻近点（$near）

在MongoDB Shell中查询

db.myCollection.find({
    location: {
        $near: {
            $geometry: {
                type: "Point",
                coordinates: [longitude, latitude]
            },
            $maxDistance: 1000 // 最大距离，单位为米
        }
    }
})

在Node.js中查询

const query = {
    location: {
        $near: {
            $geometry: {
                type: "Point",
                coordinates: [longitude, latitude]
            },
            $maxDistance: 1000 // 最大距离，单位为米
        }
    }
};
const result = await collection.find(query).toArray();
console.log(result);

在Python中查询

query = {
    'location': {
        '$near': {
            '$geometry': {
                'type': 'Point',
                'coordinates': [longitude, latitude]
            },
            '$maxDistance': 1000  # 最大距离，单位为米
        }
    }
}
result = collection.find(query)
for doc in result:
    print(doc)

在Java中查询

import static com.mongodb.client.model.Filters.*;
import static com.mongodb.client.model.Indexes.*;

FindIterable<Document> result = collection.find(near("location", 
    new Point(new Position(longitude, latitude)), 1000.0, 0.0));
for (Document doc : result) {
    System.out.println(doc.toJson());
}

查找在多边形内的点（$geoWithin）

在MongoDB Shell中查询

db.myCollection.find({
    location: {
        $geoWithin: {
            $geometry: {
                type: "Polygon",
                coordinates: [[
                    [lng1, lat1], [lng2, lat2], [lng3, lat3], [lng4, lat4], [lng1, lat1]
                ]]
            }
        }
    }
})

在Node.js中查询

const query = {
    location: {
        $geoWithin: {
            $geometry: {
                type: "Polygon",
                coordinates: [[
                    [lng1, lat1], [lng2, lat2], [lng3, lat3], [lng4, lat4], [lng1, lat1]
                ]]
            }
        }
    }
};
const result = await collection.find(query).toArray();
console.log(result);