001 /* 002 * JBoss DNA (http://www.jboss.org/dna) 003 * See the COPYRIGHT.txt file distributed with this work for information 004 * regarding copyright ownership. Some portions may be licensed 005 * to Red Hat, Inc. under one or more contributor license agreements. 006 * See the AUTHORS.txt file in the distribution for a full listing of 007 * individual contributors. 008 * 009 * JBoss DNA is free software. Unless otherwise indicated, all code in JBoss DNA 010 * is licensed to you under the terms of the GNU Lesser General Public License as 011 * published by the Free Software Foundation; either version 2.1 of 012 * the License, or (at your option) any later version. 013 * 014 * JBoss DNA is distributed in the hope that it will be useful, 015 * but WITHOUT ANY WARRANTY; without even the implied warranty of 016 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 017 * Lesser General Public License for more details. 018 * 019 * You should have received a copy of the GNU Lesser General Public 020 * License along with this software; if not, write to the Free 021 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 022 * 02110-1301 USA, or see the FSF site: http://www.fsf.org. 023 */ 024 package org.jboss.dna.common.jdbc.model.spi; 025 026 import java.util.Set; 027 import java.util.HashSet; 028 import org.jboss.dna.common.jdbc.model.api.PrimaryKey; 029 import org.jboss.dna.common.jdbc.model.api.PrimaryKeyColumn; 030 031 /** 032 * Provides database table's primary key specific metadata. 033 * 034 * @author <a href="mailto:litsenko_sergey@yahoo.com">Sergiy Litsenko</a> 035 */ 036 public class PrimaryKeyBean extends SchemaObjectBean implements PrimaryKey { 037 private static final long serialVersionUID = -6620964715120791723L; 038 private Set<PrimaryKeyColumn> columns = new HashSet<PrimaryKeyColumn>(); 039 040 /** 041 * Default constructor 042 */ 043 public PrimaryKeyBean() { 044 } 045 046 /** 047 * Retrieves primary key columns 048 * 049 * @return primary key columns 050 */ 051 public Set<PrimaryKeyColumn> getColumns() { 052 return columns; 053 } 054 055 /** 056 * Adds PrimaryKeyColumn 057 * 058 * @param primaryKeyColumn the PrimaryKeyColumn 059 */ 060 public void addColumn( PrimaryKeyColumn primaryKeyColumn ) { 061 columns.add(primaryKeyColumn); 062 } 063 064 /** 065 * deletes PrimaryKeyColumn 066 * 067 * @param primaryKeyColumn the PrimaryKeyColumn 068 */ 069 public void deleteColumn( PrimaryKeyColumn primaryKeyColumn ) { 070 columns.remove(primaryKeyColumn); 071 } 072 073 /** 074 * Returns table column for specified column name or null 075 * 076 * @param columnName the name of column 077 * @return table column for specified column name or null. 078 */ 079 public PrimaryKeyColumn findColumnByName( String columnName ) { 080 for (PrimaryKeyColumn pkc : columns) { 081 if (pkc.getName().equals(columnName)) { 082 return pkc; 083 } 084 } 085 // return nothing 086 return null; 087 } 088 089 }