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
46
47
48
49
50
|
#include "socket.h"
#include "httprequest.h"
#include <later_api.h>
#include <uv.h>
void on_Socket_close(uv_handle_t* pHandle);
void Socket::addConnection(boost::shared_ptr<HttpRequest> request) {
connections.push_back(request);
}
void Socket::removeConnection(boost::shared_ptr<HttpRequest> request) {
connections.erase(
std::remove(connections.begin(), connections.end(), request),
connections.end());
}
Socket::~Socket() {
ASSERT_BACKGROUND_THREAD()
trace("Socket::~Socket");
}
// A deleter callback for the shared_ptr<Socket>.
void delete_ppsocket(uv_handle_t* pHandle) {
boost::shared_ptr<Socket>* ppSocket = (boost::shared_ptr<Socket>*)pHandle->data;
delete ppSocket;
}
// This tells all the HttpRequests to close and deletes the
// shared_ptr<Socket>. Each HttpRequest also has a shared_ptr to this Socket.
// Once they're all closed, the Socket will be deleted. In some cases, there
// may be an extant HttpRequest object which doesn't get deleted until an R GC
// event occurs, and so this Socket will continue to exist until then.
void Socket::close() {
ASSERT_BACKGROUND_THREAD()
trace("Socket::close");
for (std::vector<boost::shared_ptr<HttpRequest> >::reverse_iterator it = connections.rbegin();
it != connections.rend();
it++) {
// std::cerr << "Request close on " << *it << std::endl;
(*it)->close();
}
uv_handle_t* pHandle = toHandle(&handle.stream);
// Delete the shared_ptr<Socket> only after uv_close() does its work. This
// will decrease the refcount and should trigger deletion of the Socket.
uv_close(pHandle, delete_ppsocket);
}
|
