com.github.zhengframework.jpa.JpaInternalModule Maven / Gradle / Ivy

Go to download
package com.github.zhengframework.jpa;

/*-
 * #%L
 * zheng-jpa
 * %%
 * Copyright (C) 2020 Zheng MingHai
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import com.github.zhengframework.jpa.JpaService.EntityManagerFactoryInternalProvider;
import com.google.inject.AbstractModule;
import com.google.inject.matcher.Matchers;
import com.google.inject.multibindings.OptionalBinder;
import com.google.inject.persist.PersistService;
import com.google.inject.persist.Transactional;
import com.google.inject.persist.UnitOfWork;
import javax.inject.Singleton;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.spi.PersistenceUnitInfo;
import org.aopalliance.intercept.MethodInterceptor;

public class JpaInternalModule extends AbstractModule {

  private final JpaConfig persistenceConfig;

  public JpaInternalModule(JpaConfig persistenceConfig) {
    this.persistenceConfig = persistenceConfig;
  }

  @Override
  protected void configure() {
    bind(JpaConfig.class).toInstance(persistenceConfig);
    OptionalBinder.newOptionalBinder(binder(), PersistenceUnitInfo.class)
        .setDefault()
        .toProvider(PersistenceUnitInfoProvider.class);

    bind(JpaService.class).in(Singleton.class);
    bind(PersistService.class).to(JpaService.class);
    bind(UnitOfWork.class).to(JpaService.class);
    bind(EntityManager.class).toProvider(JpaService.class);
    bind(EntityManagerFactory.class).toProvider(EntityManagerFactoryInternalProvider.class);

    bind(JpaManagedService.class).asEagerSingleton();

    bindInterceptor(
        Matchers.annotatedWith(Transactional.class), Matchers.any(), getTransactionInterceptor());
    bindInterceptor(
        Matchers.any(), Matchers.annotatedWith(Transactional.class), getTransactionInterceptor());

    bindInterceptor(
        Matchers.annotatedWith(javax.transaction.Transactional.class),
        Matchers.any(),
        getJavaxTransactionInterceptor());
    bindInterceptor(
        Matchers.any(),
        Matchers.annotatedWith(javax.transaction.Transactional.class),
        getJavaxTransactionInterceptor());
  }

  protected MethodInterceptor getTransactionInterceptor() {
    MethodInterceptor txInterceptor = new GuiceTransactionInterceptor();
    requestInjection(txInterceptor);
    return txInterceptor;
  }

  protected MethodInterceptor getJavaxTransactionInterceptor() {
    MethodInterceptor txInterceptor = new JavaxTransactionInterceptor();
    requestInjection(txInterceptor);
    return txInterceptor;
  }
}