01: /*
02: *
03: * The DbUnit Database Testing Framework
04: * Copyright (C)2002-2004, DbUnit.org
05: *
06: * This library is free software; you can redistribute it and/or
07: * modify it under the terms of the GNU Lesser General Public
08: * License as published by the Free Software Foundation; either
09: * version 2.1 of the License, or (at your option) any later version.
10: *
11: * This library is distributed in the hope that it will be useful,
12: * but WITHOUT ANY WARRANTY; without even the implied warranty of
13: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14: * Lesser General Public License for more details.
15: *
16: * You should have received a copy of the GNU Lesser General Public
17: * License along with this library; if not, write to the Free Software
18: * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19: *
20: */
21:
22: package org.dbunit.dataset;
23:
24: import org.slf4j.Logger;
25: import org.slf4j.LoggerFactory;
26:
27: import org.dbunit.dataset.filter.IColumnFilter;
28:
29: import java.util.ArrayList;
30: import java.util.List;
31:
32: /**
33: * @author Manuel Laflamme
34: * @version $Revision: 554 $
35: * @since Mar 8, 2002
36: */
37: public abstract class AbstractTableMetaData implements ITableMetaData {
38:
39: /**
40: * Logger for this class
41: */
42: private static final Logger logger = LoggerFactory
43: .getLogger(AbstractTableMetaData.class);
44:
45: private static final Column[] EMPTY_COLUMNS = new Column[0];
46:
47: protected static Column[] getPrimaryKeys(Column[] columns,
48: String[] keyNames) {
49: logger.debug("getPrimaryKeys(columns=" + columns
50: + ", keyNames=" + keyNames + ") - start");
51:
52: if (keyNames == null || keyNames.length == 0) {
53: return EMPTY_COLUMNS;
54: }
55:
56: List keyList = new ArrayList();
57: for (int i = 0; i < keyNames.length; i++) {
58: Column primaryKey = DataSetUtils.getColumn(keyNames[i],
59: columns);
60: if (primaryKey != null) {
61: keyList.add(primaryKey);
62: }
63: }
64:
65: return (Column[]) keyList.toArray(new Column[0]);
66: }
67:
68: protected static Column[] getPrimaryKeys(String tableName,
69: Column[] columns, IColumnFilter columnFilter) {
70: logger.debug("getPrimaryKeys(tableName=" + tableName
71: + ", columns=" + columns + ", columnFilter="
72: + columnFilter + ") - start");
73:
74: List keyList = new ArrayList();
75: for (int i = 0; i < columns.length; i++) {
76: Column column = columns[i];
77: if (columnFilter.accept(tableName, column)) {
78: keyList.add(column);
79: }
80: }
81:
82: return (Column[]) keyList.toArray(new Column[0]);
83: }
84: }
|