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

                            org.springframework.boot            spring-boot-starter-data-mongodb                                    com.alibaba            fastjson            1.2.83        

spring:  data:    mongodb:      # 有账号密码      uri: mongodb://账户名:密码@IP:端口/数据库      # 无账号密码      # uri: mongodb://IP:端口/数据库

package com.cc.mdb.entity;import lombok.Data;import org.springframework.data.annotation.Id;import org.springframework.data.mongodb.core.mapping.Document;import org.springframework.data.mongodb.core.mapping.Field;import java.time.LocalDateTime;/** * @Document("User") 存入的时候不取对象名，默认用User当做表名 */@Data@Document("User")public class User {    @Id    private String id;    /**     * 使用name当做字段key     */    @Field("name")    private String name;    /**     * 使用age1当做字段key     */    @Field("age1")    private Integer age;    /**     * 使用email当做字段key     */    @Field("email")    private String email;    /**     * 使用createDate当做字段key     */    @Field("createDate")    private LocalDateTime createDate = LocalDateTime.now();}

package com.cc.mdb.test;import com.alibaba.fastjson.JSONObject;import com.cc.mdb.BaseTest;import com.cc.mdb.entity.User;import com.mongodb.client.result.UpdateResult;import org.junit.Test;import org.springframework.data.domain.PageRequest;import org.springframework.data.domain.Pageable;import org.springframework.data.domain.Sort;import org.springframework.data.mongodb.core.MongoTemplate;import org.springframework.data.mongodb.core.query.Criteria;import org.springframework.data.mongodb.core.query.Query;import org.springframework.data.mongodb.core.query.Update;import javax.annotation.Resource;import java.util.ArrayList;import java.util.Collection;import java.util.List;import java.util.regex.Pattern;/** CRUD *  查询使用：Query *  博客：https://blog.csdn.net/weixin_44185213/article/details/125293170?spm=1001.2014.3001.5502 * @author CC * @since 2023/3/30 0030 */@SpringBootTest@RunWith(SpringRunner.class)public class TestMongoQuery{    @Resource    private MongoTemplate mongoTemplate;    //写入简单数据 —— 根据对象存数据    @Test    public void test01()throws Exception{        List vos = new ArrayList();        for (int i = 10; i < 20; i++) {            User user = new User();            String aa = i + "" + i + "" + i + "";            user.setId(aa);            user.setName("cscs"+aa);            user.setAge(123);            user.setEmail("11@qq.com");            vos.add(user);        }        Collection user2 = mongoTemplate.insert(vos, "User2");        System.out.println(user2);    }    //写入简单数据 —— 根据json存数据（JSONObject(Fastjson)）    @Test    public void test02()throws Exception{        List vos = new ArrayList();        for (int i = 1; i < 20; i++) {            JSONObject result = new JSONObject();            String aa = i + "" + i + "" + i + "";            result.put("id",aa);            result.put("name","cscs"+aa);            result.put("age",123);            result.put("email","11@qq.com");            vos.add(result);        }        Collection user3 = mongoTemplate.insert(vos, "User3");        System.out.println(user3);    }    //写入复杂数据 —— 有子文档（json数据中有下级）    @Test    public void test013()throws Exception{        List vos = new ArrayList();        for (int i = 1; i < 20; i++) {            JSONObject result = new JSONObject();            String aa = i + "" + i + "" + i + "";            result.put("id",aa);            result.put("name","cscs"+aa);            result.put("age",123);            result.put("email","11@qq.com");            List jss = new ArrayList();            for (int j = 0; j < 2; j++) {                JSONObject js = new JSONObject();                js.put("sId",j);                js.put("sName","学校" + aa);                js.put("sNo","学校编号" + aa);                js.put("sTab",1);                List whats = new ArrayList();                if (i == 1){                    for (int k = 0; k < 2; k++) {                        JSONObject kk = new JSONObject();                        kk.put("kId",k);                        kk.put("kName","k"+k);                        whats.add(kk);                    }                }                js.put("sWhat",whats);                jss.add(js);            }            result.put("school",jss);            vos.add(result);        }        Collection user3 = mongoTemplate.insert(vos, "User4");        System.out.println(user3);    }    //查找全部    @Test    public void test03()throws Exception{        List users = mongoTemplate.findAll(User.class);        System.out.println(users);    }    /** 根据json里面的字段查询     * 查询 school.sWhat.kId 的数据     */    @Test    public void test07()throws Exception{        //查询的新增方法：test013        Criteria cr = new Criteria();//        cr.and("id").is("111");//        cr.and("school.sId").is(0);        cr.and("school.sWhat.kId").is(0);        Query query = new Query();        query.addCriteria(cr);        // 执行查询并获取结果        List jsonObjects = mongoTemplate.find(query, JSONObject.class, "User4");        System.out.println(jsonObjects);    }    //**高级查询：使用Query    @Test    public void test04()throws Exception{        String search = "cscs";        int page = 1;        int size = 20;        Query query = new Query();        //一、where        //1模糊        String format = String.format("^.*%s.*$", search);        Pattern pattern = Pattern.compile(format, Pattern.CASE_INSENSITIVE);        Criteria criteria = Criteria.where("name").regex(pattern);        //模糊查询方式2（推荐）//        Criteria.where("name").regex(String.format(".*%s.*","查询的值"));        //2等于（这里比较特殊，value：数字的必须传入数字的、字符串必须传入字符串的）        criteria.and("age").is(123);        query.addCriteria(criteria);        //二、排序。按id升序        Sort sort = Sort.by(Sort.Direction.ASC, "_id");        query.with(sort);        //二、总条数（是所有条件都弄了之后）        long totalCount = mongoTemplate.count(query, JSONObject.class, "User3");        //三、分页（必须在where和查询总条数之后）        //方式1//        query.skip((page - 1) * size).limit(size);        //方式2（推荐）        Pageable pageable = PageRequest.of(page - 1, size);        query.with(pageable);        //查询。由于查询出来的是key-value的形式，可以使用JSONObject(FastJson)来接收参数。如果有对象（User），可以直接使用对象来接收        List users = mongoTemplate.find(query, JSONObject.class, "User3");        System.out.println("总条数：" + totalCount);        System.out.println(users);    }    //修改 - 根据表名+id 修改其中一个json中的某些值    @Test    public void test05() {        JSONObject user2 = mongoTemplate.findById("6433c76df108474b788cc4ce", JSONObject.class, "User3");        Object id = user2.get("_id");        Query query = new Query(Criteria.where("_id").is(id));        Update update = new Update();        update.set("name", "修改后的名字2");        update.set("age", "321");//        update.set("email", "");        //方式1：upsert方法更新与查询条件匹配的第一个文档，如果没有文档与查询条件匹配，则插入一个新文档。//        UpdateResult result = mongoTemplate.upsert(query, update, "User3");//        long count = result.getModifiedCount();//        System.out.println(count);        //方式2：update方法更新与查询条件匹配的第一个文档        UpdateResult user3 = mongoTemplate.updateFirst(query, update, "User3");        //修改计数        long modifiedCount = user3.getModifiedCount();        System.out.println(modifiedCount);        //匹配计数        long matchedCount = user3.getMatchedCount();        System.out.println(matchedCount);        //方式3：updateMulti方法更新与查询条件匹配的所有文档。//        UpdateResult user31 = mongoTemplate.updateMulti(query, update, "User3");    }    //删除 - 根据数据id删除    @Test    public void test06() {        //1删除文档中的某一条数据//        Query query = new Query(Criteria.where("_id").is("6433c76df108474b788cc4cf"));//        DeleteResult result = mongoTemplate.remove(query, "User3");//        long count = result.getDeletedCount();//        System.out.println(count);        //2删除整个文档中的数据 —— 相当于drop清空表中的数据//        DeleteResult user4 = mongoTemplate.remove(new Query(), "User4");//        long deletedCount = user4.getDeletedCount();//        System.out.println(deletedCount);        //3删除文档 —— 相当于删除表        mongoTemplate.dropCollection("User4");    }}

package com.cc.mdb.test;import com.alibaba.fastjson.JSONObject;import com.cc.mdb.BaseTest;import org.junit.Test;import org.springframework.data.domain.Sort;import org.springframework.data.mongodb.core.MongoTemplate;import org.springframework.data.mongodb.core.aggregation.*;import org.springframework.data.mongodb.core.query.Criteria;import javax.annotation.Resource;import java.util.ArrayList;import java.util.Collection;import java.util.List;/** * 聚合查询 *      $project        指定返回字段 *      $lookup            多表关联 *      $unwind            数组拆分 *      $match            筛选条件 *      $sort            排序 *      $limit            分页 *      $skip            跳过 *      $group            根据id(可自定义)分组 *      $sum            分组后求和 *      $first            分组后取第一条 *      $last            分组后取最后一条 *      $addToSet        分组后添加到list *      $cond            可以做 if-else * @author CC * @since 2023/4/20 0020 */public class TestMongoAgg extends BaseTest {    @Resource    private MongoTemplate mongoTemplate;    private static final String COLLECTION_NAME = "USER";    /** 分组     */    @Test    public void test00()throws Exception{        Aggregation groupAgg = Aggregation.newAggregation(                Aggregation.group("age") // 分组条件                        .first("age").as("age_R")     // 获取分组后查询到的该字段的第一个值//                        .first("school.sId").as("sId_R")                        .sum("age").as("sum_age_R")  // 按分组对该字段求和                        .avg("age").as("avg_age_R")  // 按分组求该字段平均值                        .sum("id").as("sum_id_R")    // 按age分组后求id的和                        .min("id").as("min_id_R")    // 按age分组后求id的最小值                        .max("id").as("max_id_R")    // 按age分组后求id的最大值        );        AggregationResults results = mongoTemplate.aggregate(groupAgg, COLLECTION_NAME, JSONObject.class);        List res = results.getMappedResults();        System.out.println(res);    }    /** 聚合查询：数组拆分 + 指定返回字段 + 排序 + 分页     */    @Test    public void testAgg01()throws Exception{        // 1. 配置聚合查询        // 1.1. 将school字段分解成数组        UnwindOperation unwind = Aggregation.unwind("school");        // 1.2. 获取school.sId的属性中大于2的对象        MatchOperation match = Aggregation.match(Criteria.where("school.sId").gt(2));        // 1.3. 显示的字段        ProjectionOperation project = Aggregation.project("_id", "school.sId",                "school.sName", "name", "id", "age", "email");        // 1.4. 查询所有        Aggregation aggregation = Aggregation.newAggregation(                unwind,                match,                project        );        // 1.5. 查询数量：把数量封装在totalCount这个值中        Aggregation countAgg = Aggregation.newAggregation(                unwind,                match,                Aggregation.count().as("totalCount")        );        // 1.6. 排序+分页        Aggregation sortAndSkipLimitAgg = Aggregation.newAggregation(                unwind,                match,                project,                // 1.6.1. 排序：先按照sId升序，再按照id降序                Aggregation.sort(Sort.Direction.ASC,"sId")                        .and(Sort.Direction.DESC,"id"),                // 1.6.2. 分页                Aggregation.skip(0L),                Aggregation.limit(2L)        );        // 2. 查询        // 2.1. 查询全部        AggregationResults results = mongoTemplate.aggregate(aggregation, COLLECTION_NAME, JSONObject.class);        // 2.1.1. 获取映射结果        List allResult = results.getMappedResults();        allResult.forEach(System.out::println);        // 2.2. 查询数量        AggregationResults resultCount = mongoTemplate.aggregate(countAgg, COLLECTION_NAME, JSONObject.class);        JSONObject jsonObject = resultCount.getUniqueMappedResult();        System.out.println(jsonObject);        Object totalCount = jsonObject.get("totalCount");        System.out.println("总数：" + totalCount);        // 2.3. 排序+分页查询        AggregationResults sortSkip = mongoTemplate.aggregate(sortAndSkipLimitAgg, COLLECTION_NAME, JSONObject.class);        List mappedResults = sortSkip.getMappedResults();        System.out.println(mappedResults);        //获取原始结果//        Document rawResults = results.getRawResults();        //获取唯一映射结果（只获取一个值）//        JSONObject uniqueMappedResult = results.getUniqueMappedResult();        //获取服务器使用情况//        String serverUsed = results.getServerUsed();        //循环：和mappedResults循环一样的//        results.forEach(result -> {//            System.out.println(result);//        });    }    //添加聚合查询的数据    @Test    public void testAgg02() throws Exception {        //写入复杂数据 —— 有子文档（json数据中有下级）        List vos = new ArrayList();        for (int i = 1; i = 10){                age = 345;            }            result.put("age", age);            result.put("email", "11@qq.com");            List jss = new ArrayList();            if (i < 6){                for (int j = 1; j < 3; j++) {                    JSONObject js = new JSONObject();                    js.put("sId", j * i);                    js.put("sName", "学校" + aa);                    js.put("sNo", "学校编号" + aa);                    js.put("sTab", j * i);                    jss.add(js);                }            }            result.put("school", jss);            vos.add(result);        }        Collection user3 = mongoTemplate.insert(vos, COLLECTION_NAME);        System.out.println(user3);    }}