使用Spring Data Rest和MongoDB更新特定字段

Question

我正在使用Spring Data MongoDB和Spring Data Rest创建一个Rest API，它允许对我的MongoDB数据库进行GET、POST、PUT和DELETE操作，除了更新操作(PUT)之外，其他操作都很好。只有当我在请求体中发送完整对象时，它才起作用。

@Document
public class User {
    @Id
    private String id;
    private String email;
    private String lastName;
    private String firstName;
    private String password;

    ...
}

我试着在这里搜索一个解决方案，但我只找到了以下帖子，但没有解决方案：如何使用MongoRepository接口更新mongo db中的特定字段？

这有没有办法实施？

Answer 1

Spring Data Rest使用Spring数据存储库使用Rest调用（查看https://docs.Spring.io/spring-data/Rest/docs/current/reference/html/#reference)自动检索和操作持久数据。

使用Spring Data MongoDB时，您有MongoOperations接口，该接口用作Restendpoint的存储库。然而，MongoOperations目前不支持特定字段更新！

PS：如果他们在Spring Data JPA中添加@DynamicUpdate这样的特性，那就太棒了

但这并不意味着它可以做，这里是我做的变通当我有这个问题。

首先让我解释一下我们要做什么：

1. 我们将创建一个控制器，该控制器将覆盖所有PUT操作，以便实现自己的更新方法。
2. 在update方法中，我们将使用MongoTemplate，它确实具有更新特定字段能力。

<dependency>
    <groupId>org.reflections</groupId>
    <artifactId>reflections</artifactId>
    <version>0.9.12</version>
</dependency>

public class UpdateUtility {

    private static final String MODEL_PACKAGE = "com.mycompany.myproject.models";
    private static boolean initialized =  false;
    private static HashMap<String, Class> classContext = new HashMap<>();

    private static void init() {
        if(!initialized) {
            Reflections reflections = new Reflections(MODEL_PACKAGE);
            Set<Class<?>> classes = reflections.getTypesAnnotatedWith(Document.class); // Get all the classes annotated with @Document in the specified package

            for(Class<?> model : classes) {
                classContext.put(model.getSimpleName().toLowerCase(), model);
            }

            initialized = true;
        }
    }

    public static Class getClassFromType(String type) throws Exception{
        init();
        if(classContext.containsKey(type)) {
            return classContext.get(type);
        }
        else {
            throw new Exception("Type " + type + " does not exists !");
        }
    }
}

public class UpdateController {

    @Autowired
    private MongoTemplate mongoTemplate;

    @PutMapping("/{type}/{id}")
    public Object update(@RequestBody HashMap<String, Object> fields,
                                  @PathVariable(name = "type") String type,
                                  @PathVariable(name = "id") String id) {
        try {
            Class classType = UpdatorUtility.getClassFromType(type); // Get the domain class from the type in the request
            Query query = new Query(Criteria.where("id").is(id)); // Update the document with the given ID
            Update update = new Update();

            // Iterate over the send fields and add them to the update object
            Iterator iterator = fields.entrySet().iterator();
            while(iterator.hasNext()) {
                HashMap.Entry entry = (HashMap.Entry) iterator.next();
                String key = (String) entry.getKey();
                Object value = entry.getValue();
                update.set(key, value);
            }

            mongoTemplate.updateFirst(query, update, classType); // Do the update
            return mongoTemplate.findById(id, classType); // Return the updated document
        } catch (Exception e) {
            // Handle your exception
        }
    }
}

现在，我们可以更新指定的字段，而不需要更改调用。因此在您的情况下，呼叫将是：

PUT http://MY-DOMAIN/user/MY-USER-ID { lastName: "My new last name" }

PS：你可以通过增加在嵌套对象中更新特定字段的可能性来改进它...