mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 22:00:47 +01:00
d6b4819e4a
By having a unique_ptr own the callback data instead of a raw pointer, the compiler helps us ensure that it's destroyed exactly once, and never used after being destroyed. (This made the callback object move-only, so I had to add support for move-only callbacks to rtc::Thread::Invoke().) BUG=webrtc:8111 Change-Id: Ia0804e4662e63e91e5cee18ecc3f38d2cfe8a26b Reviewed-on: https://webrtc-review.googlesource.com/10812 Commit-Queue: Karl Wiberg <kwiberg@webrtc.org> Reviewed-by: Taylor Brandstetter <deadbeef@webrtc.org> Cr-Commit-Position: refs/heads/master@{#20317}
75 lines
1.8 KiB
C++
75 lines
1.8 KiB
C++
/*
|
|
* Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef RTC_BASE_MESSAGEHANDLER_H_
|
|
#define RTC_BASE_MESSAGEHANDLER_H_
|
|
|
|
#include <memory>
|
|
#include <utility>
|
|
|
|
#include "rtc_base/constructormagic.h"
|
|
|
|
namespace rtc {
|
|
|
|
struct Message;
|
|
|
|
// Messages get dispatched to a MessageHandler
|
|
|
|
class MessageHandler {
|
|
public:
|
|
virtual ~MessageHandler();
|
|
virtual void OnMessage(Message* msg) = 0;
|
|
|
|
protected:
|
|
MessageHandler() {}
|
|
|
|
private:
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(MessageHandler);
|
|
};
|
|
|
|
// Helper class to facilitate executing a functor on a thread.
|
|
template <class ReturnT, class FunctorT>
|
|
class FunctorMessageHandler : public MessageHandler {
|
|
public:
|
|
explicit FunctorMessageHandler(FunctorT&& functor)
|
|
: functor_(std::forward<FunctorT>(functor)) {}
|
|
virtual void OnMessage(Message* msg) {
|
|
result_ = functor_();
|
|
}
|
|
const ReturnT& result() const { return result_; }
|
|
|
|
// Returns moved result. Should not call result() or MoveResult() again
|
|
// after this.
|
|
ReturnT MoveResult() { return std::move(result_); }
|
|
|
|
private:
|
|
FunctorT functor_;
|
|
ReturnT result_;
|
|
};
|
|
|
|
// Specialization for ReturnT of void.
|
|
template <class FunctorT>
|
|
class FunctorMessageHandler<void, FunctorT> : public MessageHandler {
|
|
public:
|
|
explicit FunctorMessageHandler(const FunctorT& functor)
|
|
: functor_(functor) {}
|
|
virtual void OnMessage(Message* msg) {
|
|
functor_();
|
|
}
|
|
void result() const {}
|
|
void MoveResult() {}
|
|
|
|
private:
|
|
FunctorT functor_;
|
|
};
|
|
|
|
} // namespace rtc
|
|
|
|
#endif // RTC_BASE_MESSAGEHANDLER_H_
|