Fix OWSPrimaryStorage+PreKeyStore methods.

7 years ago · 38b698a531
parent f38ddd052e
commit 38b698a531
2 changed files with 7 additions and 7 deletions
--- a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h
+++ b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.h
@ -10,7 +10,7 @@ NS_ASSUME_NONNULL_BEGIN
@interface OWSPrimaryStorage (PreKeyStore) <PreKeyStore>

 - (NSArray<PreKeyRecord *> *)generatePreKeyRecords;
- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(int)amount;
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(NSUInteger)batchSize;
 - (void)storePreKeyRecords:(NSArray<PreKeyRecord *> *)preKeyRecords NS_SWIFT_NAME(storePreKeyRecords(_:));

@end
--- a/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m
+++ b/SignalServiceKit/src/Storage/AxolotlStore/OWSPrimaryStorage+PreKeyStore.m
@ -18,7 +18,12 @@ NS_ASSUME_NONNULL_BEGIN

@implementation OWSPrimaryStorage (PreKeyStore)

- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(int)batchSize
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords
+{
+    return [self generatePreKeyRecords:BATCH_SIZE];
+}
+
+- (NSArray<PreKeyRecord *> *)generatePreKeyRecords:(NSUInteger)batchSize
 {
    NSMutableArray *preKeyRecords = [NSMutableArray array];

@ -42,11 +47,6 @@ NS_ASSUME_NONNULL_BEGIN
    return preKeyRecords;
 }

- (NSArray<PreKeyRecord *> *)generatePreKeyRecords
-{
-    return [self generatePreKeyRecords:BATCH_SIZE];
-}
-
 - (void)storePreKeyRecords:(NSArray<PreKeyRecord *> *)preKeyRecords
 {
    for (PreKeyRecord *record in preKeyRecords) {