Respond to CR.

// FREEBIE
pull/1/head
Matthew Chen 8 years ago
parent cf65cc3be5
commit beb4ed71e7

@ -27,8 +27,6 @@ NS_ASSUME_NONNULL_BEGIN
// This only applies for attachments being uploaded. // This only applies for attachments being uploaded.
@property (atomic) BOOL isUploaded; @property (atomic) BOOL isUploaded;
@property (atomic) BOOL hasUnsavedFilePath;
#if TARGET_OS_IPHONE #if TARGET_OS_IPHONE
- (nullable UIImage *)image; - (nullable UIImage *)image;
#endif #endif
@ -54,6 +52,8 @@ NS_ASSUME_NONNULL_BEGIN
- (CGFloat)audioDurationSecondsWithTransaction:(YapDatabaseReadWriteTransaction *)transaction; - (CGFloat)audioDurationSecondsWithTransaction:(YapDatabaseReadWriteTransaction *)transaction;
- (CGFloat)audioDurationSecondsWithoutTransaction; - (CGFloat)audioDurationSecondsWithoutTransaction;
//- (BOOL)hasUnsavedFilePath;
@end @end
NS_ASSUME_NONNULL_END NS_ASSUME_NONNULL_END

@ -42,8 +42,7 @@ NS_ASSUME_NONNULL_BEGIN
// attachments which haven't been uploaded yet. // attachments which haven't been uploaded yet.
_isUploaded = NO; _isUploaded = NO;
// This instance hasn't been persisted yet. [self ensureFilePath];
[self ensureFilePathAndPersist:NO];
return self; return self;
} }
@ -64,8 +63,7 @@ NS_ASSUME_NONNULL_BEGIN
_isUploaded = YES; _isUploaded = YES;
self.attachmentType = pointer.attachmentType; self.attachmentType = pointer.attachmentType;
// This instance hasn't been persisted yet. [self ensureFilePath];
[self ensureFilePathAndPersist:NO];
return self; return self;
} }
@ -77,9 +75,7 @@ NS_ASSUME_NONNULL_BEGIN
return self; return self;
} }
// This instance has been persisted, we need to [self ensureFilePath];
// update it in the database.
[self ensureFilePathAndPersist:YES];
return self; return self;
} }
@ -97,7 +93,7 @@ NS_ASSUME_NONNULL_BEGIN
} }
} }
- (void)ensureFilePathAndPersist:(BOOL)shouldPersist - (void)ensureFilePath
{ {
if (self.localRelativeFilePath) { if (self.localRelativeFilePath) {
return; return;
@ -127,20 +123,6 @@ NS_ASSUME_NONNULL_BEGIN
self.localRelativeFilePath = localRelativeFilePath; self.localRelativeFilePath = localRelativeFilePath;
OWSAssert(self.filePath); OWSAssert(self.filePath);
if (shouldPersist) {
self.hasUnsavedFilePath = YES;
}
}
+ (MTLPropertyStorage)storageBehaviorForPropertyWithKey:(NSString *)propertyKey
{
// Don't persist transient properties
if ([propertyKey isEqualToString:@"hasUnsavedFilePath"]) {
return MTLPropertyStorageNone;
} else {
return [super storageBehaviorForPropertyWithKey:propertyKey];
}
} }
#pragma mark - File Management #pragma mark - File Management

Loading…
Cancel
Save