Index: net/quic/quic_server_packet_writer.h |
diff --git a/net/quic/quic_server_packet_writer.h b/net/quic/quic_server_packet_writer.h |
index 63829935521019e5712e27410f70bbc38485ff40..678a18a1d530a2ad87fa0ab84035dbdedc18626e 100644 |
--- a/net/quic/quic_server_packet_writer.h |
+++ b/net/quic/quic_server_packet_writer.h |
@@ -11,38 +11,42 @@ |
#include "net/quic/quic_connection.h" |
#include "net/quic/quic_packet_writer.h" |
#include "net/quic/quic_protocol.h" |
+#include "net/quic/quic_types.h" |
#include "net/udp/udp_server_socket.h" |
namespace net { |
-struct WriteResult; |
+class QuicBlockedWriterInterface; |
// Chrome specific packet writer which uses a UDPServerSocket for writing |
// data. |
class QuicServerPacketWriter : public QuicPacketWriter { |
public: |
- QuicServerPacketWriter(); |
- explicit QuicServerPacketWriter(UDPServerSocket* socket); |
+ QuicServerPacketWriter(QuicBlockedWriterInterface* blocked_writer, |
+ UDPServerSocket* socket); |
virtual ~QuicServerPacketWriter(); |
// QuicPacketWriter |
virtual WriteResult WritePacket(const char* buffer, |
size_t buf_len, |
- const net::IPAddressNumber& self_address, |
- const net::IPEndPoint& peer_address) OVERRIDE; |
+ const IPAddressNumber& self_address, |
+ const IPEndPoint& peer_address) OVERRIDE; |
virtual bool IsWriteBlockedDataBuffered() const OVERRIDE; |
virtual bool IsWriteBlocked() const OVERRIDE; |
virtual void SetWritable() OVERRIDE; |
void OnWriteComplete(int rv); |
- void SetConnection(QuicConnection* connection) { |
- connection_ = connection; |
+ void SetWriteCallback(base::Callback<void(WriteResult wr)> callback) { |
+ callback_ = callback; |
} |
private: |
base::WeakPtrFactory<QuicServerPacketWriter> weak_factory_; |
UDPServerSocket* socket_; |
- QuicConnection* connection_; |
+ |
+ // To be notified after every successful asynchronous write. |
+ QuicBlockedWriterInterface* blocked_writer_; |
+ base::Callback<void(WriteResult wr)> callback_; |
bool write_blocked_; |
DISALLOW_COPY_AND_ASSIGN(QuicServerPacketWriter); |