firebase-ios-sdk/Firestore/Source/Util/FSTDispatchQueue.mm

/*
 * 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 <Foundation/Foundation.h>

#include <memory>
#include <utility>

#import "Firestore/Source/Util/FSTDispatchQueue.h"

#include "Firestore/core/src/firebase/firestore/util/executor_libdispatch.h"
#include "Firestore/core/src/firebase/firestore/util/hard_assert.h"
#include "absl/memory/memory.h"

using firebase::firestore::util::AsyncQueue;
using firebase::firestore::util::DelayedOperation;
using firebase::firestore::util::TimerId;
using firebase::firestore::util::internal::Executor;
using firebase::firestore::util::internal::ExecutorLibdispatch;

NS_ASSUME_NONNULL_BEGIN

#pragma mark - FSTDelayedCallback

@interface FSTDelayedCallback () {
  DelayedOperation _impl;
}

@end

@implementation FSTDelayedCallback

- (instancetype)initWithImpl:(DelayedOperation &&)impl {
  if (self = [super init]) {
    _impl = std::move(impl);
  }
  return self;
}

- (void)cancel {
  _impl.Cancel();
}

@end

#pragma mark - FSTDispatchQueue

@implementation FSTDispatchQueue {
  std::unique_ptr<AsyncQueue> _impl;
}

- (AsyncQueue *)implementation {
  return _impl.get();
}

+ (TimerId)convertTimerId:(FSTTimerID)objcTimerID {
  const TimerId converted = static_cast<TimerId>(objcTimerID);
  switch (converted) {
    case TimerId::All:
    case TimerId::ListenStreamIdle:
    case TimerId::ListenStreamConnectionBackoff:
    case TimerId::WriteStreamIdle:
    case TimerId::WriteStreamConnectionBackoff:
    case TimerId::OnlineStateTimeout:
      return converted;
    default:
      HARD_FAIL("Unknown value of enum FSTTimerID.");
  }
}

+ (instancetype)queueWith:(dispatch_queue_t)dispatchQueue {
  return [[FSTDispatchQueue alloc] initWithQueue:dispatchQueue];
}

- (instancetype)initWithQueue:(dispatch_queue_t)queue {
  if (self = [super init]) {
    _queue = queue;
    auto executor = absl::make_unique<ExecutorLibdispatch>(queue);
    _impl = absl::make_unique<AsyncQueue>(std::move(executor));
  }
  return self;
}

- (void)verifyIsCurrentQueue {
  _impl->VerifyIsCurrentQueue();
}

- (void)enterCheckedOperation:(void (^)(void))block {
  _impl->ExecuteBlocking([block] { block(); });
}

- (void)dispatchAsync:(void (^)(void))block {
  _impl->Enqueue([block] { block(); });
}

- (void)dispatchAsyncAllowingSameQueue:(void (^)(void))block {
  _impl->EnqueueRelaxed([block] { block(); });
}

- (void)dispatchSync:(void (^)(void))block {
  _impl->EnqueueBlocking([block] { block(); });
}

- (FSTDelayedCallback *)dispatchAfterDelay:(NSTimeInterval)delay
                                   timerID:(FSTTimerID)timerID
                                     block:(void (^)(void))block {
  const AsyncQueue::Milliseconds delayMs =
      std::chrono::milliseconds(static_cast<long long>(delay * 1000));
  const TimerId convertedTimerId = [FSTDispatchQueue convertTimerId:timerID];
  DelayedOperation delayed_operation =
      _impl->EnqueueAfterDelay(delayMs, convertedTimerId, [block] { block(); });
  return [[FSTDelayedCallback alloc] initWithImpl:std::move(delayed_operation)];
}

- (BOOL)containsDelayedCallbackWithTimerID:(FSTTimerID)timerID {
  return _impl->IsScheduled([FSTDispatchQueue convertTimerId:timerID]);
}

- (void)runDelayedCallbacksUntil:(FSTTimerID)lastTimerID {
  _impl->RunScheduledOperationsUntil([FSTDispatchQueue convertTimerId:lastTimerID]);
}

@end

NS_ASSUME_NONNULL_END