forked from apache/cassandra-cpp-driver
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.hpp
More file actions
456 lines (342 loc) · 16.7 KB
/
config.hpp
File metadata and controls
456 lines (342 loc) · 16.7 KB
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
/*
Copyright (c) DataStax, Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#ifndef DATASTAX_INTERNAL_CONFIG_HPP
#define DATASTAX_INTERNAL_CONFIG_HPP
#include "auth.hpp"
#include "cassandra.h"
#include "cloud_secure_connection_config.hpp"
#include "cluster_metadata_resolver.hpp"
#include "constants.hpp"
#include "execution_profile.hpp"
#include "protocol.hpp"
#include "reconnection_policy.hpp"
#include "speculative_execution.hpp"
#include "ssl.hpp"
#include "string.hpp"
#include "timestamp_generator.hpp"
#include <climits>
namespace datastax { namespace internal { namespace core {
void stderr_log_callback(const CassLogMessage* message, void* data);
class Config {
public:
Config()
: port_(CASS_DEFAULT_PORT)
, protocol_version_(ProtocolVersion::highest_supported())
, use_beta_protocol_version_(CASS_DEFAULT_USE_BETA_PROTOCOL_VERSION)
, thread_count_io_(CASS_DEFAULT_THREAD_COUNT_IO)
, queue_size_io_(CASS_DEFAULT_QUEUE_SIZE_IO)
, core_connections_per_host_(CASS_DEFAULT_NUM_CONNECTIONS_PER_HOST)
, reconnection_policy_(new ExponentialReconnectionPolicy())
, connect_timeout_ms_(CASS_DEFAULT_CONNECT_TIMEOUT_MS)
, resolve_timeout_ms_(CASS_DEFAULT_RESOLVE_TIMEOUT_MS)
, max_schema_wait_time_ms_(CASS_DEFAULT_MAX_SCHEMA_WAIT_TIME_MS)
, max_tracing_wait_time_ms_(CASS_DEFAULT_MAX_TRACING_DATA_WAIT_TIME_MS)
, retry_tracing_wait_time_ms_(CASS_DEFAULT_RETRY_TRACING_DATA_WAIT_TIME_MS)
, tracing_consistency_(CASS_DEFAULT_TRACING_CONSISTENCY)
, coalesce_delay_us_(CASS_DEFAULT_COALESCE_DELAY)
, new_request_ratio_(CASS_DEFAULT_NEW_REQUEST_RATIO)
, log_level_(CASS_DEFAULT_LOG_LEVEL)
, log_callback_(stderr_log_callback)
, log_data_(NULL)
, auth_provider_(new AuthProvider())
, tcp_nodelay_enable_(CASS_DEFAULT_TCP_NO_DELAY_ENABLED)
, tcp_keepalive_enable_(CASS_DEFAULT_TCP_KEEPALIVE_ENABLED)
, tcp_keepalive_delay_secs_(CASS_DEFAULT_TCP_KEEPALIVE_DELAY_SECS)
, connection_idle_timeout_secs_(CASS_DEFAULT_IDLE_TIMEOUT_SECS)
, connection_heartbeat_interval_secs_(CASS_DEFAULT_HEARTBEAT_INTERVAL_SECS)
, timestamp_gen_(new MonotonicTimestampGenerator())
, use_schema_(CASS_DEFAULT_USE_SCHEMA)
, use_hostname_resolution_(CASS_DEFAULT_HOSTNAME_RESOLUTION_ENABLED)
, use_randomized_contact_points_(CASS_DEFAULT_USE_RANDOMIZED_CONTACT_POINTS)
, max_reusable_write_objects_(CASS_DEFAULT_MAX_REUSABLE_WRITE_OBJECTS)
, prepare_on_all_hosts_(CASS_DEFAULT_PREPARE_ON_ALL_HOSTS)
, prepare_on_up_or_add_host_(CASS_DEFAULT_PREPARE_ON_UP_OR_ADD_HOST)
, no_compact_(CASS_DEFAULT_NO_COMPACT)
, is_client_id_set_(false)
, host_listener_(new DefaultHostListener())
, monitor_reporting_interval_secs_(CASS_DEFAULT_CLIENT_MONITOR_EVENTS_INTERVAL_SECS)
, cluster_metadata_resolver_factory_(new DefaultClusterMetadataResolverFactory())
, histogram_refresh_interval_(CASS_DEFAULT_HISTOGRAM_REFRESH_INTERVAL_NO_REFRESH) {
profiles_.set_empty_key(String());
// Assign the defaults to the cluster profile
default_profile_.set_serial_consistency(CASS_DEFAULT_SERIAL_CONSISTENCY);
default_profile_.set_request_timeout(CASS_DEFAULT_REQUEST_TIMEOUT_MS);
default_profile_.set_load_balancing_policy(new DCAwarePolicy());
default_profile_.set_retry_policy(new DefaultRetryPolicy());
default_profile_.set_speculative_execution_policy(new NoSpeculativeExecutionPolicy());
}
Config new_instance() const {
Config config = *this;
config.default_profile_.build_load_balancing_policy();
config.init_profiles(); // Initializes the profiles from default (if needed)
config.set_speculative_execution_policy(
default_profile_.speculative_execution_policy()->new_instance());
return config;
}
CassConsistency consistency() { return default_profile_.consistency(); }
void set_consistency(CassConsistency consistency) {
default_profile_.set_consistency(consistency);
}
CassConsistency serial_consistency() { return default_profile_.serial_consistency(); }
void set_serial_consistency(CassConsistency serial_consistency) {
default_profile_.set_serial_consistency(serial_consistency);
}
unsigned thread_count_io() const { return thread_count_io_; }
void set_thread_count_io(unsigned num_threads) { thread_count_io_ = num_threads; }
unsigned queue_size_io() const { return queue_size_io_; }
void set_queue_size_io(unsigned queue_size) { queue_size_io_ = queue_size; }
unsigned core_connections_per_host() const { return core_connections_per_host_; }
void set_core_connections_per_host(unsigned num_connections) {
core_connections_per_host_ = num_connections;
}
ReconnectionPolicy::Ptr reconnection_policy() const { return reconnection_policy_; }
void set_constant_reconnect(uint64_t wait_time_ms) {
reconnection_policy_.reset(new ConstantReconnectionPolicy(wait_time_ms));
}
void set_exponential_reconnect(uint64_t base_delay_ms, uint64_t max_delay_ms) {
reconnection_policy_.reset(new ExponentialReconnectionPolicy(base_delay_ms, max_delay_ms));
}
unsigned connect_timeout_ms() const { return connect_timeout_ms_; }
void set_connect_timeout(unsigned timeout_ms) { connect_timeout_ms_ = timeout_ms; }
unsigned max_schema_wait_time_ms() const { return max_schema_wait_time_ms_; }
void set_max_schema_wait_time_ms(unsigned time_ms) { max_schema_wait_time_ms_ = time_ms; }
unsigned max_tracing_wait_time_ms() const { return max_tracing_wait_time_ms_; }
void set_max_tracing_wait_time_ms(unsigned time_ms) { max_tracing_wait_time_ms_ = time_ms; }
unsigned retry_tracing_wait_time_ms() const { return retry_tracing_wait_time_ms_; }
void set_retry_tracing_wait_time_ms(unsigned time_ms) { retry_tracing_wait_time_ms_ = time_ms; }
CassConsistency tracing_consistency() const { return tracing_consistency_; }
void set_tracing_consistency(CassConsistency consistency) { tracing_consistency_ = consistency; }
uint64_t coalesce_delay_us() const { return coalesce_delay_us_; }
void set_coalesce_delay_us(uint64_t delay_us) { coalesce_delay_us_ = delay_us; }
int new_request_ratio() const { return new_request_ratio_; }
void set_new_request_ratio(int ratio) { new_request_ratio_ = ratio; }
unsigned request_timeout() { return default_profile_.request_timeout_ms(); }
void set_request_timeout(unsigned timeout_ms) {
default_profile_.set_request_timeout(timeout_ms);
}
unsigned resolve_timeout_ms() const { return resolve_timeout_ms_; }
void set_resolve_timeout(unsigned timeout_ms) { resolve_timeout_ms_ = timeout_ms; }
const AddressVec& contact_points() const { return contact_points_; }
AddressVec& contact_points() { return contact_points_; }
int port() const { return port_; }
void set_port(int port) { port_ = port; }
ProtocolVersion protocol_version() const { return protocol_version_; }
void set_protocol_version(ProtocolVersion version) { protocol_version_ = version; }
bool use_beta_protocol_version() const { return use_beta_protocol_version_; }
void set_use_beta_protocol_version(bool enable) { use_beta_protocol_version_ = enable; }
CassLogLevel log_level() const { return log_level_; }
void set_log_level(CassLogLevel log_level) { log_level_ = log_level; }
void* log_data() const { return log_data_; }
CassLogCallback log_callback() const { return log_callback_; }
void set_log_callback(CassLogCallback callback, void* data) {
log_callback_ = callback;
log_data_ = data;
}
const AuthProvider::Ptr& auth_provider() const { return auth_provider_; }
void set_auth_provider(const AuthProvider::Ptr& auth_provider) {
auth_provider_ = (!auth_provider ? AuthProvider::Ptr(new AuthProvider()) : auth_provider);
}
void set_credentials(const String& username, const String& password) {
auth_provider_.reset(new PlainTextAuthProvider(username, password));
}
const LoadBalancingPolicy::Ptr& load_balancing_policy() const {
return default_profile().load_balancing_policy();
}
LoadBalancingPolicy::Vec load_balancing_policies() const {
LoadBalancingPolicy::Vec policies;
for (ExecutionProfile::Map::const_iterator it = profiles_.begin(), end = profiles_.end();
it != end; ++it) {
if (it->second.load_balancing_policy()) {
policies.push_back(it->second.load_balancing_policy());
}
}
return policies;
}
void set_load_balancing_policy(LoadBalancingPolicy* lbp) {
default_profile_.set_load_balancing_policy(lbp);
}
void set_speculative_execution_policy(SpeculativeExecutionPolicy* sep) {
default_profile_.set_speculative_execution_policy(sep);
}
const SslContext::Ptr& ssl_context() const { return ssl_context_; }
void set_ssl_context(SslContext* ssl_context) { ssl_context_.reset(ssl_context); }
void set_ssl_context(const SslContext::Ptr& ssl_context) { ssl_context_ = ssl_context; }
bool token_aware_routing() const { return default_profile().token_aware_routing(); }
void set_token_aware_routing(bool is_token_aware) {
default_profile_.set_token_aware_routing(is_token_aware);
}
void set_token_aware_routing_shuffle_replicas(bool shuffle_replicas) {
default_profile_.set_token_aware_routing_shuffle_replicas(shuffle_replicas);
}
void set_latency_aware_routing(bool is_latency_aware) {
default_profile_.set_latency_aware_routing(is_latency_aware);
}
void set_latency_aware_routing_settings(const LatencyAwarePolicy::Settings& settings) {
default_profile_.set_latency_aware_routing_settings(settings);
}
bool tcp_nodelay_enable() const { return tcp_nodelay_enable_; }
void set_tcp_nodelay(bool enable) { tcp_nodelay_enable_ = enable; }
bool tcp_keepalive_enable() const { return tcp_keepalive_enable_; }
unsigned tcp_keepalive_delay_secs() const { return tcp_keepalive_delay_secs_; }
void set_tcp_keepalive(bool enable, unsigned delay_secs) {
tcp_keepalive_enable_ = enable;
tcp_keepalive_delay_secs_ = delay_secs;
}
unsigned connection_idle_timeout_secs() const { return connection_idle_timeout_secs_; }
void set_connection_idle_timeout_secs(unsigned timeout_secs) {
connection_idle_timeout_secs_ = timeout_secs;
}
unsigned connection_heartbeat_interval_secs() const {
return connection_heartbeat_interval_secs_;
}
void set_connection_heartbeat_interval_secs(unsigned interval_secs) {
connection_heartbeat_interval_secs_ = interval_secs;
}
TimestampGenerator* timestamp_gen() const { return timestamp_gen_.get(); }
void set_timestamp_gen(TimestampGenerator* timestamp_gen) {
if (timestamp_gen == NULL) return;
timestamp_gen_.reset(timestamp_gen);
}
void set_retry_policy(RetryPolicy* retry_policy) {
default_profile_.set_retry_policy(retry_policy);
}
bool use_schema() const { return use_schema_; }
void set_use_schema(bool enable) { use_schema_ = enable; }
bool use_hostname_resolution() const { return use_hostname_resolution_; }
void set_use_hostname_resolution(bool enable) { use_hostname_resolution_ = enable; }
bool use_randomized_contact_points() const { return use_randomized_contact_points_; }
void set_use_randomized_contact_points(bool enable) { use_randomized_contact_points_ = enable; }
unsigned max_reusable_write_objects() const { return max_reusable_write_objects_; }
void set_max_reusable_write_objects(unsigned max_reusable_write_objects) {
max_reusable_write_objects_ = max_reusable_write_objects;
}
const ExecutionProfile& default_profile() const { return default_profile_; }
ExecutionProfile& default_profile() { return default_profile_; }
const ExecutionProfile::Map& profiles() const { return profiles_; }
void set_execution_profile(const String& name, const ExecutionProfile* profile) {
ExecutionProfile copy = *profile;
copy.build_load_balancing_policy();
profiles_[name] = copy;
}
bool prepare_on_all_hosts() const { return prepare_on_all_hosts_; }
void set_prepare_on_all_hosts(bool enabled) { prepare_on_all_hosts_ = enabled; }
bool prepare_on_up_or_add_host() const { return prepare_on_up_or_add_host_; }
void set_prepare_on_up_or_add_host(bool enabled) { prepare_on_up_or_add_host_ = enabled; }
const Address& local_address() const { return local_address_; }
void set_local_address(const Address& address) { local_address_ = address; }
bool no_compact() const { return no_compact_; }
void set_no_compact(bool enabled) { no_compact_ = enabled; }
const String& application_name() const { return application_name_; }
void set_application_name(const String& application_name) {
application_name_ = application_name;
}
const String& application_version() const { return application_version_; }
void set_application_version(const String& application_version) {
application_version_ = application_version;
}
CassUuid client_id() const { return client_id_; }
bool is_client_id_set() const { return is_client_id_set_; }
void set_client_id(CassUuid client_id) {
client_id_ = client_id;
is_client_id_set_ = true;
}
const DefaultHostListener::Ptr& host_listener() const { return host_listener_; }
void set_host_listener(const DefaultHostListener::Ptr& listener) {
if (listener) {
host_listener_ = listener;
} else {
host_listener_.reset(new DefaultHostListener());
}
}
unsigned monitor_reporting_interval_secs() const { return monitor_reporting_interval_secs_; }
void set_monitor_reporting_interval_secs(unsigned interval_secs) {
monitor_reporting_interval_secs_ = interval_secs;
};
const CloudSecureConnectionConfig& cloud_secure_connection_config() const {
return cloud_secure_connection_config_;
}
bool set_cloud_secure_connection_bundle(const String& path) {
return cloud_secure_connection_config_.load(path, this);
}
const ClusterMetadataResolverFactory::Ptr& cluster_metadata_resolver_factory() const {
return cluster_metadata_resolver_factory_;
}
void set_cluster_metadata_resolver_factory(const ClusterMetadataResolverFactory::Ptr& factory) {
cluster_metadata_resolver_factory_ = factory;
}
void set_default_consistency(CassConsistency consistency) {
if (default_profile_.consistency() == CASS_CONSISTENCY_UNKNOWN) {
default_profile_.set_consistency(consistency);
}
for (ExecutionProfile::Map::iterator it = profiles_.begin(); it != profiles_.end(); ++it) {
if (it->second.consistency() == CASS_CONSISTENCY_UNKNOWN) {
it->second.set_consistency(consistency);
}
}
}
unsigned cluster_histogram_refresh_interval() const { return histogram_refresh_interval_; }
void set_cluster_histogram_refresh_interval(unsigned refresh_interval) {
histogram_refresh_interval_ = refresh_interval;
}
private:
void init_profiles();
private:
int port_;
ProtocolVersion protocol_version_;
bool use_beta_protocol_version_;
AddressVec contact_points_;
unsigned thread_count_io_;
unsigned queue_size_io_;
unsigned core_connections_per_host_;
SharedRefPtr<ReconnectionPolicy> reconnection_policy_;
unsigned connect_timeout_ms_;
unsigned resolve_timeout_ms_;
unsigned max_schema_wait_time_ms_;
unsigned max_tracing_wait_time_ms_;
unsigned retry_tracing_wait_time_ms_;
CassConsistency tracing_consistency_;
uint64_t coalesce_delay_us_;
int new_request_ratio_;
CassLogLevel log_level_;
CassLogCallback log_callback_;
void* log_data_;
AuthProvider::Ptr auth_provider_;
SslContext::Ptr ssl_context_;
bool tcp_nodelay_enable_;
bool tcp_keepalive_enable_;
unsigned tcp_keepalive_delay_secs_;
unsigned connection_idle_timeout_secs_;
unsigned connection_heartbeat_interval_secs_;
SharedRefPtr<TimestampGenerator> timestamp_gen_;
bool use_schema_;
bool use_hostname_resolution_;
bool use_randomized_contact_points_;
unsigned max_reusable_write_objects_;
ExecutionProfile default_profile_;
ExecutionProfile::Map profiles_;
bool prepare_on_all_hosts_;
bool prepare_on_up_or_add_host_;
Address local_address_;
bool no_compact_;
String application_name_;
String application_version_;
bool is_client_id_set_;
CassUuid client_id_;
DefaultHostListener::Ptr host_listener_;
unsigned monitor_reporting_interval_secs_;
CloudSecureConnectionConfig cloud_secure_connection_config_;
ClusterMetadataResolverFactory::Ptr cluster_metadata_resolver_factory_;
unsigned histogram_refresh_interval_;
};
}}} // namespace datastax::internal::core
#endif