org.tinygroup.tinydb.config.TableConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.tinydb Show documentation
There is a newer version: 2.2.3
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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.tinygroup.tinydb.config;

import org.tinygroup.commons.tools.CollectionUtil;
import org.tinygroup.tinydb.exception.TinyDbException;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 * 表信息
 * 
 * @author luoguo
 * 
 */
public class TableConfiguration implements Serializable {
	/**
	 * 
	 */
	private static final long serialVersionUID = 474364262682757321L;
	/**
	 * 表名
	 */
	private String name;
	/**
	 * 表模式
	 */
	private String schema;
	/**
	 * 主键类型
	 */
	private String keyType;
	/**
	 * 字段信息列表
	 */
	private List columns = new ArrayList();

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public List getColumns() {
		return columns;
	}

	public void setColumns(List columns) {
		this.columns = columns;
	}
	

	public String getSchema() {
		return schema;
	}

	public void setSchema(String schema) {
		this.schema = schema;
	}

	public void setPrimaryKey(String columnName) throws TinyDbException {
		boolean flag = false;
		for (ColumnConfiguration column : columns) {
			if (column.getColumnName().equals(columnName.toUpperCase())) {
				column.setPrimaryKey(true);
				flag = true;
				break;
			}
		}
		if (!flag) {
			throw new TinyDbException("表格主键字段" + columnName + "不存在");
		}
	}

	public ColumnConfiguration getPrimaryKey() throws TinyDbException {
		for (ColumnConfiguration column : columns) {
			if (column.isPrimaryKey()) {
				return column;
			}
		}
		throw new TinyDbException("找不到主键字段！");
	}
	
	public void addColumn(ColumnConfiguration column){
		columns.add(column);
	}
	
	public void removeColumn(ColumnConfiguration column){
		columns.remove(column);
	}

	public String getKeyType() {
		return keyType;
	}

	public void setKeyType(String keyType) {
		this.keyType = keyType;
	}

	public ColumnConfiguration getColumn(String columnName){
		if(!CollectionUtil.isEmpty(columns)){
			for (ColumnConfiguration column : columns) {
				if(column.getColumnName().equalsIgnoreCase(columnName)){
					return column;
				}
			}
		}
		return null;
	}
	
}