MongoDB + SpringBoot 的基礎CRUD、聚合查詢

        <!--mongodb的包-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-mongodb</artifactId>
        </dependency>
        <!--fastjson-->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.83</version>
        </dependency>

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<User> 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("[email protected]");
            vos.add(user);
        }
        Collection<User> user2 = mongoTemplate.insert(vos, "User2");
        System.out.println(user2);
    }

    //寫入簡單資料 —— 根據json存資料（JSONObject(Fastjson)）
    @Test
    public void test02()throws Exception{
        List<JSONObject> 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","[email protected]");
            vos.add(result);
        }
        Collection<JSONObject> user3 = mongoTemplate.insert(vos, "User3");
        System.out.println(user3);
    }

    //寫入復雜資料 —— 有子檔案（json資料中有下級）
    @Test
    public void test013()throws Exception{
        List<JSONObject> 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","[email protected]");

            List<JSONObject> 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<JSONObject> 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<JSONObject> user3 = mongoTemplate.insert(vos, "User4");
        System.out.println(user3);
    }

    //查找全部
    @Test
    public void test03()throws Exception{
        List<User> 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<JSONObject> 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<JSONObject> 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<JSONObject> results = mongoTemplate.aggregate(groupAgg, COLLECTION_NAME, JSONObject.class);
        List<JSONObject> 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<JSONObject> results = mongoTemplate.aggregate(aggregation, COLLECTION_NAME, JSONObject.class);
        // 2.1.1. 獲取映射結果
        List<JSONObject> allResult = results.getMappedResults();
        allResult.forEach(System.out::println);
        // 2.2. 查詢數量
        AggregationResults<JSONObject> 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<JSONObject> sortSkip = mongoTemplate.aggregate(sortAndSkipLimitAgg, COLLECTION_NAME, JSONObject.class);
        List<JSONObject> 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<JSONObject> 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);
            int age = 123;
            if (i >= 10){
                age = 345;
            }
            result.put("age", age);
            result.put("email", "[email protected]");

            List<JSONObject> 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<JSONObject> user3 = mongoTemplate.insert(vos, COLLECTION_NAME);
        System.out.println(user3);
    }


}