mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-20 00:20:47 +00:00
e43a2e5086
We introduce to the Query request a new "VectorSearch" parameter, which take a mandatory "QueryVector" (a value which must be a numeric vector of the right length) and "Limit". The "Limit" of a vector search (Query with VectorSearch) determines the number of nearest neighbors to return, and does not allow pagination (ExclusiveKeyStart is not allowed). ConsistentRead=True is also not allowed on a vector search query. The "Select"/"ProjectionExpression"/"AttributesToGet" parameters are also supported, requesting which attributes to fetch. Using Select= ALL_PROJECTED_ATTRIBUTES means read only the attributes found in the vector index - currently only the key columns - so it is significantly faster than ALL_ATTRIBUTES because it doesn't require reading the items from the base table. The "FilterExpression" parameter is also supported. Like in DynamoDB's traditional Query, this does post-filtering, i.e., removing some of the results returned by the vector index that don't match the filter, and as a result fewer than Limit results may be returned. Pre-filtering (done on the vector store, and always returns Limit results) is not yet implemented.
109 lines
3.0 KiB
C++
109 lines
3.0 KiB
C++
/*
|
|
* Copyright (C) 2021-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.1
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <seastar/core/sharded.hh>
|
|
#include <seastar/core/smp.hh>
|
|
|
|
#include "transport/protocol_server.hh"
|
|
|
|
namespace service {
|
|
class storage_proxy;
|
|
class storage_service;
|
|
class migration_manager;
|
|
class memory_limiter;
|
|
}
|
|
|
|
namespace db {
|
|
class system_distributed_keyspace;
|
|
class config;
|
|
}
|
|
|
|
namespace cdc {
|
|
class generation_service;
|
|
}
|
|
|
|
namespace gms {
|
|
|
|
class gossiper;
|
|
|
|
}
|
|
|
|
namespace auth {
|
|
class service;
|
|
}
|
|
|
|
namespace qos {
|
|
class service_level_controller;
|
|
}
|
|
|
|
namespace vector_search {
|
|
class vector_store_client;
|
|
}
|
|
|
|
namespace alternator {
|
|
|
|
// This is the official DynamoDB API version.
|
|
// It represents the last major reorganization of that API, and all the features
|
|
// that were added since did NOT increment this version string.
|
|
constexpr const char* version = "2012-08-10";
|
|
|
|
using namespace seastar;
|
|
|
|
class executor;
|
|
class server;
|
|
|
|
class controller : public protocol_server {
|
|
sharded<gms::gossiper>& _gossiper;
|
|
sharded<service::storage_proxy>& _proxy;
|
|
sharded<service::storage_service>& _ss;
|
|
sharded<service::migration_manager>& _mm;
|
|
sharded<db::system_distributed_keyspace>& _sys_dist_ks;
|
|
sharded<cdc::generation_service>& _cdc_gen_svc;
|
|
sharded<service::memory_limiter>& _memory_limiter;
|
|
sharded<auth::service>& _auth_service;
|
|
sharded<qos::service_level_controller>& _sl_controller;
|
|
sharded<vector_search::vector_store_client>& _vsc;
|
|
const db::config& _config;
|
|
|
|
std::vector<socket_address> _listen_addresses;
|
|
sharded<executor> _executor;
|
|
sharded<server> _server;
|
|
std::optional<smp_service_group> _ssg;
|
|
|
|
public:
|
|
controller(
|
|
sharded<gms::gossiper>& gossiper,
|
|
sharded<service::storage_proxy>& proxy,
|
|
sharded<service::storage_service>& ss,
|
|
sharded<service::migration_manager>& mm,
|
|
sharded<db::system_distributed_keyspace>& sys_dist_ks,
|
|
sharded<cdc::generation_service>& cdc_gen_svc,
|
|
sharded<service::memory_limiter>& memory_limiter,
|
|
sharded<auth::service>& auth_service,
|
|
sharded<qos::service_level_controller>& sl_controller,
|
|
sharded<vector_search::vector_store_client>& vsc,
|
|
const db::config& config,
|
|
seastar::scheduling_group sg);
|
|
|
|
virtual sstring name() const override;
|
|
virtual sstring protocol() const override;
|
|
virtual sstring protocol_version() const override;
|
|
virtual std::vector<socket_address> listen_addresses() const override;
|
|
virtual future<> start_server() override;
|
|
virtual future<> stop_server() override;
|
|
virtual future<> request_stop_server() override;
|
|
// This virtual function is called (on each shard separately) when the
|
|
// virtual table "system.clients" is read. It is expected to generate a
|
|
// list of clients connected to this server (on this shard).
|
|
virtual future<utils::chunked_vector<foreign_ptr<std::unique_ptr<client_data>>>> get_client_data() override;
|
|
};
|
|
|
|
}
|