1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/*
* Copyright (c) 2013 Eike Stepper (Berlin, Germany) and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Eike Stepper - initial API and implementation
*/
package org.eclipse.emf.cdo.tests.db;
import org.eclipse.emf.cdo.common.CDOCommonRepository.IDGenerationLocation;
import org.eclipse.net4j.db.DBUtil;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
/**
* @author Victor Roldan Betancort
*/
public abstract class AbstractSetupDBConfig extends DBConfig
{
private static final long serialVersionUID = 1L;
private transient DataSource setupDataSource;
private transient List<String> databases = new ArrayList<String>();
public AbstractSetupDBConfig(String name, boolean supportingAudits, boolean supportingBranches, boolean withRanges,
boolean copyOnBranch, IDGenerationLocation idGenerationLocation)
{
super(name, supportingAudits, supportingBranches, withRanges, copyOnBranch, idGenerationLocation);
}
@Override
protected DataSource createDataSource(String repoName)
{
String dbName = "test_" + repoName;
initDatabase(dbName);
return createDataSourceForDB(dbName);
}
@Override
protected void deactivateRepositories()
{
super.deactivateRepositories();
for (String dbName : databases)
{
dropDatabase(dbName);
}
}
private void initDatabase(String dbName)
{
dropDatabase(dbName);
Connection connection = null;
Statement stmt = null;
try
{
connection = getSetupDataSource().getConnection();
stmt = connection.createStatement();
stmt.execute("CREATE DATABASE " + dbName);
}
catch (SQLException ignore)
{
}
finally
{
DBUtil.close(stmt);
DBUtil.close(connection);
}
}
private void dropDatabase(String dbName)
{
Connection connection = null;
Statement stmt = null;
try
{
connection = getSetupDataSource().getConnection();
stmt = connection.createStatement();
stmt.execute("DROP DATABASE " + dbName);
}
catch (SQLException ignore)
{
}
finally
{
DBUtil.close(stmt);
DBUtil.close(connection);
}
}
private DataSource getSetupDataSource()
{
if (setupDataSource == null)
{
setupDataSource = createDataSourceForDB(null);
}
return setupDataSource;
}
/**
* Note that <code>dbName</code> can be <code>null</code>, in which case a <i>setup</i> datasource must be returned.
* A connection form a setup< datasource can be used to create or drop other databases.
*/
protected abstract DataSource createDataSourceForDB(String dbName);
}
|