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.utils.SonarException; 023 import org.sonar.core.persistence.Database; 024 import org.sonar.jpa.entity.SchemaMigration; 025 026 import java.sql.Connection; 027 import java.sql.SQLException; 028 029 public class DefaultDatabaseConnector extends AbstractDatabaseConnector { 030 031 public DefaultDatabaseConnector(Database database) { 032 super(database); 033 } 034 035 @Override 036 public boolean isOperational() { 037 if (isStarted() && getDatabaseVersion() != SchemaMigration.LAST_VERSION) { 038 // connector was started and connection OK but schema version was not OK 039 // call start again to check if this is now ok (schema created by rails) 040 start(); 041 } 042 return super.isOperational(); 043 } 044 045 @Override 046 public void start() { 047 if (!isStarted()) { 048 createDatasource(); 049 } 050 if (!super.isOperational()) { 051 super.start(); 052 } 053 } 054 055 private void createDatasource() { 056 try { 057 CustomHibernateConnectionProvider.setDatasourceForConfig(database.getDataSource()); 058 } catch (Exception e) { 059 throw new SonarException("Fail to connect to database", e); 060 } 061 } 062 063 public Connection getConnection() throws SQLException { 064 return database != null && database.getDataSource() != null ? database.getDataSource().getConnection() : null; 065 } 066 }