kylefang/MagicalRecord
1
0
Fork 0
mirror of https://github.com/zhigang1992/MagicalRecord.git synced 2026-01-12 17:32:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for Importing ordered relationships; Added shouldImport: callback for any pre-import validation

Browse Source
This commit is contained in:
Saul Mora
2011-11-29 23:36:46 -07:00
parent f3d3f2b082
commit 3e19b336ef
1 changed files with 27 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
Source/Categories/NSManagedObject+MagicalDataImport.m
View File

@@ -115,16 +115,30 @@ NSString * const kMagicalRecordImportRelationshipTypeKey = @"type";
NSAssert2([relatedObject entity] == [relationshipInfo destinationEntity], @"related object entity %@ not same as destination entity %@", [relatedObject entity], [relationshipInfo destinationEntity]);
//add related object to set
NSString *addRelationMessageFormat = [relationshipInfo isToMany] ? @"add%@Object:" : @"set%@:";
NSString *addRelationMessageFormat = @"set%@:";
id relationshipSource = self;
if ([relationshipInfo isToMany])
{
addRelationMessageFormat = @"add%@Object:";
if ([relationshipInfo isOrdered])
{
//Need to get the ordered set
NSString *selectorName = [[relationshipInfo name] stringByAppendingString:@"Set"];
relationshipSource = [self performSelector:NSSelectorFromString(selectorName)];
addRelationMessageFormat = @"addObject:";
}
}
NSString *addRelatedObjectToSetMessage = [NSString stringWithFormat:addRelationMessageFormat, attributeNameFromString([relationshipInfo name])];
SEL selector = NSSelectorFromString(addRelatedObjectToSetMessage);
MRLog(@"add selector: %@", addRelatedObjectToSetMessage);
@try
{
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
[self performSelector:selector withObject:relatedObject];
[relationshipSource performSelector:selector withObject:relatedObject];
#pragma clang diagnostic pop
}
@catch (NSException *exception)
@@ -152,18 +166,28 @@ NSString * const kMagicalRecordImportRelationshipTypeKey = @"type";
continue;
}
SEL shouldImportSelector = @selector(shouldImport:);
BOOL implementsShouldImport = [self respondsToSelector:shouldImportSelector];
if (![self MR_importValue:relatedObjectData forKey:relationshipName])
{
if ([relationshipInfo isToMany])
{
for (id singleRelatedObjectData in relatedObjectData)
{
if (implementsShouldImport && !(BOOL)[self performSelector:shouldImportSelector withObject:singleRelatedObjectData])
{
continue;
}
setRelationshipBlock(relationshipInfo, singleRelatedObjectData);
}
}
else
{
setRelationshipBlock(relationshipInfo, relatedObjectData);
if (!(implementsShouldImport && !(BOOL)[self performSelector:shouldImportSelector withObject:relatedObjectData]))
{
setRelationshipBlock(relationshipInfo, relatedObjectData);
}
}
}
}