io.mongock.driver.mongodb.springdata.v4.SpringDataMongoV4DriverBase 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
There is a newer version: 5.5.0
Show newest version
package io.mongock.driver.mongodb.springdata.v4;

import com.github.cloudyrock.mongock.driver.mongodb.springdata.v4.decorator.impl.*;
import com.mongodb.client.*;
import io.mongock.api.exception.*;
import io.mongock.driver.api.driver.*;
import io.mongock.driver.api.entry.*;
import io.mongock.driver.mongodb.sync.v4.driver.*;
import io.mongock.utils.annotation.*;
import org.slf4j.*;
import org.springframework.data.mongodb.*;
import org.springframework.data.mongodb.core.*;
import org.springframework.transaction.*;
import org.springframework.transaction.support.*;

import java.util.*;

@NotThreadSafe
public abstract class SpringDataMongoV4DriverBase> extends MongoSync4DriverGeneric implements TenantSelectable {

  protected static final Logger logger = LoggerFactory.getLogger(SpringDataMongoV4DriverBase.class);

  protected final MongoTemplate mongoTemplate;
  protected MongoTransactionManager txManager;

  protected SpringDataMongoV4DriverBase(MongoTemplate mongoTemplate,
                                        long lockAcquiredForMillis,
                                        long lockQuitTryingAfterMillis,
                                        long lockTryFrequencyMillis) {
    super(lockAcquiredForMillis, lockQuitTryingAfterMillis, lockTryFrequencyMillis);
    this.mongoTemplate = mongoTemplate;
    disableTransaction();
  }

  @Override
  protected MongoDatabase getDataBase() {
    return mongoTemplate.getDb();
  }
  @Override
  public void runValidation() throws MongockException {
    super.runValidation();
    if (this.mongoTemplate == null) {
      throw new MongockException("MongoTemplate must not be null");
    }
  }


  @Override
  public void specificInitialization() {
    super.specificInitialization();
    dependencies.add(new ChangeSetDependencyBuildable(
        MongockTemplate.class,
        MongoTemplate.class,
        impl -> new MongockTemplate((MongoTemplate) impl),
        true));
    dependencies.add(new ChangeSetDependency(MongoTemplate.class, this.mongoTemplate));

    txManager = new MongoTransactionManager(mongoTemplate.getMongoDatabaseFactory(), txOptions);
  }

  public MongockTemplate getMongockTemplate() {
    if (!isInitialized()) {
      throw new MongockException("Mongock Driver hasn't been initialized yet");
    }
    return dependencies
        .stream()
        .filter(dependency -> MongockTemplate.class.isAssignableFrom(dependency.getType()))
        .map(ChangeSetDependency::getInstance)
        .map(instance -> (MongockTemplate) instance)
        .findAny()
        .orElseThrow(() -> new MongockException("Mongock Driver hasn't been initialized yet"));

  }

  @Override
  public ChangeEntryService getChangeEntryService() {
    if (changeEntryRepository == null) {
      changeEntryRepository = new SpringDataMongoV4ChangeEntryRepository(mongoTemplate, getMigrationRepositoryName(), getReadWriteConfiguration(), isTransactionable());
      changeEntryRepository.setIndexCreation(isIndexCreation());
    }
    return changeEntryRepository;
  }

  @Override
  public Optional getTransactioner() {
    return Optional.ofNullable(transactionEnabled ? this : null);
  }

  @Override
  public void executeInTransaction(Runnable operation) {
    TransactionStatus txStatus = getTxStatus(txManager);
    try {
      mongoTemplate.setSessionSynchronization(SessionSynchronization.ALWAYS);
      operation.run();
      txManager.commit(txStatus);
    } catch (Exception ex) {
      logger.warn("Error in Mongock's transaction", ex);
      txManager.rollback(txStatus);
      throw new MongockException(ex);
    }

  }

  protected TransactionStatus getTxStatus(PlatformTransactionManager txManager) {
    DefaultTransactionDefinition def = new DefaultTransactionDefinition();
// explicitly setting the transaction name is something that can be done only programmatically
    def.setName("mongock-transaction-spring-data-3");
    def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);
    return txManager.getTransaction(def);
  }


  @Deprecated
  public void enableTransactionWithTxManager(PlatformTransactionManager txManager) {
    enableTransaction();
  }
}