#ifndef BASE_MEMORY_REF_COUNTED_DELETE_ON_MESSAGE_LOOP_H_
#define BASE_MEMORY_REF_COUNTED_DELETE_ON_MESSAGE_LOOP_H_
#include "base/location.h"
#include "base/logging.h"
#include "base/memory/ref_counted.h"
#include "base/message_loop/message_loop_proxy.h"
namespace base {
template <class T>
class RefCountedDeleteOnMessageLoop : public subtle::RefCountedThreadSafeBase {
public:
RefCountedDeleteOnMessageLoop(
const scoped_refptr<MessageLoopProxy>& proxy) : proxy_(proxy) {
DCHECK(proxy_.get());
}
void AddRef() const {
subtle::RefCountedThreadSafeBase::AddRef();
}
void Release() const {
if (subtle::RefCountedThreadSafeBase::Release())
DestructOnMessageLoop();
}
protected:
friend class DeleteHelper<RefCountedDeleteOnMessageLoop>;
~RefCountedDeleteOnMessageLoop() {}
void DestructOnMessageLoop() const {
const T* t = static_cast<const T*>(this);
if (proxy_->BelongsToCurrentThread())
delete t;
else
proxy_->DeleteSoon(FROM_HERE, t);
}
scoped_refptr<MessageLoopProxy> proxy_;
private:
DISALLOW_COPY_AND_ASSIGN(RefCountedDeleteOnMessageLoop);
};
}
#endif