io.mongock.driver.mongodb.springdata.v4.SpringDataMongoV4ChangeEntryRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongodb-springdata-v4-driver Show documentation

The newest version!

package io.mongock.driver.mongodb.springdata.v4;

import io.mongock.api.exception.*;
import io.mongock.driver.api.entry.*;
import io.mongock.driver.core.entry.*;
import io.mongock.driver.mongodb.sync.v4.repository.*;
import org.bson.*;
import org.springframework.data.mongodb.core.*;
import org.springframework.data.mongodb.core.query.*;

public class SpringDataMongoV4ChangeEntryRepository extends MongoSync4ChangeEntryRepository implements ChangeEntryRepositoryWithEntity {

  private final MongoOperations mongoOperations;
  private final boolean transactionable;

  public SpringDataMongoV4ChangeEntryRepository(MongoOperations mongoOperations,
                                                String collectionName,
                                                ReadWriteConfiguration readWriteConfiguration,
                                                boolean transactionable) {
    super(mongoOperations.getCollection(collectionName), readWriteConfiguration);
    this.mongoOperations = mongoOperations;
    this.transactionable = transactionable;
  }


  @Override
  public void saveOrUpdate(ChangeEntry changeEntry) throws MongockException {

    if (transactionable) {
      //if transaction is enabled, it will delegate the write concern to the transactionManager
      Query filter = new Query().addCriteria(new Criteria()
          .andOperator(
              Criteria.where(KEY_EXECUTION_ID).is(changeEntry.getExecutionId()),
              Criteria.where(KEY_CHANGE_ID).is(changeEntry.getChangeId()),
              Criteria.where(KEY_AUTHOR).is(changeEntry.getAuthor())));
      mongoOperations.upsert(filter, getUpdateFromEntity(changeEntry), collection.getNamespace().getCollectionName());
    } else {
      super.saveOrUpdate(changeEntry);
    }
  }


  private Update getUpdateFromEntity(ChangeEntry changeEntry) {
    Update updateChangeEntry = new Update();
    toEntity(changeEntry).forEach(updateChangeEntry::set);
    return updateChangeEntry;
  }
}