diff options
Diffstat (limited to 'libtransport/src/core/pending_interest.h')
-rw-r--r-- | libtransport/src/core/pending_interest.h | 93 |
1 files changed, 93 insertions, 0 deletions
diff --git a/libtransport/src/core/pending_interest.h b/libtransport/src/core/pending_interest.h new file mode 100644 index 000000000..87fed5073 --- /dev/null +++ b/libtransport/src/core/pending_interest.h @@ -0,0 +1,93 @@ +/* + * Copyright (c) 2017-2019 Cisco and/or its affiliates. + * 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. + */ + +#pragma once + +#include <hicn/transport/config.h> +#include <hicn/transport/core/content_object.h> +#include <hicn/transport/core/interest.h> +#include <hicn/transport/core/name.h> +#include <hicn/transport/portability/portability.h> + +#include <utils/deadline_timer.h> + +#include <asio/steady_timer.hpp> + +namespace transport { + +namespace core { + +class HicnForwarderInterface; +class VPPForwarderInterface; +class RawSocketInterface; + +template <typename ForwarderInt> +class Portal; + +typedef std::function<void(Interest::Ptr &&, ContentObject::Ptr &&)> + OnContentObjectCallback; +typedef std::function<void(Interest::Ptr &&)> OnInterestTimeoutCallback; +typedef std::function<void(const std::error_code &)> TimerCallback; + +class PendingInterest { + friend class Portal<HicnForwarderInterface>; + friend class Portal<VPPForwarderInterface>; + friend class Portal<RawSocketInterface>; + + public: + using Ptr = utils::ObjectPool<PendingInterest>::Ptr; + PendingInterest(); + + PendingInterest(Interest::Ptr &&interest, + std::unique_ptr<asio::steady_timer> &&timer); + + PendingInterest(Interest::Ptr &&interest, + OnContentObjectCallback &&on_content_object, + OnInterestTimeoutCallback &&on_interest_timeout, + std::unique_ptr<asio::steady_timer> &&timer); + + ~PendingInterest(); + + template <typename Handler> + TRANSPORT_ALWAYS_INLINE void startCountdown(Handler &&cb) { + timer_->expires_from_now( + std::chrono::milliseconds(interest_->getLifetime())); + timer_->async_wait(std::forward<Handler &&>(cb)); + } + + void cancelTimer(); + + Interest::Ptr &&getInterest(); + + void setInterest(Interest::Ptr &&interest); + + const OnContentObjectCallback &getOnDataCallback() const; + + void setOnContentObjectCallback(OnContentObjectCallback &&on_content_object); + + const OnInterestTimeoutCallback &getOnTimeoutCallback() const; + + void setOnTimeoutCallback(OnInterestTimeoutCallback &&on_interest_timeout); + + private: + Interest::Ptr interest_; + std::unique_ptr<asio::steady_timer> timer_; + OnContentObjectCallback on_content_object_callback_; + OnInterestTimeoutCallback on_interest_timeout_callback_; +}; + +} // end namespace core + +} // end namespace transport |