baseten / Sources / NSEntityDescription+BXPGAdditions.m

//
// NSEntityDescription+BXPGAdditions.m
// BaseTen
//
// Copyright (C) 2006-2008 Marko Karppinen & Co. LLC.
//
// Before using this software, please review the available licensing options
// by visiting http://basetenframework.org/licensing/ or by contacting
// us at sales@karppinen.fi. Without an additional license, this software
// may be distributed only in compliance with the GNU General Public License.
//
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License, version 2.0,
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
//
// $Id$
//

#import "NSEntityDescription+BXPGAdditions.h"
#import "NSAttributeDescription+BXPGAdditions.h"
#import "BXLogger.h"
#import "BXEnumerate.h"
#import "PGTSOids.h"


@implementation NSEntityDescription (BXPGAdditions)
- (NSString *) BXPGCreateStatementWithIDColumn: (BOOL) addSerialIDColumn 
									  inSchema: (NSString *) schemaName
										errors: (NSMutableArray *) errors
{
	Expect (schemaName);

	NSString* name = [self name];
	NSEntityDescription* superentity = [self superentity];
	NSDictionary* attributes = [self attributesByName];
    NSMutableArray* attributeDefs = [NSMutableArray arrayWithCapacity: 1 + [attributes count]];    
    if (YES == addSerialIDColumn)
        [attributeDefs addObject: @"id SERIAL"];
	
	BXEnumerate (currentAttribute, e, [attributes objectEnumerator])
	{
		//Transient values are not stored
		if ([currentAttribute isTransient])
			continue;
		
		//Superentities' attributes won't be repeated here.
		if (! [[currentAttribute entity] isEqual: self])
			continue;
		
		NSError* attrError = nil;
		if (! [currentAttribute BXCanAddAttribute: &attrError])
			[errors addObject: attrError];
		else
		{
			NSString* attrDef = [currentAttribute BXPGAttributeDefinition];
			[attributeDefs addObject: attrDef];
		}
	}
	
	NSString* addition = @"";
	if (superentity)
		addition = [NSString stringWithFormat: @"INHERITS (\"%@\".\"%@\")", schemaName, [superentity name]];
	
	NSString* statementFormat = @"CREATE TABLE \"%@\".\"%@\" (%@) %@;";
	NSString* retval = [NSString stringWithFormat: statementFormat, schemaName, name,
						[attributeDefs componentsJoinedByString: @", "], addition];
	
	return retval;
}

- (NSString *) BXPGPrimaryKeyConstraintInSchema: (NSString *) schemaName
{
	NSString* format = @"ALTER TABLE \"%@\".\"%@\" ADD PRIMARY KEY (id);";
	NSString* constraint = [NSString stringWithFormat: format, schemaName, [self name]];
	return constraint;
}
@end
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.