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.jpa.entity.SchemaMigration; 023 import org.sonar.persistence.Database; 024 025 import javax.persistence.EntityManager; 026 027 public class MemoryDatabaseConnector extends DefaultDatabaseConnector { 028 private int version; 029 030 public MemoryDatabaseConnector(Database database) { 031 super(database); 032 version = SchemaMigration.LAST_VERSION; 033 } 034 035 public MemoryDatabaseConnector(Database database, int version) { 036 this(database); 037 this.version = version; 038 } 039 040 @Override 041 public void start() { 042 try { 043 super.start(); 044 } catch (DatabaseException ex) { 045 if (!isStarted()) { 046 throw ex; 047 } 048 } 049 setEntityManagerFactory(createEntityManagerFactory()); 050 setupSchemaVersion(version); 051 } 052 053 protected void setupSchemaVersion(int version) { 054 SchemaMigration migration = new SchemaMigration(); 055 migration.setVersion(version); 056 EntityManager manager = null; 057 try { 058 manager = createEntityManager(); 059 manager.getTransaction().begin(); 060 manager.persist(migration); 061 manager.getTransaction().commit(); 062 063 } finally { 064 if (manager != null) { 065 manager.close(); 066 } 067 } 068 } 069 }