73 lines
1.9 KiB
Objective-C
73 lines
1.9 KiB
Objective-C
//
|
|
// TSYapDatabaseObject.m
|
|
// TextSecureKit
|
|
//
|
|
// Created by Frederic Jacobs on 16/11/14.
|
|
// Copyright (c) 2014 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "TSYapDatabaseObject.h"
|
|
#import "TSStorageManager.h"
|
|
|
|
@implementation TSYapDatabaseObject
|
|
|
|
- (id)init{
|
|
if (self = [super init])
|
|
{
|
|
_uniqueId = [[NSUUID UUID] UUIDString];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithUniqueId:(NSString *)aUniqueId{
|
|
if (self = [super init]) {
|
|
_uniqueId = aUniqueId;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (void)saveWithTransaction:(YapDatabaseReadWriteTransaction *)transaction{
|
|
[transaction setObject:self forKey:self.uniqueId inCollection:[[self class] collection]];
|
|
}
|
|
|
|
- (void)save{
|
|
[[TSStorageManager sharedManager].dbConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
|
|
[self saveWithTransaction:transaction];
|
|
}];
|
|
}
|
|
|
|
- (void)removeWithTransaction:(YapDatabaseReadWriteTransaction *)transaction{
|
|
[transaction removeObjectForKey:self.uniqueId inCollection:[[self class] collection]];
|
|
}
|
|
|
|
|
|
- (void)remove{
|
|
[[TSStorageManager sharedManager].dbConnection readWriteWithBlock:^(YapDatabaseReadWriteTransaction *transaction) {
|
|
[self removeWithTransaction:transaction];
|
|
[[transaction ext:@"relationships"] flush];
|
|
}];
|
|
}
|
|
|
|
|
|
#pragma mark Class Methods
|
|
|
|
+ (NSString *)collection{
|
|
return NSStringFromClass([self class]);
|
|
}
|
|
|
|
+ (instancetype) fetchObjectWithUniqueID:(NSString *)uniqueID transaction:(YapDatabaseReadTransaction *)transaction {
|
|
return [transaction objectForKey:uniqueID inCollection:[self collection]];
|
|
}
|
|
|
|
+ (instancetype) fetchObjectWithUniqueID:(NSString *)uniqueID{
|
|
__block id object;
|
|
|
|
[[TSStorageManager sharedManager].dbConnection readWithBlock:^(YapDatabaseReadTransaction *transaction) {
|
|
object = [transaction objectForKey:uniqueID inCollection:[self collection]];
|
|
}];
|
|
|
|
return object;
|
|
}
|
|
|
|
@end
|