Line data Source code
1 : /*
2 : ___________________________________________
3 : | _ ___ _ |
4 : | | | |__ \ | | |
5 : | | |__ ) |__ _ __ _ ___ _ __ | |_ |
6 : | | '_ \ / // _` |/ _` |/ _ \ '_ \| __| | HTTP/2 AGENT FOR MOCK TESTING
7 : | | | | |/ /| (_| | (_| | __/ | | | |_ | Version 0.0.z
8 : | |_| |_|____\__,_|\__, |\___|_| |_|\__| | https://github.com/testillano/h2agent
9 : | __/ | |
10 : | |___/ |
11 : |___________________________________________|
12 :
13 : Licensed under the MIT License <http://opensource.org/licenses/MIT>.
14 : SPDX-License-Identifier: MIT
15 : Copyright (c) 2021 Eduardo Ramos
16 :
17 : Permission is hereby granted, free of charge, to any person obtaining a copy
18 : of this software and associated documentation files (the "Software"), to deal
19 : in the Software without restriction, including without limitation the rights
20 : to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
21 : copies of the Software, and to permit persons to whom the Software is
22 : furnished to do so, subject to the following conditions:
23 :
24 : The above copyright notice and this permission notice shall be included in all
25 : copies or substantial portions of the Software.
26 :
27 : THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
28 : IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
29 : FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
30 : AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
31 : LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
32 : OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
33 : SOFTWARE.
34 : */
35 :
36 : #include <string>
37 : #include <regex>
38 :
39 : #include <nlohmann/json.hpp>
40 :
41 : #include <ert/tracing/Logger.hpp>
42 :
43 : #include <AdminClientEndpointData.hpp>
44 : #include <Configuration.hpp>
45 :
46 :
47 : namespace h2agent
48 : {
49 : namespace model
50 : {
51 :
52 173 : AdminClientEndpointData::AdminClientEndpointData() {
53 173 : client_endpoint_schema_.setJson(h2agent::adminSchemas::client_endpoint); // won't fail
54 173 : }
55 :
56 3 : std::string AdminClientEndpointData::asJsonString() const {
57 :
58 3 : nlohmann::json result = nlohmann::json::array();
59 :
60 3 : read_guard_t guard(rw_mutex_);
61 6 : for (auto it = map_.begin(); it != map_.end(); it++) {
62 : //result.push_back(it->second->getJson());
63 3 : result.push_back(it->second->asJson()); // json_ is altered with connection status
64 : };
65 :
66 : // Client endpoint configuration is shown as an array regardless if there is 1 item, N items or none ([]):
67 6 : return (result.dump());
68 3 : }
69 :
70 26 : AdminClientEndpointData::LoadResult AdminClientEndpointData::loadSingle(const nlohmann::json &j, const common_resources_t &cr) {
71 :
72 26 : std::string error;
73 26 : if (!client_endpoint_schema_.validate(j, error)) {
74 4 : return BadSchema;
75 : }
76 :
77 : // Client endpoint object to fill:
78 22 : auto clientEndpoint = std::make_shared<AdminClientEndpoint>();
79 :
80 22 : if (clientEndpoint->load(j)) {
81 :
82 : // Metrics data:
83 17 : clientEndpoint->setMetricsData(cr.MetricsPtr, cr.ResponseDelaySecondsHistogramBucketBoundaries, cr.MessageSizeBytesHistogramBucketBoundaries, cr.ApplicationName);
84 :
85 : // Push the key in the map:
86 17 : admin_client_endpoint_key_t key = clientEndpoint->getKey();
87 :
88 : // First find (avoid deadlock):
89 17 : auto registeredClientEndpoint = find(key);
90 :
91 : // Then write:
92 17 : write_guard_t guard(rw_mutex_);
93 :
94 : // host, port or secure changes implies re-creation for the client connection id:
95 17 : if (registeredClientEndpoint) {
96 14 : if (registeredClientEndpoint->getHost() != clientEndpoint->getHost() ||
97 14 : registeredClientEndpoint->getPort() != clientEndpoint->getPort() ||
98 5 : registeredClientEndpoint->getSecure() != clientEndpoint->getSecure()) {
99 :
100 3 : registeredClientEndpoint->load(j);
101 3 : if (!cr.ConfigurationPtr->getLazyClientConnection()) registeredClientEndpoint->connect(true /* from scratch */);
102 3 : return Accepted;
103 : }
104 :
105 4 : LOGINFORMATIONAL(
106 : bool updated = (registeredClientEndpoint->getPermit() != clientEndpoint->getPermit());
107 : ert::tracing::Logger::informational(ert::tracing::Logger::asString("Client endpoint '%s' has been updated%s", key.c_str(), updated ? "":" but no changes detected"), ERT_FILE_LOCATION);
108 : );
109 : }
110 : else {
111 10 : add(key, clientEndpoint);
112 10 : if (!cr.ConfigurationPtr->getLazyClientConnection()) clientEndpoint->connect();
113 : }
114 :
115 14 : return Success;
116 :
117 17 : }
118 :
119 5 : return BadContent;
120 26 : }
121 :
122 20 : AdminClientEndpointData::LoadResult AdminClientEndpointData::load(const nlohmann::json &j, const common_resources_t &cr) {
123 :
124 20 : if (j.is_array()) {
125 8 : bool oneAccepted = false;
126 30 : for (auto it : j) // "it" is of type json::reference and has no key() member
127 : {
128 14 : LoadResult result = loadSingle(it, cr);
129 14 : if (result != Success && result != Accepted)
130 3 : return result;
131 11 : if (result == Accepted) oneAccepted = true;
132 14 : }
133 :
134 5 : return oneAccepted ? Accepted:Success;
135 : }
136 :
137 12 : return loadSingle(j, cr);
138 : }
139 :
140 9 : bool AdminClientEndpointData::clear()
141 : {
142 9 : write_guard_t guard(rw_mutex_);
143 :
144 9 : bool result = (size() != 0);
145 :
146 9 : map_.clear();
147 :
148 9 : return result;
149 9 : }
150 :
151 19 : std::shared_ptr<AdminClientEndpoint> AdminClientEndpointData::find(const admin_client_endpoint_key_t &id) const {
152 :
153 19 : read_guard_t guard(rw_mutex_);
154 19 : auto it = get(id);
155 19 : if (it != end())
156 9 : return it->second;
157 :
158 10 : return nullptr;
159 19 : }
160 :
161 : }
162 : }
163 :
|
