aboutsummaryrefslogtreecommitdiffhomepage
path: root/Firebase/Messaging/FIRMessagingPubSubRegistrar.m
blob: 6268302e825998dad75ffe2054aec72e11903e18 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
 * Copyright 2017 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import "FIRMessagingPubSubRegistrar.h"

#import "FIRMessagingCheckinService.h"
#import "FIRMessagingDefines.h"
#import "FIRMessagingPubSubRegistrar.h"
#import "FIRMessagingTopicsCommon.h"
#import "NSError+FIRMessaging.h"

@interface FIRMessagingPubSubRegistrar ()

@property(nonatomic, readwrite, strong) FIRMessagingCheckinService *checkinService;

@property(nonatomic, readonly, strong) NSOperationQueue *topicOperations;
// Common errors, instantiated, to avoid generating multiple copies
@property(nonatomic, readwrite, strong) NSError *operationInProgressError;

@end

@implementation FIRMessagingPubSubRegistrar

- (instancetype)init {
  FIRMessagingInvalidateInitializer();
}

- (instancetype)initWithCheckinService:(FIRMessagingCheckinService *)checkinService {
  self = [super init];
  if (self) {
    _checkinService = checkinService;
    _topicOperations = [[NSOperationQueue alloc] init];
    // Do 10 topic operations at a time; it's enough to keep the TCP connection to the host alive,
    // saving hundreds of milliseconds on each request (compared to a serial queue).
    _topicOperations.maxConcurrentOperationCount = 10;
  }
  return self;
}

- (void)stopAllSubscriptionRequests {
  [self.topicOperations cancelAllOperations];
}

- (void)updateSubscriptionToTopic:(NSString *)topic
                        withToken:(NSString *)token
                          options:(NSDictionary *)options
                     shouldDelete:(BOOL)shouldDelete
                          handler:(FIRMessagingTopicOperationCompletion)handler {

  FIRMessagingTopicAction action = FIRMessagingTopicActionSubscribe;
  if (shouldDelete) {
    action = FIRMessagingTopicActionUnsubscribe;
  }
  FIRMessagingTopicOperation *operation =
      [[FIRMessagingTopicOperation alloc] initWithTopic:topic
                                                 action:action
                                                  token:token
                                                options:options
                                         checkinService:self.checkinService
                                             completion:handler];
  [self.topicOperations addOperation:operation];

}

@end