Spring MongoDB实现LBS功能的其中一种实现

1.pom.xml 增加依赖

<dependency>
    <groupId>org.mongodb</groupId>
    <artifactId>mongo-java-driver</artifactId>
    <version>3.4.2</version>
</dependency>
<dependency>
    <groupId>org.springframework.data</groupId>
    <artifactId>spring-data-mongodb</artifactId>
    <version>1.10.7.RELEASE</version>
</dependency>

2.MongoDB 连接配置

#mongoDB连接配置
mongo.hostport=127.0.0.1:27017
mongo.connectionsPerHost=8
mongo.threadsAllowedToBlockForConnectionMultiplier=4
#连接超时时间
mongo.connectTimeout=1000
#等待时间
mongo.maxWaitTime=1500
mongo.autoConnectRetry=true
mongo.socketKeepAlive=true
#Socket超时时间
mongo.socketTimeout=1500
mongo.slaveOk=true

3.spring-mongodb配置

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:context="http://www.springframework.org/schema/context"
       xmlns:mongo="http://www.springframework.org/schema/data/mongo"
       xsi:schemaLocation="http://www.springframework.org/schema/context
        http://www.springframework.org/schema/context/spring-context-3.0.xsd
        http://www.springframework.org/schema/data/mongo
        http://www.springframework.org/schema/data/mongo/spring-mongo-1.10.2.xsd
        http://www.springframework.org/schema/beans
        http://www.springframework.org/schema/beans/spring-beans-3.0.xsd">


    <context:property-placeholder location="classpath:conf/mongodbconfig.properties"/>
    <mongo:mongo-client id="mongo" replica-set="${mongo.hostport}">
        <!-- 一些连接属性的设置 -->
        <mongo:client-options
                connections-per-host="${mongo.connectionsPerHost}"
                threads-allowed-to-block-for-connection-multiplier="${mongo.threadsAllowedToBlockForConnectionMultiplier}"
                connect-timeout="${mongo.connectTimeout}"
                max-wait-time="${mongo.maxWaitTime}"
                socket-keep-alive="${mongo.socketKeepAlive}"
                socket-timeout="${mongo.socketTimeout}"/>
    </mongo:mongo-client>
    <!-- 设置使用的数据库 名-->
    <mongo:db-factory id="mongoDbFactory" dbname="lbs" mongo-ref="mongo"/>
    <!-- mongodb的模板 -->
    <bean id="mongoTemplate" class="org.springframework.data.mongodb.core.MongoTemplate">
        <constructor-arg name="mongoDbFactory" ref="mongoDbFactory"/>
    </bean>

</beans>

在spring主文件里引入spring-mongodb.xml

<import resource="classpath:spring/spring-mongodb.xml"/>

4.mongodb 添加索引

我的数据格式

{
    "loc": {
        "type": "Point",
        "coordinates": [
            151.91518839682385,
            17.140996455288267
        ]
    },
    "playerId":  2869,
    "name": "name",
    "createTime": 1511921687
    }
}

目前项目中使用的是2dsphere模型

输入命令 mongo 进入 use lbs

查看目前索引：

 db.lbs.getIndexes() 可以看到目前只有_id的主见索引

[
{
"v" : 2,
"key" : {
"_id" : 1
},
"name" : "_id_",
"ns" : "lbs.lbs"
}
]

建立索引 ：

db.lbs.ensureIndex({loc:"2dsphere"})

{
"createdCollectionAutomatically" : false,
"numIndexesBefore" : 1,
"numIndexesAfter" : 2,
"ok" : 1
}
然后执行：

db.lbs.getIndexes() 

[
{
"v" : 2,
"key" : {
"_id" : 1
},
"name" : "_id_",
"ns" : "lbs.lbs"
},
{
"v" : 2,
"key" : {
"loc" : "2dsphere"
},
"name" : "loc_2dsphere",
"ns" : "lbs.lbs",
"2dsphereIndexVersion" : 3
}
]

可以看到增加了loc这个key的索引

5.上代码

5.1 插入方法

public boolean createLocation(PlayerInfoPO playerInfoPO, String location) {
    long time = System.currentTimeMillis();
    log.info("MongoDB插入位置开始；id：{},name：{},location：{}", playerInfoPO.getId(), playerInfoPO.getNickname(), location);
    DBObject positionInfo = new BasicDBObject();
    DBObject loc = new BasicDBObject();
    loc.put("type", "Point");
    String[] locationStrArray = location.split(",");
    double[] locationDoubleArray = StringUtils.isEmpty(location) ? new double[]{90.0, 90.0} : new double[]{Double.parseDouble(locationStrArray[0]), Double.parseDouble(locationStrArray[1])};
    loc.put("coordinates", locationDoubleArray);
    positionInfo.put("loc", loc);
    positionInfo.put("playerId", playerInfoPO.getId());
    positionInfo.put("name", playerInfoPO.getNickname());
    positionInfo.put("createTime", System.currentTimeMillis() / 1000);
    mongoTemplate.insert(positionInfo, "lbs");

    //获取_id
    playerInfoPO.setMapId(positionInfo.get("_id").toString());    playerInfoPOMapper.updateByPrimaryKey(playerInfoPO);    log.info("MongoDB插入位置结束" + (System.currentTimeMillis() - time));    return true;}

5.2 检索周边 分页查询

/**
 * 获取周边
 *
 * @param center 坐标
 * @param radius 查询半径 米
 * @param limit  每页条数
 * @param page   第几页
 * @return
 */
public Map<String, String> getAroundByCenter(Long playerId, String center, double radius, int limit, int page) {
    log.info("MongoDB获取周边开始；playerId：{},center：{},radius：{},limit：{},page：{}", playerId, center, radius, limit, page);
    long time = System.currentTimeMillis();
    Map<String, String> result = new HashMap<>(3);
    String[] centerArray = center.split(",");

    DBCollection dbCollection = mongoTemplate.getCollection("lbs");
    DBObject query = new BasicDBObject();

    query.put("loc",
            new BasicDBObject("$nearSphere",
                    new BasicDBObject("$geometry",
                            new BasicDBObject("type", "Point")
                                    .append("coordinates", new double[]{Double.parseDouble(centerArray[0]), Double.parseDouble(centerArray[1])}))
                            .append("$minDistance", 0)
                            .append("$maxDistance", radius)
            )
    );
    long count = dbCollection.count(query);
    List<DBObject> dbObjectList = dbCollection.find(query).limit(limit).skip((page - 1) * limit).toArray();

    result.put("count", String.valueOf(count));
    result.put("datas", JSON.toJSONString(dbObjectList));
    log.info("MongoDB获取周边结束" + (System.currentTimeMillis() - time));
    log.info("MongoDB获取周边结束" + JSON.toJSONString(result));

    return result;
}

删改方法网上一大堆，不再啰嗦