MongoDB + SpringBoot 的基础CRUD、聚合查询

1、数据准备

1.1、springboot导包

• springboot版本：2.7.10
  

点击查看代码

1.         
2.         <dependency>
3.             <groupId>org.springframework.boot</groupId>
4.             <artifactId>spring-boot-starter-data-mongodb</artifactId>
5.         </dependency>
6.         
7.         <dependency>
8.             <groupId>com.alibaba</groupId>
9.             <artifactId>fastjson</artifactId>
10.             <version>1.2.83</version>
11.         </dependency>

复制代码

1.2、配置文件yml

点击查看代码

1. spring:
2.   data:
3.     mongodb:
4.       # 有账号密码
5.       uri: mongodb://账户名:密码@IP:端口/数据库
6.       # 无账号密码
7.       # uri: mongodb://IP:端口/数据库

复制代码

1.3、实体类

• 也可以不使用实体类来存数据、查询数据
  

点击查看代码

1. package com.cc.mdb.entity;
3. import lombok.Data;
4. import org.springframework.data.annotation.Id;
5. import org.springframework.data.mongodb.core.mapping.Document;
6. import org.springframework.data.mongodb.core.mapping.Field;
8. import java.time.LocalDateTime;
10. /**
11. * @Document("User") 存入的时候不取对象名，默认用User当做表名
12. */
13. @Data
14. @Document("User")
15. public class User {
17.     @Id
18.     private String id;
19.     /**
20.      * 使用name当做字段key
21.      */
22.     @Field("name")
23.     private String name;
24.     /**
25.      * 使用age1当做字段key
26.      */
27.     @Field("age1")
28.     private Integer age;
29.     /**
30.      * 使用email当做字段key
31.      */
32.     @Field("email")
33.     private String email;
34.     /**
35.      * 使用createDate当做字段key
36.      */
37.     @Field("createDate")
38.     private LocalDateTime createDate = LocalDateTime.now();
39. }

复制代码

2、基础CRUD

• 参考：https://blog.csdn.net/weixin_44185213/article/details/125293170?spm=1001.2014.3001.5502
  
• 注入：SpringBoot中使用 MongoTemplate 来操作MongoDB
  

点击查看代码

1. package com.cc.mdb.test;
3. import com.alibaba.fastjson.JSONObject;
4. import com.cc.mdb.BaseTest;
5. import com.cc.mdb.entity.User;
6. import com.mongodb.client.result.UpdateResult;
7. import org.junit.Test;
8. import org.springframework.data.domain.PageRequest;
9. import org.springframework.data.domain.Pageable;
10. import org.springframework.data.domain.Sort;
11. import org.springframework.data.mongodb.core.MongoTemplate;
12. import org.springframework.data.mongodb.core.query.Criteria;
13. import org.springframework.data.mongodb.core.query.Query;
14. import org.springframework.data.mongodb.core.query.Update;
16. import javax.annotation.Resource;
17. import java.util.ArrayList;
18. import java.util.Collection;
19. import java.util.List;
20. import java.util.regex.Pattern;
22. /** CRUD
23. *  查询使用：Query
24. *  博客：https://blog.csdn.net/weixin_44185213/article/details/125293170?spm=1001.2014.3001.5502
25. * @author CC
26. * @since 2023/3/30 0030
27. */
28. @SpringBootTest
29. @RunWith(SpringRunner.class)
30. public class TestMongoQuery{
32.     @Resource
33.     private MongoTemplate mongoTemplate;
35.     //写入简单数据 —— 根据对象存数据
36.     @Test
37.     public void test01()throws Exception{
38.         List<User> vos = new ArrayList<>();
39.         for (int i = 10; i < 20; i++) {
40.             User user = new User();
41.             String aa = i + "" + i + "" + i + "";
42.             user.setId(aa);
43.             user.setName("cscs"+aa);
44.             user.setAge(123);
45.             user.setEmail("11@qq.com");
46.             vos.add(user);
47.         }
48.         Collection<User> user2 = mongoTemplate.insert(vos, "User2");
49.         System.out.println(user2);
50.     }
52.     //写入简单数据 —— 根据json存数据（JSONObject(Fastjson)）
53.     @Test
54.     public void test02()throws Exception{
55.         List<JSONObject> vos = new ArrayList<>();
56.         for (int i = 1; i < 20; i++) {
57.             JSONObject result = new JSONObject();
58.             String aa = i + "" + i + "" + i + "";
59.             result.put("id",aa);
60.             result.put("name","cscs"+aa);
61.             result.put("age",123);
62.             result.put("email","11@qq.com");
63.             vos.add(result);
64.         }
65.         Collection<JSONObject> user3 = mongoTemplate.insert(vos, "User3");
66.         System.out.println(user3);
67.     }
69.     //写入复杂数据 —— 有子文档（json数据中有下级）
70.     @Test
71.     public void test013()throws Exception{
72.         List<JSONObject> vos = new ArrayList<>();
73.         for (int i = 1; i < 20; i++) {
74.             JSONObject result = new JSONObject();
75.             String aa = i + "" + i + "" + i + "";
76.             result.put("id",aa);
77.             result.put("name","cscs"+aa);
78.             result.put("age",123);
79.             result.put("email","11@qq.com");
81.             List<JSONObject> jss = new ArrayList<>();
82.             for (int j = 0; j < 2; j++) {
83.                 JSONObject js = new JSONObject();
84.                 js.put("sId",j);
85.                 js.put("sName","学校" + aa);
86.                 js.put("sNo","学校编号" + aa);
87.                 js.put("sTab",1);
89.                 List<JSONObject> whats = new ArrayList<>();
90.                 if (i == 1){
91.                     for (int k = 0; k < 2; k++) {
92.                         JSONObject kk = new JSONObject();
93.                         kk.put("kId",k);
94.                         kk.put("kName","k"+k);
95.                         whats.add(kk);
96.                     }
97.                 }
98.                 js.put("sWhat",whats);
99.                 jss.add(js);
100.             }
101.             result.put("school",jss);
102.             vos.add(result);
103.         }
104.         Collection<JSONObject> user3 = mongoTemplate.insert(vos, "User4");
105.         System.out.println(user3);
106.     }
108.     //查找全部
109.     @Test
110.     public void test03()throws Exception{
111.         List<User> users = mongoTemplate.findAll(User.class);
112.         System.out.println(users);
113.     }
115.     /** 根据json里面的字段查询
116.      * 查询 school.sWhat.kId 的数据
117.      */
118.     @Test
119.     public void test07()throws Exception{
120.         //查询的新增方法：test013
122.         Criteria cr = new Criteria();
123. //        cr.and("id").is("111");
124. //        cr.and("school.sId").is(0);
125.         cr.and("school.sWhat.kId").is(0);
126.         Query query = new Query();
127.         query.addCriteria(cr);
129.         // 执行查询并获取结果
130.         List<JSONObject> jsonObjects = mongoTemplate.find(query, JSONObject.class, "User4");
131.         System.out.println(jsonObjects);
132.     }
134.     //**高级查询：使用Query
135.     @Test
136.     public void test04()throws Exception{
137.         String search = "cscs";
138.         int page = 1;
139.         int size = 20;
140.         Query query = new Query();
141.         //一、where
142.         //1模糊
143.         String format = String.format("^.*%s.*$", search);
144.         Pattern pattern = Pattern.compile(format, Pattern.CASE_INSENSITIVE);
145.         Criteria criteria = Criteria.where("name").regex(pattern);
146.         //模糊查询方式2（推荐）
147. //        Criteria.where("name").regex(String.format(".*%s.*","查询的值"));
149.         //2等于（这里比较特殊，value：数字的必须传入数字的、字符串必须传入字符串的）
150.         criteria.and("age").is(123);
151.         query.addCriteria(criteria);
153.         //二、排序。按id升序
154.         Sort sort = Sort.by(Sort.Direction.ASC, "_id");
155.         query.with(sort);
157.         //二、总条数（是所有条件都弄了之后）
158.         long totalCount = mongoTemplate.count(query, JSONObject.class, "User3");
160.         //三、分页（必须在where和查询总条数之后）
161.         //方式1
162. //        query.skip((page - 1) * size).limit(size);
163.         //方式2（推荐）
164.         Pageable pageable = PageRequest.of(page - 1, size);
165.         query.with(pageable);
167.         //查询。由于查询出来的是key-value的形式，可以使用JSONObject(FastJson)来接收参数。如果有对象（User），可以直接使用对象来接收
168.         List<JSONObject> users = mongoTemplate.find(query, JSONObject.class, "User3");
170.         System.out.println("总条数：" + totalCount);
171.         System.out.println(users);
172.     }
174.     //修改 - 根据表名+id 修改其中一个json中的某些值
175.     @Test
176.     public void test05() {
177.         JSONObject user2 = mongoTemplate.findById("6433c76df108474b788cc4ce", JSONObject.class, "User3");
178.         Object id = user2.get("_id");
180.         Query query = new Query(Criteria.where("_id").is(id));
182.         Update update = new Update();
183.         update.set("name", "修改后的名字2");
184.         update.set("age", "321");
185. //        update.set("email", "");
187.         //方式1：upsert方法更新与查询条件匹配的第一个文档，如果没有文档与查询条件匹配，则插入一个新文档。
188. //        UpdateResult result = mongoTemplate.upsert(query, update, "User3");
189. //        long count = result.getModifiedCount();
190. //        System.out.println(count);
192.         //方式2：update方法更新与查询条件匹配的第一个文档
193.         UpdateResult user3 = mongoTemplate.updateFirst(query, update, "User3");
194.         //修改计数
195.         long modifiedCount = user3.getModifiedCount();
196.         System.out.println(modifiedCount);
197.         //匹配计数
198.         long matchedCount = user3.getMatchedCount();
199.         System.out.println(matchedCount);
201.         //方式3：updateMulti方法更新与查询条件匹配的所有文档。
202. //        UpdateResult user31 = mongoTemplate.updateMulti(query, update, "User3");
203.     }
205.     //删除 - 根据数据id删除
206.     @Test
207.     public void test06() {
208.         //1删除文档中的某一条数据
209. //        Query query = new Query(Criteria.where("_id").is("6433c76df108474b788cc4cf"));
210. //        DeleteResult result = mongoTemplate.remove(query, "User3");
211. //        long count = result.getDeletedCount();
212. //        System.out.println(count);
214.         //2删除整个文档中的数据 —— 相当于drop清空表中的数据
215. //        DeleteResult user4 = mongoTemplate.remove(new Query(), "User4");
216. //        long deletedCount = user4.getDeletedCount();
217. //        System.out.println(deletedCount);
219.         //3删除文档 —— 相当于删除表
220.         mongoTemplate.dropCollection("User4");
221.     }
225. }

复制代码

3、聚合查询

• 聚合意思：自己见管网
  

3.1、聚合查询关键字

1. *      $project            指定返回字段
2. *      $lookup                    多表关联
3. *      $unwind                    数组拆分
4. *      $match                    筛选条件
5. *      $sort                    排序
6. *      $limit                    分页
7. *      $skip                    跳过
8. *      $group                    根据id(可自定义)分组
9. *      $sum                    分组后求和
10. *      $first                    分组后取第一条
11. *      $last                    分组后取最后一条
12. *      $addToSet           分组后添加到list
13. *      $cond                    可以做 if-else

复制代码

3.2、聚合查询

点击查看代码

1. package com.cc.mdb.test;
3. import com.alibaba.fastjson.JSONObject;
4. import com.cc.mdb.BaseTest;
5. import org.junit.Test;
6. import org.springframework.data.domain.Sort;
7. import org.springframework.data.mongodb.core.MongoTemplate;
8. import org.springframework.data.mongodb.core.aggregation.*;
9. import org.springframework.data.mongodb.core.query.Criteria;
11. import javax.annotation.Resource;
12. import java.util.ArrayList;
13. import java.util.Collection;
14. import java.util.List;
16. /**
17. * 聚合查询
18. *      $project                指定返回字段
19. *      $lookup                    多表关联
20. *      $unwind                    数组拆分
21. *      $match                    筛选条件
22. *      $sort                    排序
23. *      $limit                    分页
24. *      $skip                    跳过
25. *      $group                    根据id(可自定义)分组
26. *      $sum                    分组后求和
27. *      $first                    分组后取第一条
28. *      $last                    分组后取最后一条
29. *      $addToSet                分组后添加到list
30. *      $cond                    可以做 if-else
31. * @author CC
32. * @since 2023/4/20 0020
33. */
34. public class TestMongoAgg extends BaseTest {
36.     @Resource
37.     private MongoTemplate mongoTemplate;
39.     private static final String COLLECTION_NAME = "USER";
41.     /** 分组
42.      */
43.     @Test
44.     public void test00()throws Exception{
45.         Aggregation groupAgg = Aggregation.newAggregation(
46.                 Aggregation.group("age") // 分组条件
47.                         .first("age").as("age_R")     // 获取分组后查询到的该字段的第一个值
48. //                        .first("school.sId").as("sId_R")
49.                         .sum("age").as("sum_age_R")  // 按分组对该字段求和
50.                         .avg("age").as("avg_age_R")  // 按分组求该字段平均值
51.                         .sum("id").as("sum_id_R")    // 按age分组后求id的和
52.                         .min("id").as("min_id_R")    // 按age分组后求id的最小值
53.                         .max("id").as("max_id_R")    // 按age分组后求id的最大值
54.         );
56.         AggregationResults<JSONObject> results = mongoTemplate.aggregate(groupAgg, COLLECTION_NAME, JSONObject.class);
57.         List<JSONObject> res = results.getMappedResults();
58.         System.out.println(res);
61.     }
63.     /** 聚合查询：数组拆分 + 指定返回字段 + 排序 + 分页
64.      */
65.     @Test
66.     public void testAgg01()throws Exception{
67.         // 1. 配置聚合查询
68.         // 1.1. 将school字段分解成数组
69.         UnwindOperation unwind = Aggregation.unwind("school");
70.         // 1.2. 获取school.sId的属性中大于2的对象
71.         MatchOperation match = Aggregation.match(Criteria.where("school.sId").gt(2));
72.         // 1.3. 显示的字段
73.         ProjectionOperation project = Aggregation.project("_id", "school.sId",
74.                 "school.sName", "name", "id", "age", "email");
76.         // 1.4. 查询所有
77.         Aggregation aggregation = Aggregation.newAggregation(
78.                 unwind,
79.                 match,
80.                 project
81.         );
82.         // 1.5. 查询数量：把数量封装在totalCount这个值中
83.         Aggregation countAgg = Aggregation.newAggregation(
84.                 unwind,
85.                 match,
86.                 Aggregation.count().as("totalCount")
87.         );
88.         // 1.6. 排序+分页
89.         Aggregation sortAndSkipLimitAgg = Aggregation.newAggregation(
90.                 unwind,
91.                 match,
92.                 project,
93.                 // 1.6.1. 排序：先按照sId升序，再按照id降序
94.                 Aggregation.sort(Sort.Direction.ASC,"sId")
95.                         .and(Sort.Direction.DESC,"id"),
96.                 // 1.6.2. 分页
97.                 Aggregation.skip(0L),
98.                 Aggregation.limit(2L)
99.         );
101.         // 2. 查询
102.         // 2.1. 查询全部
103.         AggregationResults<JSONObject> results = mongoTemplate.aggregate(aggregation, COLLECTION_NAME, JSONObject.class);
104.         // 2.1.1. 获取映射结果
105.         List<JSONObject> allResult = results.getMappedResults();
106.         allResult.forEach(System.out::println);
107.         // 2.2. 查询数量
108.         AggregationResults<JSONObject> resultCount = mongoTemplate.aggregate(countAgg, COLLECTION_NAME, JSONObject.class);
109.         JSONObject jsonObject = resultCount.getUniqueMappedResult();
110.         System.out.println(jsonObject);
111.         Object totalCount = jsonObject.get("totalCount");
112.         System.out.println("总数：" + totalCount);
114.         // 2.3. 排序+分页查询
115.         AggregationResults<JSONObject> sortSkip = mongoTemplate.aggregate(sortAndSkipLimitAgg, COLLECTION_NAME, JSONObject.class);
116.         List<JSONObject> mappedResults = sortSkip.getMappedResults();
117.         System.out.println(mappedResults);
119.         //获取原始结果
120. //        Document rawResults = results.getRawResults();
121.         //获取唯一映射结果（只获取一个值）
122. //        JSONObject uniqueMappedResult = results.getUniqueMappedResult();
123.         //获取服务器使用情况
124. //        String serverUsed = results.getServerUsed();
125.         //循环：和mappedResults循环一样的
126. //        results.forEach(result -> {
127. //            System.out.println(result);
128. //        });
129.     }
131.     //添加聚合查询的数据
132.     @Test
133.     public void testAgg02() throws Exception {
134.         //写入复杂数据 —— 有子文档（json数据中有下级）
135.         List<JSONObject> vos = new ArrayList<>();
136.         for (int i = 1; i < 20; i++) {
137.             JSONObject result = new JSONObject();
138.             String aa = i + "" + i + "" + i + "";
139.             result.put("id", aa);
140.             result.put("name", "cscs" + aa);
141.             int age = 123;
142.             if (i >= 10){
143.                 age = 345;
144.             }
145.             result.put("age", age);
146.             result.put("email", "11@qq.com");
148.             List<JSONObject> jss = new ArrayList<>();
149.             if (i < 6){
150.                 for (int j = 1; j < 3; j++) {
151.                     JSONObject js = new JSONObject();
152.                     js.put("sId", j * i);
153.                     js.put("sName", "学校" + aa);
154.                     js.put("sNo", "学校编号" + aa);
155.                     js.put("sTab", j * i);
156.                     jss.add(js);
157.                 }
158.             }
159.             result.put("school", jss);
160.             vos.add(result);
161.         }
162.         Collection<JSONObject> user3 = mongoTemplate.insert(vos, COLLECTION_NAME);
163.         System.out.println(user3);
164.     }
167. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！