io.github.mmm.orm.jdbc.tx.JdbcSessionHolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mmm-orm-jdbc Show documentation

JDBC support for mmm-orm.

The newest version!

/* Copyright (c) The m-m-m Team, Licensed under the Apache License, Version 2.0
 * http://www.apache.org/licenses/LICENSE-2.0 */
package io.github.mmm.orm.jdbc.tx;

import java.util.HashMap;
import java.util.Map;

import io.github.mmm.orm.jdbc.session.JdbcSession;
import io.github.mmm.orm.source.DbSource;

/**
 * Holder for {@link JdbcSession} by {@link DbSource}.
 */
public class JdbcSessionHolder {

  JdbcSession session;

  private Map sessionMap;

  /**
   * The constructor.
   */
  public JdbcSessionHolder() {

    super();
  }

  /**
   * @param source the {@link DbSource}.
   * @return the {@link JdbcSession}.
   */
  public JdbcSession getSession(DbSource source) {

    if ((source == null) || (DbSource.get() == source)) {
      return this.session;
    }
    if (this.sessionMap == null) {
      return null;
    }
    return this.sessionMap.get(source);
  }

  void setSession(DbSource source, JdbcSession session) {

    if ((source == null) || (DbSource.get() == source)) {
      this.session = session;
    } else {
      if (this.sessionMap == null) {
        this.sessionMap = new HashMap<>();
      }
      this.sessionMap.put(source, session);
    }

  }

}