/*****************************************************************
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
****************************************************************/
package org.apache.cayenne.access.dbsync;
import org.apache.cayenne.CayenneRuntimeException;
import org.apache.cayenne.ObjectContext;
import org.apache.cayenne.access.DataNode;
import org.apache.cayenne.dba.DbAdapter;
import org.apache.cayenne.di.Inject;
import org.apache.cayenne.map.DataMap;
import org.apache.cayenne.query.SQLTemplate;
import org.apache.cayenne.unit.di.server.ServerCase;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
public class SchemaUpdateStrategyBase extends ServerCase {
@Inject
protected ObjectContext context;
@Inject
protected DataNode node;
@Inject
protected DbAdapter adapter;
@Override
public void cleanUpDB() throws Exception {
DataMap map = node.getEntityResolver().getDataMap("sus-map");
for (String name : existingTables()) {
for (String drop : adapter.dropTableStatements(map.getDbEntity(name))) {
context.performGenericQuery(new SQLTemplate(Object.class, drop));
}
}
}
protected void setStrategy(Class<? extends SchemaUpdateStrategy> type) throws Exception {
node.setSchemaUpdateStrategy(type.newInstance());
}
protected Collection<String> existingTables() {
Collection<String> present = new ArrayList<String>();
for (Entry<String, Boolean> e : tablesMap().entrySet()) {
if (e.getValue()) {
present.add(e.getKey());
}
}
return present;
}
protected void createOneTable(String entityName) {
DataMap map = node.getEntityResolver().getDataMap("sus-map");
String createTable = adapter.createTable(map.getDbEntity(entityName));
context.performGenericQuery(new SQLTemplate(Object.class, createTable));
}
protected Map<String, Boolean> tablesMap() {
DataMap map = node.getEntityResolver().getDataMap("sus-map");
Map<String, String> tables = new HashMap<>();
// add upper/lower case permutations
for (String name : map.getDbEntityMap().keySet()) {
tables.put(name.toLowerCase(), name);
tables.put(name.toUpperCase(), name);
}
Map<String, Boolean> presentInDB = new HashMap<>();
for (String name : map.getDbEntityMap().keySet()) {
presentInDB.put(name, false);
}
String tableLabel = node.getAdapter().tableTypeForTable();
try (Connection con = node.getDataSource().getConnection();) {
try (ResultSet rs = con.getMetaData().getTables(null, null, "%", new String[] { tableLabel });) {
while (rs.next()) {
String dbName = rs.getString("TABLE_NAME");
String name = tables.get(dbName);
if (name != null) {
presentInDB.put(name, true);
}
}
}
} catch (SQLException e) {
throw new CayenneRuntimeException(e);
}
return presentInDB;
}
}