Spring Mongo 仅通过新 POJO 的非空字段进行更新

Question

用户模型：

@Document(collection = "USER")
public class UserModel {
    private String id;
    private String username;
    @Indexed(unique = true)
    private String email;
    private String password;
    private String firstName;
    private String lastName;
}

用户存储库：

@Repository
public interface UserRepository extends MongoRepository<UserModel, String> {

    Optional<UserModel> findByEmail(String email);
    Optional<UserModel> findByUsername(String username);
}

像这样保存用户文档：

@Autowired
UserRepository userRepo;

public someMethod() {
    UserModel user = new UserModel();
    //Set all fields....
    userRepo.save(user);
}

尝试像这样更新：

@Autowired
UserRepository userRepo;

public someMethod() {
    UserModel user = new UserModel();
    user.setId("A valid Id got from database by last insert")
    user.setFirstName("New first name");
    // Remaining fields are null now
    userRepo.save(user);
}

这里的 null 值将由 MongoRepository 插入。

我可以通过什么设置告诉 MongoRepository 在更新时避免空值？

如果我们通过MongoTemplate updateFirst() 方法执行此操作，我们必须为 POJO 的所有字段添加if 条件。在这种情况下，最好的解决方案是什么？

Answer 1

尝试先从数据库中获取用户，然后更新您的数据并保存在数据库中。

UserModel user = userRepo.findById(id); // Fetch from database
user.setFirstName("New first name");    // Update the data
userRepo.save(user);                    // Save in the database

或使用自定义查询仅更新选定的字段。这里要使其动态化，从UserModel 创建一个Map<String, Object> 并删除空值。

ObjectMapper oMapper = new ObjectMapper();
Map<String, Object> dataMap = oMapper.convertValue(user, Map.class);
map.values().removeIf(Objects::isNull);

现在在Update 中设置地图，然后在数据库中更新。

Update update = new Update();
dataMap.forEach(update::set);
Query query = new Query().addCriteria(where("_id").is(id));
mongoTemplate.update(UserModel.class).matching(query).apply(update).first();