f3eade2f62
Drop the AGPL license in favor of a source-available license. See the blog post [1] for details. [1] https://www.scylladb.com/2024/12/18/why-were-moving-to-a-source-available-license/
70 lines
1.8 KiB
C++
70 lines
1.8 KiB
C++
/*
|
|
*
|
|
* Modified by ScyllaDB
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (LicenseRef-ScyllaDB-Source-Available-1.0 and Apache-2.0)
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "gms/inet_address.hh"
|
|
#include "streaming/stream_fwd.hh"
|
|
#include "streaming/stream_session.hh"
|
|
#include "streaming/session_info.hh"
|
|
#include <map>
|
|
|
|
namespace streaming {
|
|
|
|
/**
|
|
* {@link StreamCoordinator} is a helper class that abstracts away maintaining multiple
|
|
* StreamSession and ProgressInfo instances per peer.
|
|
*
|
|
* This class coordinates multiple SessionStreams per peer in both the outgoing StreamPlan context and on the
|
|
* inbound StreamResultFuture context.
|
|
*/
|
|
class stream_coordinator {
|
|
public:
|
|
using inet_address = gms::inet_address;
|
|
|
|
private:
|
|
class host_streaming_data;
|
|
std::map<locator::host_id, shared_ptr<stream_session>> _peer_sessions;
|
|
bool _is_receiving;
|
|
|
|
public:
|
|
stream_coordinator(bool is_receiving = false)
|
|
: _is_receiving(is_receiving) {
|
|
}
|
|
public:
|
|
/**
|
|
* @return true if any stream session is active
|
|
*/
|
|
bool has_active_sessions() const;
|
|
|
|
std::vector<shared_ptr<stream_session>> get_all_stream_sessions() const;
|
|
|
|
bool is_receiving() const;
|
|
|
|
void connect_all_stream_sessions();
|
|
std::set<locator::host_id> get_peers() const;
|
|
|
|
public:
|
|
shared_ptr<stream_session> get_or_create_session(stream_manager& mgr, locator::host_id peer) {
|
|
auto& session = _peer_sessions[peer];
|
|
if (!session) {
|
|
session = make_shared<stream_session>(mgr, peer);
|
|
}
|
|
return session;
|
|
}
|
|
|
|
std::vector<session_info> get_all_session_info() const;
|
|
std::vector<session_info> get_peer_session_info(locator::host_id peer) const;
|
|
|
|
void abort_all_stream_sessions();
|
|
};
|
|
|
|
} // namespace streaming
|