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 or by contacting
// us at 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
// 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 "PGTSFunctions.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"];
	TSEnumerate (currentAttribute, e, [attributes objectEnumerator])
		//Transient values are not stored
		if ([currentAttribute isTransient])
		//Superentities' attributes won't be repeated here.
		if (! [[currentAttribute entity] isEqual: self])
		NSError* attrError = nil;
		if (! [currentAttribute BXCanAddAttribute: &attrError])
			[errors addObject: attrError];
			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;
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
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.