001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.jpa.session; 021 022 import org.sonar.api.database.DatabaseSession; 023 024 public class ThreadLocalDatabaseSessionFactory implements DatabaseSessionFactory { 025 026 private final ThreadLocal<JpaDatabaseSession> threadSession = new ThreadLocal<JpaDatabaseSession>(); 027 private final DatabaseConnector connector; 028 029 public ThreadLocalDatabaseSessionFactory(DatabaseConnector connector) { 030 this.connector = connector; 031 } 032 033 public DatabaseSession getSession() { 034 JpaDatabaseSession session = threadSession.get(); 035 if (session == null) { 036 session = new JpaDatabaseSession(connector); 037 session.start(); 038 threadSession.set(session); 039 } 040 return session; 041 } 042 043 public void clear() { 044 JpaDatabaseSession session = threadSession.get(); 045 if (session != null) { 046 session.stop(); 047 } 048 threadSession.set(null); 049 } 050 051 public void stop() { 052 clear(); 053 } 054 }