blob: 24a8b52c41844e4de3c02c4f17a8ea0aff84c084 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// This is a wrapper class to help ref-counting a protobuf message.
// This file should only be inclued on host_message_dispatcher.cc and
// client_message_dispatche.cc.
// A single protobuf can contain multiple messages that will be handled by
// different message handlers. We use this wrapper to ensure that the
// protobuf is only deleted after all the handlers have finished executing.
#ifndef REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_
#define REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_
#include "base/ref_counted.h"
#include "base/task.h"
namespace remoting {
namespace protocol {
template <typename T>
class RefCountedMessage : public base::RefCounted<RefCountedMessage<T> > {
public:
RefCountedMessage(T* message) : message_(message) { }
T* message() { return message_.get(); }
private:
scoped_ptr<T> message_;
};
// Dummy methods to destroy messages.
template <class T>
static void DeleteMessage(scoped_refptr<T> message) { }
template <class T>
static Task* NewDeleteTask(scoped_refptr<T> message) {
return NewRunnableFunction(&DeleteMessage<T>, message);
}
} // namespace protocol
} // namespace remoting
#endif // REMOTING_PROTOCOL_REF_COUNTED_MESSAGE_H_
|