com.github.zhengframework.jdbc.migrate.LiquibaseMigrateModule Maven / Gradle / Ivy

Go to download
package com.github.zhengframework.jdbc.migrate;

import static com.google.inject.name.Names.named;

import com.github.zhengframework.configuration.ConfigurationAwareModule;
import com.github.zhengframework.configuration.ConfigurationBeanMapper;
import com.github.zhengframework.jdbc.ManagedSchema;
import com.google.inject.Key;
import com.google.inject.multibindings.OptionalBinder;
import java.util.Map;
import java.util.Map.Entry;
import lombok.EqualsAndHashCode;


@EqualsAndHashCode(callSuper = false)
public class LiquibaseMigrateModule extends ConfigurationAwareModule {

  @Override
  protected void configure() {
    Map liquibaseConfigMap = ConfigurationBeanMapper
        .resolve(getConfiguration(), LiquibaseConfig.class);
    for (Entry entry : liquibaseConfigMap.entrySet()) {
      String name = entry.getKey();
      LiquibaseConfig liquibaseConfig = entry.getValue();
      if (name.isEmpty()) {
        bind(LiquibaseConfig.class).toInstance(liquibaseConfig);
        OptionalBinder.newOptionalBinder(binder(), Key.get(ManagedSchema.class))
            .setBinding().toInstance(new LiquibaseManagedSchema(liquibaseConfig));
      } else {
        bind(Key.get(LiquibaseConfig.class, named(name))).toInstance(liquibaseConfig);
        OptionalBinder.newOptionalBinder(binder(), Key.get(ManagedSchema.class, named(name)))
            .setBinding().toInstance(new LiquibaseManagedSchema(liquibaseConfig));
      }
    }
  }

}