Files
scylladb/cql3/functions/user_function.hh
Avi Kivity fcb8d040e8 treewide: use Software Package Data Exchange (SPDX) license identifiers
Instead of lengthy blurbs, switch to single-line, machine-readable
standardized (https://spdx.dev) license identifiers. The Linux kernel
switched long ago, so there is strong precedent.

Three cases are handled: AGPL-only, Apache-only, and dual licensed.
For the latter case, I chose (AGPL-3.0-or-later and Apache-2.0),
reasoning that our changes are extensive enough to apply our license.

The changes we applied mechanically with a script, except to
licenses/README.md.

Closes #9937
2022-01-18 12:15:18 +01:00

66 lines
1.9 KiB
C++

/*
* Copyright (C) 2019-present ScyllaDB
*
* Modified by ScyllaDB
*/
/*
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
#pragma once
#include "abstract_function.hh"
#include "scalar_function.hh"
#include "lang/lua.hh"
#include "lang/wasm.hh"
namespace cql3 {
namespace functions {
class user_function final : public abstract_function, public scalar_function {
public:
struct lua_context {
sstring bitcode;
// FIXME: We should not need a copy in each function. It is here
// because user_function::execute is only passed the
// cql_serialization_format and the runtime arguments. We could
// avoid it by having a runtime->execute(user_function) instead,
// but that is a large refactoring. We could also store a
// lua_runtime in a thread_local variable, but that is one extra
// global.
lua::runtime_config cfg;
};
using context = std::variant<lua_context, wasm::context>;
private:
std::vector<sstring> _arg_names;
sstring _body;
sstring _language;
bool _called_on_null_input;
context _ctx;
public:
user_function(function_name name, std::vector<data_type> arg_types, std::vector<sstring> arg_names, sstring body,
sstring language, data_type return_type, bool called_on_null_input, context ctx);
const std::vector<sstring>& arg_names() const { return _arg_names; }
const sstring& body() const { return _body; }
const sstring& language() const { return _language; }
bool called_on_null_input() const { return _called_on_null_input; }
virtual bool is_pure() const override;
virtual bool is_native() const override;
virtual bool is_aggregate() const override;
virtual bool requires_thread() const override;
virtual bytes_opt execute(cql_serialization_format sf, const std::vector<bytes_opt>& parameters) override;
};
}
}