395dec35c1
Templates are slower to compile and more difficult to read, in this case generalization is not needed and can be replaced by std::function.
119 lines
5.2 KiB
C++
119 lines
5.2 KiB
C++
/*
|
|
* Modified by ScyllaDB
|
|
*
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (AGPL-3.0-or-later and Apache-2.0)
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "function.hh"
|
|
#include "cql3/assignment_testable.hh"
|
|
#include "cql3/cql3_type.hh"
|
|
#include "cql3/functions/function_name.hh"
|
|
#include "schema/schema.hh"
|
|
#include <unordered_map>
|
|
|
|
namespace cql3 {
|
|
|
|
namespace functions {
|
|
//forward declarations
|
|
class user_function;
|
|
class user_aggregate;
|
|
class change_batch;
|
|
|
|
using declared_t = std::unordered_multimap<function_name, shared_ptr<function>>;
|
|
void add_agg_functions(declared_t& funcs);
|
|
|
|
class functions {
|
|
friend class change_batch;
|
|
std::unordered_multimap<function_name, shared_ptr<function>> init() noexcept;
|
|
protected:
|
|
using declared_t = cql3::functions::declared_t;
|
|
declared_t _declared;
|
|
|
|
struct skip_init {};
|
|
functions(skip_init) {};
|
|
public:
|
|
lw_shared_ptr<column_specification> make_arg_spec(const sstring& receiver_ks, std::optional<const std::string_view> receiver_cf,
|
|
const function& fun, size_t i) const;
|
|
|
|
functions() : _declared(init()) {}
|
|
|
|
shared_ptr<function> get(data_dictionary::database db,
|
|
const sstring& keyspace,
|
|
const function_name& name,
|
|
const std::vector<shared_ptr<assignment_testable>>& provided_args,
|
|
const sstring& receiver_ks,
|
|
std::optional<const std::string_view> receiver_cf,
|
|
const column_specification* receiver = nullptr) const;
|
|
template <typename AssignmentTestablePtrRange>
|
|
shared_ptr<function> get(data_dictionary::database db,
|
|
const sstring& keyspace,
|
|
const function_name& name,
|
|
AssignmentTestablePtrRange&& provided_args,
|
|
const sstring& receiver_ks,
|
|
std::optional<const std::string_view> receiver_cf,
|
|
const column_specification* receiver = nullptr) const {
|
|
const std::vector<shared_ptr<assignment_testable>> args(std::begin(provided_args), std::end(provided_args));
|
|
return get(db, keyspace, name, args, receiver_ks, receiver_cf, receiver);
|
|
}
|
|
std::vector<shared_ptr<user_function>> get_user_functions(const sstring& keyspace) const;
|
|
std::vector<shared_ptr<user_aggregate>> get_user_aggregates(const sstring& keyspace) const;
|
|
boost::iterator_range<declared_t::const_iterator> find(const function_name& name) const;
|
|
declared_t::const_iterator find_iter(const function_name& name, const std::vector<data_type>& arg_types) const;
|
|
shared_ptr<function> find(const function_name& name, const std::vector<data_type>& arg_types) const;
|
|
shared_ptr<function> mock_get(const function_name& name, const std::vector<data_type>& arg_types) const;
|
|
void add_function(shared_ptr<function>);
|
|
void replace_function(shared_ptr<function>);
|
|
void remove_function(const function_name& name, const std::vector<data_type>& arg_types);
|
|
std::optional<function_name> used_by_user_aggregate(shared_ptr<user_function>) const;
|
|
std::optional<function_name> used_by_user_function(const ut_name& user_type) const;
|
|
private:
|
|
void with_udf_iter(const function_name& name, const std::vector<data_type>& arg_types, std::function<void(declared_t::iterator)> f);
|
|
|
|
template <typename F>
|
|
std::vector<shared_ptr<F>> get_filtered_transformed(const sstring& keyspace) const;
|
|
|
|
// This method and matchArguments are somewhat duplicate, but this method allows us to provide more precise errors in the common
|
|
// case where there is no override for a given function. This is thus probably worth the minor code duplication.
|
|
void validate_types(data_dictionary::database db,
|
|
const sstring& keyspace,
|
|
const schema* schema_opt,
|
|
shared_ptr<function> fun,
|
|
const std::vector<shared_ptr<assignment_testable>>& provided_args,
|
|
const sstring& receiver_ks,
|
|
std::optional<const std::string_view> receiver_cf) const;
|
|
assignment_testable::test_result match_arguments(data_dictionary::database db, const sstring& keyspace,
|
|
const schema* schema_opt,
|
|
shared_ptr<function> fun,
|
|
const std::vector<shared_ptr<assignment_testable>>& provided_args,
|
|
const sstring& receiver_ks,
|
|
std::optional<const std::string_view> receiver_cf) const;
|
|
|
|
bool type_equals(const std::vector<data_type>& t1, const std::vector<data_type>& t2) const;
|
|
};
|
|
|
|
// Getter for static functions object.
|
|
const functions& instance();
|
|
|
|
class change_batch : public functions {
|
|
public:
|
|
// Skip init as we copy data from static instance.
|
|
change_batch() : functions(skip_init{}) {
|
|
_declared = instance()._declared;
|
|
}
|
|
|
|
// Commit allows to atomically apply changes to main functions instance.
|
|
void commit();
|
|
|
|
// Used only by unittest.
|
|
void clear_functions() noexcept;
|
|
};
|
|
|
|
}
|
|
}
|