kz.greetgo.db.GreetgoTransactionManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of greetgo.db Show documentation

DB utils using in greetgo!

The newest version!

package kz.greetgo.db;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;

public class GreetgoTransactionManager implements TransactionManager {

  private static final ExceptionCatcher DEFAULT_EXCEPTION_CATCHER = Throwable::printStackTrace;

  private ExceptionCatcher exceptionCatcher = DEFAULT_EXCEPTION_CATCHER;

  private final ExceptionCatcherGetter exceptionCatcherGetter = () -> exceptionCatcher;

  public void setExceptionCatcher(ExceptionCatcher exceptionCatcher) {
    this.exceptionCatcher = exceptionCatcher == null ? DEFAULT_EXCEPTION_CATCHER : exceptionCatcher;
  }

  private final ThreadLocal threadLocalTM
    = ThreadLocal.withInitial(() -> new ThreadLocalTransactionManager(exceptionCatcherGetter));

  @Override
  public Connection getConnection(DataSource dataSource) {
    try {
      return threadLocalTM.get().getConnection(dataSource);
    } catch (SQLException e) {
      throw new SQLRuntimeException(e);
    }
  }

  @Override
  public void upLevel(CallMeta callMeta) {
    threadLocalTM.get().upLevel(callMeta);
  }

  @Override
  public void downLevel(Throwable throwable) {
    threadLocalTM.get().downLevel(throwable);
  }

  @Override
  public void closeConnection(DataSource dataSource, Connection connection) {
    try {
      threadLocalTM.get().closeConnection(dataSource, connection);
    } catch (SQLException e) {
      throw new SQLRuntimeException(e);
    }
  }
}