SpringBoot使用mongoTemplate

1.配置文件

POM文件

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>

YML文件

spring:
  data:
    mongodb:
      uri: mongodb://localhost:27017 #连接的地址
      database: curd #数据库

2.常用方法

mongoTemplate.insert(Class)新增
mongoTemplate.remove(Query, Class)删除
mongoTemplate.findAll(Class)查询全部数据
mongoTemplate.findById(<id>,Class)根据ID查询数据
mongoTemplate.find(Query, Class)根据Query条件查询
mongoTemplate.upsert(Query, Update, Class)修改,当查找不到对应的行时,会进行insert一条新记录,查找到时,只修改查找到的第一行
mongoTemplate.updateFirst(Query, Update, Class)修改查找到的第一行
mongoTemplate.updateMulti(Query, Update, Class)修改多行
mongoTemplate.aggregate(Aggregation, Class, Class)聚合查询

Query对象
static query(CriteriaDefinition)设置查询标准,并创建Query对象
skip(long)设置跳过document数量,在返回结果集之前
limit(int)限制返回document的数量
with(Sort sort)排序
with(Pageable)分页方式返回数据
案例

//获取第一页的10条数据,根据username,降序排序
PageRequest pageRequest = PageRequest.of(0, 10, Sort.by(Sort.Direction.DESC, "username"));
//查询age>=18的数据
Query query = Query.query(Criteria.where("age").gte(18)).with(pageRequest);
List<User> list = userDAO.list(query);
log.info("length: {} ,listTest: {}", list.size(), list); //length: 10 ,listTest:...

Criteria对象
static where(String key)根据key进行查询,并创建Criteria对象
is(Object)等于
ne(Object)不等于
lt(Object)小于
lte(Object)小于等于
gt(Object)大于
gte(Object)大于等于
ne(Object)不等于
in(Object...)包含
in(Collection)包含
nin(Object...)不包含
nin(Collection)不包含
mod(Number value, Number remainder)取模value等于remainder的数据
all(Object...)同时包含,与in的区别就是,in是查询某个字段的值,all是查询某个数组类型字段是否都包含
all(Collection)同时包含

//查询address字段同时包含"深圳市","珠海市"的数据
Query query = Query.query(Criteria.where("address").all(Arrays.asList("深圳市","珠海市")));

size(int)查询数组类型字段数量匹配的数据

//查询address数量为2的用户数据
db.user.find({'address':{$size:2}})
Query query = Query.query(Criteria.where("address").size(2));

exists(boolean)是否存在该字段

//查询存在address字段的数据
Query query = Query.query(Criteria.where("address").exists(true));

type(int)查询字段是对应类型的数据

Type Number Alias Notes
Double 1 “double”
String 2 “string”
Object 3 “object”
Array 4 “array”
Binary data 5 “binData”
Undefined 6 “undefined” Deprecated.
ObjectId 7 “objectId”
Boolean 8 “bool”
Date 9 “date”
Null 10 “null”
Regular Expression 11 “regex”
DBPointer 12 “dbPointer” Deprecated.
JavaScript 13 “javascript”
Symbol 14 “symbol” Deprecated.
JavaScript (with scope) 15 “javascriptWithScope”
32-bit integer 16 “int”
Timestamp 17 “timestamp”
64-bit integer 18 “long”
Decimal128 19 “decimal” New in version 3.4.
Min key -1 “minKey”
Max key 127 “maxKey”
//查询address字段的数据类型为数组的数据
Query query = Query.query(Criteria.where("address").type(4));

type(Type... types)查询字段是对应类型的数据
not(Object)取反
regex(String)正则
regex(Pattern)正则

3.使用

model

@Document(collection = "user")
@Data
public class User implements Serializable {

    private static final long serialVersionUID = 4349568198872630000L;
    @Id
    private String id;

    private String username;

    private String password;

    @Field("nick_name")
    private String nickName;

    private Integer age;

    private List<String> address;
}

dao

@Repository
public class BaseDAO<T> {

    private Class<T> clazz;

    @Autowired
    MongoTemplate mongoTemplate;

    public BaseDAO() {
        clazz = getSuperClassGenericType();
    }

    public T insert(T t) {
        T insert = mongoTemplate.insert(t);
        return insert;
    }

    public List<T> insertAll(List<T> ts) {
        List<T> collection = (List<T>) mongoTemplate.insertAll(ts);
        return collection;
    }


    public List<T> list() {
        Query query = new Query();
        List<T> all = mongoTemplate.findAll(clazz);
        return all;
    }

    public List<T> list(Query query) {
        List<T> all = mongoTemplate.find(query, clazz);
        return all;
    }

    public T getById(String id) {
        T byId = mongoTemplate.findById(id, clazz);
        return byId;
    }

    public UpdateResult upsert(Query query, UpdateDefinition updateDefinition) {
        UpdateResult upsert = mongoTemplate.upsert(query, updateDefinition, clazz);
        return upsert;
    }

    public UpdateResult updateFirst(Query query, UpdateDefinition updateDefinition) {
        UpdateResult upsert = mongoTemplate.updateFirst(query, updateDefinition, clazz);
        return upsert;
    }

    public UpdateResult updateMulti(Query query, UpdateDefinition updateDefinition) {
        UpdateResult upsert = mongoTemplate.updateMulti(query, updateDefinition, clazz);
        return upsert;
    }

    public DeleteResult remove(Query query) {
        DeleteResult deleteResult = mongoTemplate.remove(query, clazz);
        return deleteResult;
    }

    public List<T> aggregate(Aggregation aggregation) {
        List<T> aggregate = aggregate(aggregation, clazz);
        return aggregate;
    }

    public <R> List<R> aggregate(Aggregation aggregation, Class<R> outputClass) {
        List<R> aggregate = aggregate(aggregation, clazz, outputClass);
        return aggregate;
    }

    public <R> List<R> aggregate(Aggregation aggregation, Class inputClass, Class<R> outputClass) {
        AggregationResults<R> aggregate = mongoTemplate.aggregate(aggregation, inputClass, outputClass);
        List<R> mappedResults = aggregate.getMappedResults();
        return mappedResults;
    }

    private Class getSuperClassGenericType() {
        Type type = this.getClass().getGenericSuperclass();
        if (type instanceof ParameterizedType) {
            ParameterizedType parameterizedType = (ParameterizedType) type;
            clazz = (Class<T>) parameterizedType.getActualTypeArguments()[0];
        }
        return clazz;
    }

}


@Repository
public class UserDAO extends BaseDAO<User> {

}

测试用例

@SpringBootTest
@Slf4j
public class UserDAOTest {

    @Autowired
    UserDAO userDAO;

    @Test
    public void insertTest() {
        User user = new User();
        user.setAge(18);
        user.setNickName("超级好看");
        user.setUsername("张三");
        user.setPassword("123456");
        User insert = userDAO.insert(user);
        log.info("insertTest: {}", insert);
    }

    @Test
    public void insertAllTest() {
        List<User> collect = IntStream.range(0, 100).mapToObj(num -> {
            User user = new User();
            user.setAge(18);
            user.setNickName("超级好看");
            user.setUsername("张三" + num);
            user.setPassword("123456");
            return user;
        }).collect(Collectors.toList());
        userDAO.insertAll(collect);
    }

    @Test
    public void listTest() {
        List<User> list = userDAO.list();
        log.info("listTest: {}", list);
    }

    @Test
    public void listTest2() {
        Query query = new Query(Criteria.where("age").gte(18));
        query.with(PageRequest.of(0, 10, Sort.by(Sort.Direction.DESC, "username")));
        List<User> list = userDAO.list(query);
        log.info("length: {} ,listTest: {}", list.size(), list);
    }

    @Test
    public void listTest3() {
        PageRequest pageRequest = PageRequest.of(0, 10, Sort.by(Sort.Direction.DESC, "username"));
        Query query = Query.query(Criteria.where("age").gte(18)).with(pageRequest).limit(5);
        List<User> list = userDAO.list(query);
        log.info("length: {} ,listTest: {}", list.size(), list);
    }

    @Test
    public void listTest4() {
        Query query = Query.query(Criteria.where("address").size(2));
        List<User> list = userDAO.list(query);
        log.info("length: {} ,listTest: {}", list.size(), list);
    }

    @Test
    public void listTest5() {
        Query query = Query.query(Criteria.where("address").all(Arrays.asList("深圳市", "珠海市")));
        List<User> list = userDAO.list(query);
        log.info("length: {} ,listTest: {}", list.size(), list);
    }

    @Test
    public void getByIdTest() {
        User byId = userDAO.getById("5f03f180e7d7c8591deac8f9");
        log.info("getByIdTest: {}", byId);
    }

    @Test
    public void upsertTest() {
        UpdateResult updateResult = userDAO.upsert(Query.query(Criteria.where("username").is("何一")), Update.update("age", 19));
        log.info("upsertTest: {}", updateResult);
    }

    @Test
    public void updateFirstTest() {
        UpdateResult updateResult = userDAO.updateFirst(Query.query(Criteria.where("username").is("许二")), Update.update("age", 19));
        log.info("updateFirstTest: {}", updateResult);
    }

    @Test
    public void updateMultiTest() {
        UpdateResult updateResult = userDAO.updateMulti(Query.query(Criteria.where("age").is(19)), Update.update("age", 18));
        log.info("updateMultiTest: {}", updateResult);
    }

    @Test
    public void removeTest() {
        DeleteResult remove = userDAO.remove(Query.query(Criteria.where("age").is(18)));
        log.info("removeTest: {}", remove);
    }

    @Test
    public void aggregateTest() {
        //获取以好看结尾的nickName,age>=16和age<=18的数据,然后以id进行分组,将username添加到username,nickName添加到nickName,同时统计条数
        List<User> aggregate = userDAO.aggregate(Aggregation.newAggregation(
                Aggregation.match(Criteria.where("nickName").regex("好看$")
                        .andOperator(Criteria.where("age").gte(16)
                                , Criteria.where("age").lte(18))),
                Aggregation.group("id").push("username").as("username").push("nickName").as("nick_name").count().as("count")
        ));
        log.info("aggregateTest: {}", aggregate);
    }

}

4.小练习

model

@Document(collection = "favorite_unit ")
@Data
public class FavoriteUnit implements Serializable {

    private static final long serialVersionUID = 1L;
    @Id
    private String id;

    private String agentId; //代理人ID

    private String unitId; //单元ID

    private LocalDateTime createTime;//创建时间

}

题目:获取上个月某个代理人添加的unitId列表和对应unitId的添加次数

答案

    @Test
    public void aggregateTest() {
        LocalDateTime localDateTime = LocalDateTime.now().minusMonths(1);
        LocalDateTime firstDayOfMonth = localDateTime.with(TemporalAdjusters.firstDayOfMonth());
        firstDayOfMonth = LocalDateTime.of(firstDayOfMonth.toLocalDate(), LocalTime.MIN);
        LocalDateTime lastDayOfMonth = localDateTime.with(TemporalAdjusters.lastDayOfMonth());
        lastDayOfMonth = LocalDateTime.of(lastDayOfMonth.toLocalDate(), LocalTime.MAX);
        //统计某个agentId,上个月开始时间到结束时间的数据,然后只获取"agentId", "unitId", "createTime"字段,最后根据unitId分组,并统计条数
        List<User> aggregate = userDAO.aggregate(Aggregation.newAggregation(
                Aggregation.match(Criteria.where("agentId").is("5c1cacc718aad600067b7aa0")
                        .andOperator(Criteria.where("createTime").gte(firstDayOfMonth)
                                , Criteria.where("createTime").lte(lastDayOfMonth))),
                Aggregation.project("agentId", "unitId", "createTime"),
                Aggregation.group("unitId").count().as("count")
        ));
        log.info("aggregateTest: {}", aggregate);
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,616评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,020评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,078评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,040评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,154评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,265评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,298评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,072评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,491评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,795评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,970评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,654评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,272评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,985评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,815评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,852评论 2 351