111 lines
3.8 KiB
Objective-C
111 lines
3.8 KiB
Objective-C
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "YapDatabaseTransaction+OWS.h"
|
|
#import <Curve25519Kit/Curve25519.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
@implementation YapDatabaseReadTransaction (OWS)
|
|
|
|
- (nullable id)objectForKey:(NSString *)key inCollection:(NSString *)collection ofExpectedType:(Class) class {
|
|
id _Nullable value = [self objectForKey:key inCollection:collection];
|
|
return value;
|
|
}
|
|
|
|
- (nullable NSDictionary *)dictionaryForKey : (NSString *)key inCollection : (NSString *)collection
|
|
{
|
|
return [self objectForKey:key inCollection:collection ofExpectedType:[NSDictionary class]];
|
|
}
|
|
|
|
- (nullable NSString *)stringForKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
return [self objectForKey:key inCollection:collection ofExpectedType:[NSString class]];
|
|
}
|
|
|
|
- (BOOL)boolForKey:(NSString *)key inCollection:(NSString *)collection defaultValue:(BOOL)defaultValue
|
|
{
|
|
NSNumber *_Nullable value = [self objectForKey:key inCollection:collection ofExpectedType:[NSNumber class]];
|
|
return value ? [value boolValue] : defaultValue;
|
|
}
|
|
|
|
- (nullable NSData *)dataForKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
return [self objectForKey:key inCollection:collection ofExpectedType:[NSData class]];
|
|
}
|
|
|
|
- (nullable ECKeyPair *)keyPairForKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
return [self objectForKey:key inCollection:collection ofExpectedType:[ECKeyPair class]];
|
|
}
|
|
|
|
- (int)intForKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
NSNumber *_Nullable number = [self objectForKey:key inCollection:collection ofExpectedType:[NSNumber class]];
|
|
return [number intValue];
|
|
}
|
|
|
|
- (nullable NSDate *)dateForKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
NSNumber *_Nullable value = [self objectForKey:key inCollection:collection ofExpectedType:[NSNumber class]];
|
|
if (value) {
|
|
return [NSDate dateWithTimeIntervalSince1970:value.doubleValue];
|
|
} else {
|
|
return nil;
|
|
}
|
|
}
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation YapDatabaseReadWriteTransaction (OWS)
|
|
|
|
#pragma mark - Debug
|
|
|
|
#if DEBUG
|
|
- (void)snapshotCollection:(NSString *)collection snapshotFilePath:(NSString *)snapshotFilePath
|
|
{
|
|
NSMutableDictionary<NSString *, id> *snapshot = [NSMutableDictionary new];
|
|
[self enumerateKeysAndObjectsInCollection:collection
|
|
usingBlock:^(NSString *_Nonnull key, id _Nonnull value, BOOL *_Nonnull stop) {
|
|
snapshot[key] = value;
|
|
}];
|
|
NSData *_Nullable data = [NSKeyedArchiver archivedDataWithRootObject:snapshot];
|
|
BOOL success = [data writeToFile:snapshotFilePath atomically:YES];
|
|
}
|
|
|
|
- (void)restoreSnapshotOfCollection:(NSString *)collection snapshotFilePath:(NSString *)snapshotFilePath
|
|
{
|
|
|
|
NSData *_Nullable data = [NSData dataWithContentsOfFile:snapshotFilePath];
|
|
NSMutableDictionary<NSString *, id> *_Nullable snapshot = [NSKeyedUnarchiver unarchiveObjectWithData:data];
|
|
|
|
[self removeAllObjectsInCollection:collection];
|
|
[snapshot enumerateKeysAndObjectsUsingBlock:^(NSString *_Nonnull key, id _Nonnull value, BOOL *_Nonnull stop) {
|
|
[self setObject:value forKey:key inCollection:collection];
|
|
}];
|
|
}
|
|
#endif
|
|
|
|
- (void)setBool:(BOOL)value forKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
NSNumber *_Nullable oldValue = [self objectForKey:key inCollection:collection ofExpectedType:[NSNumber class]];
|
|
if (oldValue && [@(value) isEqual:oldValue]) {
|
|
// Skip redundant writes.
|
|
return;
|
|
}
|
|
|
|
[self setObject:@(value) forKey:key inCollection:collection];
|
|
}
|
|
|
|
- (void)setDate:(NSDate *)value forKey:(NSString *)key inCollection:(NSString *)collection
|
|
{
|
|
[self setObject:@(value.timeIntervalSince1970) forKey:key inCollection:collection];
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|