31
31
import io .micronaut .data .model .PersistentProperty ;
32
32
import io .micronaut .data .model .naming .NamingStrategy ;
33
33
import io .micronaut .data .model .runtime .RuntimeEntityRegistry ;
34
+ import io .micronaut .data .mongodb .operations .MongoCollectionNameProvider ;
34
35
import io .micronaut .inject .qualifiers .Qualifiers ;
35
36
import org .slf4j .Logger ;
36
37
import org .slf4j .LoggerFactory ;
@@ -80,10 +81,12 @@ protected <M> M getMongoFactory(Class<M> mongoFactoryClass, BeanLocator beanLoca
80
81
* @param runtimeEntityRegistry The entity registry
81
82
* @param mongoConfigurations The configuration
82
83
* @param databaseOperationsProvider The database provider
84
+ * @param mongoCollectionNameProvider The Mongo collection name provider
83
85
*/
84
86
protected void initialize (RuntimeEntityRegistry runtimeEntityRegistry ,
85
87
List <AbstractMongoConfiguration > mongoConfigurations ,
86
- DatabaseOperationsProvider <Dtbs > databaseOperationsProvider ) {
88
+ DatabaseOperationsProvider <Dtbs > databaseOperationsProvider ,
89
+ MongoCollectionNameProvider mongoCollectionNameProvider ) {
87
90
88
91
for (AbstractMongoConfiguration mongoConfiguration : mongoConfigurations ) {
89
92
// TODO: different initializer per conf
@@ -99,7 +102,7 @@ protected void initialize(RuntimeEntityRegistry runtimeEntityRegistry,
99
102
for (PersistentEntity entity : entities ) {
100
103
Dtbs database = databaseOperations .find (entity );
101
104
Set <String > collections = databaseOperations .listCollectionNames (database );
102
- String persistedName = entity . getPersistedName ( );
105
+ String persistedName = mongoCollectionNameProvider . provide ( entity );
103
106
if (collections .add (persistedName )) {
104
107
if (LOG .isInfoEnabled ()) {
105
108
LOG .info ("Creating collection: {} in database: {}" , persistedName , databaseOperations .getDatabaseName (database ));
0 commit comments