]>
git.ipfire.org Git - thirdparty/squid.git/blob - src/mgr/Inquirer.cc
2 * Copyright (C) 1996-2018 The Squid Software Foundation and contributors
4 * Squid software is distributed under GPLv2+ license and includes
5 * contributions from numerous individuals and organizations.
6 * Please see the COPYING and CONTRIBUTORS files for details.
9 /* DEBUG: section 16 Cache Manager API */
12 #include "base/TextException.h"
14 #include "comm/Connection.h"
15 #include "comm/Write.h"
16 #include "CommCalls.h"
17 #include "errorpage.h"
18 #include "HttpReply.h"
19 #include "HttpRequest.h"
20 #include "ipc/UdsOp.h"
21 #include "mgr/ActionWriter.h"
22 #include "mgr/Command.h"
23 #include "mgr/Inquirer.h"
24 #include "mgr/IntParam.h"
25 #include "mgr/Request.h"
26 #include "mgr/Response.h"
27 #include "SquidTime.h"
31 CBDATA_NAMESPACED_CLASS_INIT(Mgr
, Inquirer
);
33 Mgr::Inquirer::Inquirer(Action::Pointer anAction
,
34 const Request
&aCause
, const Ipc::StrandCoords
&coords
):
35 Ipc::Inquirer(aCause
.clone(), applyQueryParams(coords
, aCause
.params
.queryParams
), anAction
->atomic() ? 10 : 100),
39 Ipc::ImportFdIntoComm(conn
, SOCK_STREAM
, IPPROTO_TCP
, Ipc::fdnHttpSocket
);
41 debugs(16, 5, HERE
<< conn
<< " action: " << aggrAction
);
43 closer
= asyncCall(16, 5, "Mgr::Inquirer::noteCommClosed",
44 CommCbMemFunT
<Inquirer
, CommCloseCbParams
>(this, &Inquirer::noteCommClosed
));
45 comm_add_close_handler(conn
->fd
, closer
);
48 /// closes our copy of the client HTTP connection socket
50 Mgr::Inquirer::cleanup()
52 if (Comm::IsConnOpen(conn
)) {
59 Mgr::Inquirer::removeCloseHandler()
62 comm_remove_close_handler(conn
->fd
, closer
);
68 Mgr::Inquirer::start()
71 Ipc::Inquirer::start();
72 Must(Comm::IsConnOpen(conn
));
73 Must(aggrAction
!= NULL
);
75 std::unique_ptr
<MemBuf
> replyBuf
;
76 if (strands
.empty()) {
77 const char *url
= aggrAction
->command().params
.httpUri
.termedBuf();
78 const MasterXaction::Pointer mx
= new MasterXaction(XactionInitiator::initIpc
);
79 HttpRequest
*req
= HttpRequest::FromUrl(url
, mx
);
80 ErrorState
err(ERR_INVALID_URL
, Http::scNotFound
, req
);
81 std::unique_ptr
<HttpReply
> reply(err
.BuildHttpReply());
82 replyBuf
.reset(reply
->pack());
84 std::unique_ptr
<HttpReply
> reply(new HttpReply
);
85 reply
->setHeaders(Http::scOkay
, NULL
, "text/plain", -1, squid_curtime
, squid_curtime
);
86 reply
->header
.putStr(Http::HdrType::CONNECTION
, "close"); // until we chunk response
87 replyBuf
.reset(reply
->pack());
89 writer
= asyncCall(16, 5, "Mgr::Inquirer::noteWroteHeader",
90 CommCbMemFunT
<Inquirer
, CommIoCbParams
>(this, &Inquirer::noteWroteHeader
));
91 Comm::Write(conn
, replyBuf
.get(), writer
);
94 /// called when we wrote the response header
96 Mgr::Inquirer::noteWroteHeader(const CommIoCbParams
& params
)
100 Must(params
.flag
== Comm::OK
);
101 Must(params
.conn
.getRaw() == conn
.getRaw());
102 Must(params
.size
!= 0);
103 // start inquiries at the initial pos
107 /// called when the HTTP client or some external force closed our socket
109 Mgr::Inquirer::noteCommClosed(const CommCloseCbParams
& params
)
112 Must(!Comm::IsConnOpen(conn
) && params
.conn
.getRaw() == conn
.getRaw());
114 mustStop("commClosed");
118 Mgr::Inquirer::aggregate(Ipc::Response::Pointer aResponse
)
120 Mgr::Response
& response
= static_cast<Response
&>(*aResponse
);
121 if (response
.hasAction())
122 aggrAction
->add(response
.getAction());
127 Mgr::Inquirer::sendResponse()
129 if (!strands
.empty() && aggrAction
->aggregatable()) {
130 removeCloseHandler();
131 AsyncJob::Start(new ActionWriter(aggrAction
, conn
));
132 conn
= NULL
; // should not close because we passed it to ActionWriter
137 Mgr::Inquirer::doneAll() const
139 return !writer
&& Ipc::Inquirer::doneAll();
143 Mgr::Inquirer::applyQueryParams(const Ipc::StrandCoords
& aStrands
, const QueryParams
& aParams
)
145 Ipc::StrandCoords sc
;
147 QueryParam::Pointer processesParam
= aParams
.get("processes");
148 QueryParam::Pointer workersParam
= aParams
.get("workers");
150 if (processesParam
== NULL
|| workersParam
== NULL
) {
151 if (processesParam
!= NULL
) {
152 IntParam
* param
= dynamic_cast<IntParam
*>(processesParam
.getRaw());
153 if (param
!= NULL
&& param
->type
== QueryParam::ptInt
) {
154 const std::vector
<int>& processes
= param
->value();
155 for (Ipc::StrandCoords::const_iterator iter
= aStrands
.begin();
156 iter
!= aStrands
.end(); ++iter
) {
157 if (std::find(processes
.begin(), processes
.end(), iter
->kidId
) != processes
.end())
161 } else if (workersParam
!= NULL
) {
162 IntParam
* param
= dynamic_cast<IntParam
*>(workersParam
.getRaw());
163 if (param
!= NULL
&& param
->type
== QueryParam::ptInt
) {
164 const std::vector
<int>& workers
= param
->value();
165 for (int i
= 0; i
< (int)aStrands
.size(); ++i
) {
166 if (std::find(workers
.begin(), workers
.end(), i
+ 1) != workers
.end())
167 sc
.push_back(aStrands
[i
]);
175 debugs(16, 4, HERE
<< "strands kid IDs = ");
176 for (Ipc::StrandCoords::const_iterator iter
= sc
.begin(); iter
!= sc
.end(); ++iter
) {
177 debugs(16, 4, HERE
<< iter
->kidId
);