Spring Data Mongodb DBRef聯級儲存
Spring Data MongoDB by default does not support cascade operations on referenced objects with @DBRef annotations as reference says:
The mapping framework does not handle cascading saves. If you change an Account object that is referenced by a Person object, you mustsave the Account object separately
. Calling save on the Person object will not automatically save the Account objects in the property accounts.
That’s quite problematic because in order to achieve saving child objects you need to override save method in repository in parent or create additional service
methods like it is presented in
In this article I will show you how it can be achieved for all documents using generic implementation of AbstractMongoEventListener.
@CascadeSave annotation
Because we can’t change @DBRef annotation by adding cascade property lets create new annotation @CascadeSave that will be used to mark which fields should be saved when parent object is saved.
1 2 3 4 5 6 7 8 9 10 11 12 |
|
CascadingMongoEventListener
Next part is to implement handler for this annotation. We will use for that powerful Spring Application Event mechanism. In particular we will extendAbstractMongoEventListener to catch saved object before it is converted to Mongo’s DBObject.
How does it work? When object MongoTemplate#save method is called, before object is actually saved it is being converted into DBObject from MongoDB api. CascadingMongoEventListener implemented below provides hook that catches object before its converted and:
- goes through all its fields to check if there are fields annotated with @DBRef and @CascadeSave at once.
- when field is found it checks if @Id annotation is present
- child object is being saved
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
|
Mapping requirements
As you can see in order to make thing work you need to follow some rules:
- parent’s class child property has to be mapped with @DBRef and @CascadeSave
- child class needs to have property annotated with @Id and if that id is supposed to be autogenerated it should by type of ObjectId
Usage
In order to use cascade saving in your project you need just to register CascadingMongoEventListener in Spring Context:
1
|
|
Let’s test it
In order to show an example I made two document classes:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
|
1 2 3 4 5 6 7 8 9 10 11 12 |
|
In test there is one user with address created and then user is saved. Test will cover only positive scenario and its just meant to show that it actually works (applcationContext-tests.xml
contains
only default Spring Data MongoDB beans and CascadingMongoEventListener registered):
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
|
We can check that also in Mongo console:
1 2 3 4 |
|
Summary
With this simple solution we can finally save child objects with one method call without implementing anything special for each document class.
I believe that we will find this functionality together with cascade delete as part Spring Data Mongo