org.sonar.core.persistence.DryRunDatabaseFactory.java Source code

Introduction

Here is the source code for org.sonar.core.persistence.DryRunDatabaseFactory.java

Source

/*
 * Sonar, open source software quality management tool.
 * Copyright (C) 2008-2012 SonarSource
 * mailto:contact AT sonarsource DOT com
 *
 * Sonar is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * Sonar is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Sonar; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02
 */
package org.sonar.core.persistence;

import com.google.common.io.Files;
import org.apache.commons.dbcp.BasicDataSource;
import org.sonar.api.ServerComponent;
import org.sonar.api.platform.ServerFileSystem;
import org.sonar.api.utils.SonarException;
import org.sonar.core.review.ReviewDto;

import javax.annotation.Nullable;
import javax.sql.DataSource;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;

public class DryRunDatabaseFactory implements ServerComponent {
    private static final String DIALECT = "h2";
    private static final String DRIVER = "org.h2.Driver";
    private static final String URL = "jdbc:h2:";
    private static final String USER = "sonar";
    private static final String PASSWORD = "sonar";

    private final Database database;
    private final ServerFileSystem serverFileSystem;

    public DryRunDatabaseFactory(Database database, ServerFileSystem serverFileSystem) {
        this.database = database;
        this.serverFileSystem = serverFileSystem;
    }

    public byte[] createDatabaseForDryRun(@Nullable Long projectId) {
        String name = serverFileSystem.getTempDir().getAbsolutePath() + "db-" + System.nanoTime();

        try {
            DataSource source = database.getDataSource();
            BasicDataSource destination = create(DIALECT, DRIVER, USER, PASSWORD, URL + name);

            copy(source, destination, projectId);
            close(destination);

            return dbFileContent(name);
        } catch (SQLException e) {
            throw new SonarException("Unable to create database for DryRun", e);
        }
    }

    private void copy(DataSource source, DataSource dest, @Nullable Long projectId) {
        DbTemplate template = new DbTemplate();
        template.copyTable(source, dest, "active_rules").copyTable(source, dest, "active_rule_parameters")
                .copyTable(source, dest, "characteristics").copyTable(source, dest, "characteristic_edges")
                .copyTable(source, dest, "characteristic_properties").copyTable(source, dest, "metrics")
                .copyTable(source, dest, "quality_models").copyTable(source, dest, "rules")
                .copyTable(source, dest, "rules_parameters").copyTable(source, dest, "rules_profiles");
        if (projectId != null) {
            String snapshotCondition = "islast=" + database.getDialect().getTrueSqlValue() + " and (project_id="
                    + projectId + " or root_project_id=" + projectId + ")";
            template.copyTable(source, dest, "projects", "(id=" + projectId + " or root_id=" + projectId + ")")
                    .copyTable(source, dest, "reviews", "project_id=" + projectId,
                            "status<>'" + ReviewDto.STATUS_CLOSED + "'")
                    .copyTable(source, dest, "rule_failures",
                            "snapshot_id in (select id from snapshots where " + snapshotCondition + ")")
                    .copyTable(source, dest, "snapshots", snapshotCondition);
        }
    }

    private BasicDataSource create(String dialect, String driver, String user, String password, String url) {
        BasicDataSource dataSource = new DbTemplate().dataSource(driver, user, password, url);
        new DbTemplate().createSchema(dataSource, dialect);
        return dataSource;
    }

    private void close(BasicDataSource destination) throws SQLException {
        destination.close();
    }

    private byte[] dbFileContent(String name) {
        try {
            File dbFile = new File(name + ".h2.db");
            byte[] content = Files.toByteArray(dbFile);
            dbFile.delete();
            return content;
        } catch (IOException e) {
            throw new SonarException("Unable to read h2 database file", e);
        }
    }
}