依赖
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-lang3</artifactId>
<version>3.11</version>
</dependency>
代码
package com.my.controller;
import com.my.entity.Shop;
import com.my.service.ShopService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.geo.Distance;
import org.springframework.data.geo.GeoResults;
import org.springframework.data.geo.Metrics;
import org.springframework.data.geo.Point;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.CriteriaDefinition;
import org.springframework.data.mongodb.core.query.NearQuery;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.web.bind.annotation.*;
import java.math.BigDecimal;
import java.util.List;
import java.util.regex.Pattern;
/**
* @author hutf
* @createTime 2021年07月11日 19:23:00
*/
@RestController
@RequestMapping(value = "mongo")
public class MongoController {
@Autowired
private MongoTemplate mongoTemplate;
@Autowired
private ShopService shopService;
@PostMapping("/add")
public String add(@RequestBody Shop shop) {
mongoTemplate.save(shop);
return "OK";
}
@GetMapping("del")
public String del(String id) {
Query query = new Query();
query.addCriteria(Criteria.where("id").is(id));
mongoTemplate.remove(query, Shop.class);
return "OK";
}
@PostMapping("update")
public String update() {
return "OK";
}
@GetMapping("findList")
public List<Shop> findList() {
return mongoTemplate.findAll(Shop.class);
}
@GetMapping("getPageList")
public List<Shop> getPageList() {
Shop shop = new Shop();
// shop.setCapital(21000);
// shop.setType(1);
shop.setName("下");
List<Shop> list = null;
Query query = new Query();
if (shop.getCapital() != null) {// 范围查找
query.addCriteria(Criteria.where("capital").lt(shop.getCapital()));
}
if (shop.getType() != null) {// 精准查询
query.addCriteria(Criteria.where("type").is(shop.getType()));
}
if (StringUtils.isNotBlank(shop.getName())) {// 模糊查询
Pattern pattern = Pattern.compile("" + shop.getName() + "$", Pattern.CASE_INSENSITIVE);
query.addCriteria(Criteria.where("name").regex(".*"+ shop.getName() +".*"));
}
//
// List<BigDecimal> loc = shop.getLoc();
// if (loc != null && loc.size() == 2) {
// query.addCriteria(Criteria.where())
// }
// 排序
query.with(Sort.by("capital").ascending());
// 分页
query.limit(5);
query.skip(0);
List<Shop> resultList = mongoTemplate.find(query, Shop.class, "shop");
return resultList;
}
/**
* 参考文章 https://www.cnblogs.com/wenbronk/p/7217802.html
* @return
*/
@GetMapping("/distance")
public GeoResults<Shop> distanceList() {
// 当前位置 金铂广场 (113.281218 23.238349) 距离到嘉禾望岗地铁口(113.287727,23.236857) 一公里左右
// 距离 嘉禾广场(113.573161, 23.281602)27.004公里
// 据 王钢小学 0.70 公里
Point location = new Point(113.281218 , 23.238349);
NearQuery query = NearQuery.near(location).maxDistance(new Distance(10, Metrics.KILOMETERS));// 单位公里
GeoResults<Shop> geoResults = mongoTemplate.geoNear(query, Shop.class);
return geoResults;
}
@GetMapping("/distanceV2")
public GeoResults<Store> distanceV2(Double d) {
Point currentPoint = new Point(113.234819,23.154207);
// 方式二
NearQuery query = NearQuery.near(currentPoint).maxDistance(new Distance(d, Metrics.KILOMETERS));// 单位公里
GeoResults<Store> geoResults = mongoTemplate.geoNear(query, Store.class);
return geoResults;
}
}