MongoTemplate 操作数组字段

import cn.hutool.json.JSONUtil;
import com.google.common.collect.Lists;
import com.mongodb.BasicDBList;
import com.mongodb.BasicDBObject;
import com.mongodb.client.MongoClients;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.mapping.Document;
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 java.util.List;

**
 * @author zxp
 * @version 1.0
 * @date 2021/11/1 9:40
 */
@Slf4j
public class MongoArrOptTest {

    private MongoTemplate mongoTemplate;

    @Data
    @Document(collection = "user")
    static class UserEntity {
        @Id
        private String id;

        /**
         * 用户ID
         */
        private Integer userId;
        /**
         * 用户车辆列表
         */
        private List<UserCar> carList;
    }

    @Data
    @AllArgsConstructor
    @NoArgsConstructor
    static class UserCar {
        /**
         * 车牌号
         */
        private String carNo;
        /**
         * 车品牌
         */
        private String carBrand;
    }

    @BeforeEach
    public void before() {

        mongoTemplate = new MongoTemplate(MongoClients.create(), "test");

        // 初始化数据
        UserEntity userEntity = new UserEntity();
        userEntity.setUserId(1);
        userEntity.setCarList(Lists.newArrayList());
        mongoTemplate.insert(userEntity);
    }

    @AfterEach
    public void after() {
        List<UserEntity> all = mongoTemplate.findAll(UserEntity.class);
        System.out.println(JSONUtil.toJsonPrettyStr(all));
        // 删除测试集合
        mongoTemplate.dropCollection(UserEntity.class);
    }



    @Test
    @DisplayName("数组添加一条数据")
    public void push() {
        Query query = new Query(Criteria.where("userId").is(1));
        Update update = new Update();
        update.push("carList", new UserCar("闽A12345", "奥迪"));
        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }


    @Test
    @DisplayName("数组添加多条数据")
    public void pushAll() {
        Query query = new Query(Criteria.where("userId").is(1));
        Update update = new Update();
        update.push("carList").each(Lists.newArrayList(
                new UserCar("闽A11111", "奥迪"),
                new UserCar("闽A22222", "奥迪")));
        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }

    @Test
    @DisplayName("不存在则插入到数组，已存在将不操作")
    public void addToSet() {
        Query query = new Query(Criteria.where("userId").is(1));
        Update update = new Update();
        update.addToSet("carList", new UserCar("闽A11111", "奥迪"));
        // 只会插入一条
        mongoTemplate.updateFirst(query, update, UserEntity.class);
        mongoTemplate.updateFirst(query, update, UserEntity.class);

        log.debug("results: {}", JSONUtil.toJsonPrettyStr(mongoTemplate.findAll(UserEntity.class)));

        update.addToSet("carList").each(Lists.newArrayList(
                new UserCar("闽A11111", "奥迪"),
                new UserCar("闽A22222", "奥迪")));
        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }

    @Test
    @DisplayName("更新数组内的某条记录")
    public void update() {
        // 先插入一些数据
        this.pushAll();

        // 更新车牌号为 闽A22222 的 车辆品牌信息为大众
        Query query = new Query(Criteria
                .where("userId").is(1)
                .and("carList").elemMatch(Criteria.where("carNo").is("闽A22222"))
        );
        Update update = new Update();
        update.set("carList.$", new UserCar("闽A22222", "大众"));
        // 只会插入一条
        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }

    @Test
    @DisplayName("数组删除数据")
    public void pull() {
        // 先插入一些数据
        this.pushAll();

        Query query = new Query(Criteria
                .where("userId").is(1)
        );
        Update update = new Update();
        update.pull("carList", new UserCar("闽A11111", "奥迪"));

        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }

    @Test
    @DisplayName("数组按条件删除指定数据")
    public void pullBy() {
        // 先插入一些数据
        this.pushAll();
        this.pushAll();

        Query query = new Query(Criteria
                .where("userId").is(1)
        );
        Update update = new Update();
        // 删除所有 carNo = 闽A11111 的数据
        update.pull("carList", new BasicDBObject("carNo", "闽A11111"));

        // 删除 carNo 在数组中的数据
        BasicDBList basicDBList = new BasicDBList();
        basicDBList.addAll(Lists.newArrayList("闽A11111", "闽A22222"));
        update.pull("carList", new BasicDBObject("carNo", new BasicDBObject("$in", basicDBList)));

        mongoTemplate.updateFirst(query, update, UserEntity.class);
    }

}