Files
RestKit/Code/CoreData/RKManagedObjectMapping.m
Blake Watters f3c0995d5e Implementation of Object Mapping 2.0 design:
* Removed RestKit from inheritance hierarchy
  * Mappings are implemented as concrete classes
  * Mapper is much more flexible & powerful
  * Much more robust error handling
  * Serialization is reimplemented as an object mapping operation
  * Added ability to serialize to JSON natively
  * Reworked Core Data integration
  * Simplified the codebase substantially
2011-06-11 19:26:56 -04:00

77 lines
2.6 KiB
Objective-C

//
// RKManagedObjectMapping.m
// RestKit
//
// Created by Blake Watters on 5/31/11.
// Copyright 2011 Two Toasters. All rights reserved.
//
#import "RKManagedObjectMapping.h"
#import "NSManagedObject+ActiveRecord.h"
@implementation RKManagedObjectMapping
@synthesize entity = _entity;
@synthesize primaryKeyAttribute = _primaryKeyAttribute;
+ (id)mappingForClass:(Class)objectClass {
return [self mappingForEntityWithName:NSStringFromClass(objectClass)];
}
+ (RKManagedObjectMapping*)mappingForEntity:(NSEntityDescription*)entity {
return [[[self alloc] initWithEntity:entity] autorelease];
}
+ (RKManagedObjectMapping*)mappingForEntityWithName:(NSString*)entityName {
return [self mappingForEntity:[NSEntityDescription entityForName:entityName inManagedObjectContext:[NSManagedObject managedObjectContext]]];
}
- (id)initWithEntity:(NSEntityDescription*)entity {
NSAssert(entity, @"Cannot initialize an RKManagedObjectMapping without an entity. Maybe you want RKObjectMapping instead?");
self = [self init];
if (self) {
self.objectClass = NSClassFromString([entity managedObjectClassName]);
_entity = [entity retain];
}
return self;
}
- (id)init {
self = [super init];
if (self) {
_relationshipToPrimaryKeyMappings = [[NSMutableDictionary alloc] init];
}
return self;
}
- (void)dealloc {
[_entity release];
[_relationshipToPrimaryKeyMappings release];
[super dealloc];
}
- (NSDictionary*)relationshipsAndPrimaryKeyAttributes {
return _relationshipToPrimaryKeyMappings;
}
- (void)connectRelationship:(NSString*)relationshipName withObjectForPrimaryKeyAttribute:(NSString*)primaryKeyAttribute {
NSAssert([_relationshipToPrimaryKeyMappings objectForKey:relationshipName] == nil, @"Cannot add connect relationship %@ by primary key, a mapping already exists.", relationshipName);
[_relationshipToPrimaryKeyMappings setObject:primaryKeyAttribute forKey:relationshipName];
}
- (void)connectRelationshipsWithObjectsForPrimaryKeyAttributes:(NSString*)firstRelationshipName, ... {
va_list args;
va_start(args, firstRelationshipName);
for (NSString* relationshipName = firstRelationshipName; relationshipName != nil; relationshipName = va_arg(args, NSString*)) {
NSString* primaryKeyAttribute = va_arg(args, NSString*);
NSAssert(primaryKeyAttribute != nil, @"Cannot connect a relationship without an attribute containing the primary key");
[self connectRelationship:relationshipName withObjectForPrimaryKeyAttribute:primaryKeyAttribute];
// TODO: Raise proper exception here, argument error...
}
va_end(args);
}
@end