/*
 * 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.    
 */
#include "apache/das/rdb/TableData.h"

namespace apache {
	namespace das {
		namespace rdb {

TableData::TableData(const Table& table, ResultSetPtr resultSet) : PKObject(table) {
	dataObject = 0;
	const std::map<std::string, const Column*>& columns = table.getColumns();
	std::map<std::string, const Column*>::const_iterator columnIterator;
	const ResultSetMetaData& metaData = resultSet->getResultSetMetaData();
	const KeyDataList& primaryKeys = getPrimaryKeys();
	
	for (columnIterator = columns.begin() ; columnIterator != columns.end() ; 
		columnIterator++) {

			const Column& column = *columnIterator->second;
			std::string columnName = column.getName();

			if (metaData.containsColumn(table.getTableName(), columnName)) {
				ColumnData* columnData = new ColumnData(column, resultSet);

				columnsData.insert(std::make_pair(columnName, columnData));

				if (column.isPK()) {
					addPrimaryKey(columnName, *columnData);
				}

			}

	}

}

TableData::~TableData(void) {
	std::map<std::string, ColumnData*>::iterator it;

	for (it = columnsData.begin() ; it != columnsData.end() ; it++) {

		if (!(it->second->getColumn().isPK())) {
			delete it->second;
		}

	}
	
}

bool TableData::hasPK(void) const {
	const KeyDataList& primaryKeys = getPrimaryKeys();
	return (primaryKeys.size() == getTable().getPKColumnCount() && primaryKeys.size() > 0);
}

commonj::sdo::DataObjectPtr TableData::getGraphObject(void) const {
	return dataObject;
}

void TableData::populateDataGraph(GraphBuilder& graphBuilder) {
	dataObject = graphBuilder.getRoot()->createDataObject(getTable().getTableName().c_str());
	std::map<std::string, ColumnData*>::iterator it;

	for (it = columnsData.begin() ; it != columnsData.end() ; it++) {
		it->second->populateDataGraph(*this);
	}

}

ColumnData* TableData::getColumnData(std::string columnName) const {
	std::map<std::string, ColumnData*>::const_iterator it = columnsData.find(columnName);

	if (it == columnsData.end()) {
		return 0;
	}

	return it->second;

}

		};
	};
};