mirror of
https://github.com/oxen-io/session-ios.git
synced 2023-12-13 21:30:14 +01:00
34 lines
1.1 KiB
Objective-C
34 lines
1.1 KiB
Objective-C
//
|
|
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
#import "OWSPrimaryStorage+messageIDs.h"
|
|
#import <YapDatabase/YapDatabase.h>
|
|
|
|
#define TSStorageParametersCollection @"TSStorageParametersCollection"
|
|
#define TSMessagesLatestId @"TSMessagesLatestId"
|
|
|
|
@implementation OWSPrimaryStorage (messageIDs)
|
|
|
|
+ (NSString *)getAndIncrementMessageIdWithTransaction:(YapDatabaseReadWriteTransaction *)transaction
|
|
{
|
|
NSString *messageId = [transaction objectForKey:TSMessagesLatestId inCollection:TSStorageParametersCollection];
|
|
if (!messageId) {
|
|
messageId = @"0";
|
|
}
|
|
|
|
NSNumberFormatter *numberFormatter = [[NSNumberFormatter alloc] init];
|
|
numberFormatter.numberStyle = NSNumberFormatterDecimalStyle;
|
|
NSNumber *myNumber = [numberFormatter numberFromString:messageId];
|
|
|
|
unsigned long long nextMessageId = [myNumber unsignedLongLongValue];
|
|
nextMessageId++;
|
|
|
|
NSString *nextMessageIdString = [[NSNumber numberWithUnsignedLongLong:nextMessageId] stringValue];
|
|
|
|
[transaction setObject:nextMessageIdString forKey:TSMessagesLatestId inCollection:TSStorageParametersCollection];
|
|
|
|
return messageId;
|
|
}
|
|
|
|
@end
|