使用MongoDB实现增删改查

使用MongoDB实现增删改查

简单实现MongoDB的增删改查功能，MongoDB使用的是本地数据库（默认提前安装好了），图形化界面使用的是MongoDB Compass。

0、手动添加数据库中的文档（提前建好db和collection，我的db叫testdb，collection是test）

{
  "_id": {
    "$oid": "657bc100ff585a84cb0dda68"
  },
  "name":"name",
  "age":18,
  "gender":"男",
  "phone":"18888888888"
}

1、添加MongoDB的maven依赖

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

2、编写yaml文件中MongoDB的配置

spring:
  data:
    mongodb:
      uri: mongodb://127.0.0.1:27017/testdb

3、编写实体类对应collection

/**
 * @Author: wxy
 * @Date: 2023/12/15 08:40
 * @Description: person实体
 */
@NoArgsConstructor
@AllArgsConstructor
@Data
@Document(collection = "person")
public class Person {

    /**
     * id
     */
    @Id
    private String id;
    /**
     * 姓名
     */
    private String name;
    /**
     * 年龄
     */
    private Integer age;
    /**
     * 性别
     */
    private String gender;
    /**
     * 电话
     */
    private String phone;

}

4、编写对应类的控制器

/**
 * @Author: wxy
 * @Date: 2023/12/15 08:44
 * @Description: person控制器
 */
@RestController
@RequestMapping("/person")
@RequiredArgsConstructor
public class PersonController {

    private final MongoTemplate mongoTemplate;

    /**
     * 查询person全部数据
     *
     * @return person全部数据
     */
    @GetMapping("/list")
    public R getAllPerson() {
        return R.success(mongoTemplate.findAll(Person.class));
    }

    /**
     * 根据id查询person单条数据
     *
     * @param id person id
     * @return person单条数据
     */
    @GetMapping("/{id}")
    public R getPerson(@PathVariable("id") String id) {
        return R.success(mongoTemplate.findById(id, Person.class));
    }

    /**
     * 新增一条person信息
     *
     * @param person 前端传来的person
     * @return 是否成功
     */
    @PostMapping("/save")
    public R savePerson(@RequestBody Person person) {
        return R.success(mongoTemplate.save(person, "person"));
    }

    /**
     * 根据id更新person信息
     *
     * @param person 包含id以及更新的所有字段
     * @return 是否成功
     */
    @PutMapping("/update")
    public R updatePerson(@RequestBody Person person) {
        Query query = new Query(Criteria.where("_id").is(person.getId()));
        Update update = new Update()
                .set("name", person.getName())
                .set("age", person.getAge())
                .set("gender", person.getGender())
                .set("phone", person.getPhone());
        return R.success(mongoTemplate.updateMulti(query, update, Person.class));
    }

    /**
     * 根据id更新person的第一条信息
     *
     * @param person 包含id，name的json
     * @return 是否更新成功
     */
    @PutMapping("/updateFirst")
    public R updateFirstPerson(@RequestBody Person person) {
        Query query = new Query(Criteria.where("_id").is(person.getId()));
        Update update = new Update().set("name", person.getName());
        return R.success(mongoTemplate.updateFirst(query, update, Person.class));
    }

    /**
     * 根据ids批量删除person信息
     *
     * @param ids 前端传来的id串，用逗号,分割
     * @return 是否成功
     */
    @DeleteMapping("/remove")
    public R removePerson(String ids) {
        List<String> idsList = Arrays.stream(ids.split(",")).collect(Collectors.toList());
        Query query = new Query(Criteria.where("_id").in(idsList));
        return R.success(mongoTemplate.remove(query, Person.class));
    }
    
}

5、总结：

这里使用的是mongoTemplate进行的增删改查，也可以用MongoDB的持久层接口，代码如下：

/**
 * @Author: wxy
 * @Date: 2023/12/15 10:51
 * @Description:
 */
// 这里的泛型String代表的是实体中id的类型
public interface PersonRepository extends MongoRepository<Person, String> {
}

使用时直接注入使用即可，具体的API就不再重新编写代码展示，但是应该都是大差不差的使用方法。

在第0步中我遇到了一个错误，我自己添加的时候我直接编写的 _id 类型是string类型，导致我后面根据id查询、修改、删除时都匹配不到对应的数据，这里我搜索了以下资料并没有查阅到原因，我自己的理解应该使用 _id 查询数据库的时候会自动查询ObjectId类型的，我String类型的 _id 是查询不到的，因为理解不一定对所以写到最后面仅供参考。