alarm_impl.h

Go to the documentation of this file.

/*
 *
 * Copyright 2015 gRPC authors.
 *
 * 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.
 *
 */

#ifndef GRPCPP_ALARM_IMPL_H
#define GRPCPP_ALARM_IMPL_H

#include <functional>

#include <grpc/grpc.h>
#include <grpcpp/impl/codegen/completion_queue.h>
#include <grpcpp/impl/codegen/completion_queue_tag.h>
#include <grpcpp/impl/codegen/grpc_library.h>
#include <grpcpp/impl/codegen/time.h>
#include <grpcpp/impl/grpc_library.h>

namespace grpc_impl {

class Alarm : private ::grpc::GrpcLibraryCodegen {
 public:
  Alarm();

  ~Alarm();

  template <typename T>
  Alarm(::grpc::CompletionQueue* cq, const T& deadline, void* tag) : Alarm() {
    SetInternal(cq, ::grpc::TimePoint<T>(deadline).raw_time(), tag);
  }

  template <typename T>
  void Set(::grpc::CompletionQueue* cq, const T& deadline, void* tag) {
    SetInternal(cq, ::grpc::TimePoint<T>(deadline).raw_time(), tag);
  }

  Alarm(const Alarm&) = delete;
  Alarm& operator=(const Alarm&) = delete;

  Alarm(Alarm&& rhs) : alarm_(rhs.alarm_) { rhs.alarm_ = nullptr; }
  Alarm& operator=(Alarm&& rhs) {
    alarm_ = rhs.alarm_;
    rhs.alarm_ = nullptr;
    return *this;
  }

  void Cancel();

  class experimental_type {
   public:
    explicit experimental_type(Alarm* alarm) : alarm_(alarm) {}

    template <typename T>
    void Set(const T& deadline, std::function<void(bool)> f) {
      alarm_->SetInternal(::grpc::TimePoint<T>(deadline).raw_time(),
                          std::move(f));
    }

   private:
    Alarm* alarm_;
  };

  experimental_type experimental() { return experimental_type(this); }

 private:
  void SetInternal(::grpc::CompletionQueue* cq, gpr_timespec deadline,
                   void* tag);
  void SetInternal(gpr_timespec deadline, std::function<void(bool)> f);

  ::grpc::internal::CompletionQueueTag* alarm_;
};

}  // namespace grpc_impl

#endif  // GRPCPP_ALARM_IMPL_H