mirror of
https://github.com/scylladb/scylladb.git
synced 2026-05-28 10:41:12 +00:00
3993498fb4
connection_notifier.hh defines a number of template-specialized variables in a header. This is illegal since you're allowed to define something multiple times if it's a template, but not if it's fully specialized. gcc doesn't care but clang notices and complains. Fix by defining the variiables as inline variables, which are allowed to have definitions in multiple translation units. Closes #7519
107 lines
3.9 KiB
C++
107 lines
3.9 KiB
C++
/*
|
|
* Copyright (C) 2019 ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* This file is part of Scylla.
|
|
*
|
|
* Scylla is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Scylla is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#pragma once
|
|
|
|
#include "db/query_context.hh"
|
|
|
|
#include <seastar/net/inet_address.hh>
|
|
#include <seastar/core/sstring.hh>
|
|
#include "seastarx.hh"
|
|
|
|
#include <optional>
|
|
|
|
namespace db::system_keyspace {
|
|
extern const char *const CLIENTS;
|
|
}
|
|
|
|
enum class client_type {
|
|
cql = 0,
|
|
thrift,
|
|
alternator,
|
|
};
|
|
|
|
sstring to_string(client_type ct);
|
|
|
|
enum class changed_column {
|
|
username = 0,
|
|
connection_stage,
|
|
driver_name,
|
|
driver_version,
|
|
hostname,
|
|
protocol_version,
|
|
};
|
|
|
|
template <changed_column column> constexpr const char* column_literal = "";
|
|
template <> inline constexpr const char* column_literal<changed_column::username> = "username";
|
|
template <> inline constexpr const char* column_literal<changed_column::connection_stage> = "connection_stage";
|
|
template <> inline constexpr const char* column_literal<changed_column::driver_name> = "driver_name";
|
|
template <> inline constexpr const char* column_literal<changed_column::driver_version> = "driver_version";
|
|
template <> inline constexpr const char* column_literal<changed_column::hostname> = "hostname";
|
|
template <> inline constexpr const char* column_literal<changed_column::protocol_version> = "protocol_version";
|
|
|
|
enum class client_connection_stage {
|
|
established = 0,
|
|
authenticating,
|
|
ready,
|
|
};
|
|
|
|
template <client_connection_stage ccs> constexpr const char* connection_stage_literal = "";
|
|
template <> inline constexpr const char* connection_stage_literal<client_connection_stage::established> = "ESTABLISHED";
|
|
template <> inline constexpr const char* connection_stage_literal<client_connection_stage::authenticating> = "AUTHENTICATING";
|
|
template <> inline constexpr const char* connection_stage_literal<client_connection_stage::ready> = "READY";
|
|
|
|
// Representation of a row in `system.clients'. std::optionals are for nullable cells.
|
|
struct client_data {
|
|
net::inet_address ip;
|
|
int32_t port;
|
|
client_type ct;
|
|
client_connection_stage connection_stage = client_connection_stage::established;
|
|
int32_t shard_id; /// ID of server-side shard which is processing the connection.
|
|
|
|
// `optional' column means that it's nullable (possibly because it's
|
|
// unimplemented yet). If you want to fill ("implement") any of them,
|
|
// remember to update the query in `notify_new_client()'.
|
|
std::optional<sstring> driver_name;
|
|
std::optional<sstring> driver_version;
|
|
std::optional<sstring> hostname;
|
|
std::optional<int32_t> protocol_version;
|
|
std::optional<sstring> ssl_cipher_suite;
|
|
std::optional<bool> ssl_enabled;
|
|
std::optional<sstring> ssl_protocol;
|
|
std::optional<sstring> username;
|
|
};
|
|
|
|
future<> notify_new_client(client_data cd);
|
|
future<> notify_disconnected_client(net::inet_address addr, int port, client_type ct);
|
|
future<> clear_clientlist();
|
|
|
|
template <changed_column column_enum_val>
|
|
struct notify_client_change {
|
|
template <typename T>
|
|
future<> operator()(net::inet_address addr, int port, client_type ct, T&& value) {
|
|
const static sstring req
|
|
= format("UPDATE system.{} SET {}=? WHERE address=? AND port=? AND client_type=?;",
|
|
db::system_keyspace::CLIENTS, column_literal<column_enum_val>);
|
|
|
|
return db::execute_cql(req, std::forward<T>(value), std::move(addr), port, to_string(ct)).discard_result();
|
|
}
|
|
};
|