cfd2a4588d
By moving the initialization to the constructor, we can now be certain that all wasm-related objects (wasm instance cache, compilation thread runner, and wasm engine, which was already passed in the constructor) are initialized when we try to use them because we have to use the query processor to access them anyway. The change is also motivated by the fact that we're planning to take Wasm UDFs out of experimental, after which they should stop getting special treatment.
62 lines
1.7 KiB
C++
62 lines
1.7 KiB
C++
/*
|
|
* Copyright (C) 2021-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "types/types.hh"
|
|
#include <seastar/core/future.hh>
|
|
#include "db/functions/function_name.hh"
|
|
#include "rust/wasmtime_bindings.hh"
|
|
#include "lang/wasm_alien_thread_runner.hh"
|
|
#include "db/config.hh"
|
|
#include "replica/database.hh"
|
|
|
|
namespace wasm {
|
|
|
|
class instance_cache;
|
|
|
|
struct exception : public std::exception {
|
|
std::string _msg;
|
|
public:
|
|
explicit exception(std::string_view msg) : _msg(msg) {}
|
|
const char* what() const noexcept {
|
|
return _msg.c_str();
|
|
}
|
|
};
|
|
|
|
struct instance_corrupting_exception : public exception {
|
|
explicit instance_corrupting_exception(std::string_view msg) : exception(msg) {}
|
|
};
|
|
|
|
struct startup_context {
|
|
std::shared_ptr<alien_thread_runner> alien_runner;
|
|
std::shared_ptr<rust::Box<wasmtime::Engine>> engine;
|
|
size_t cache_size;
|
|
size_t instance_size;
|
|
seastar::lowres_clock::duration timer_period;
|
|
|
|
startup_context(db::config& cfg, replica::database_config& dbcfg);
|
|
};
|
|
|
|
struct context {
|
|
wasmtime::Engine& engine_ptr;
|
|
std::optional<rust::Box<wasmtime::Module>> module;
|
|
std::string function_name;
|
|
instance_cache& cache;
|
|
uint64_t yield_fuel;
|
|
uint64_t total_fuel;
|
|
|
|
context(wasmtime::Engine& engine_ptr, std::string name, instance_cache& cache, uint64_t yield_fuel, uint64_t total_fuel);
|
|
};
|
|
|
|
seastar::future<> precompile(alien_thread_runner& alien_runner, context& ctx, const std::vector<sstring>& arg_names, std::string script);
|
|
|
|
seastar::future<bytes_opt> run_script(const db::functions::function_name& name, context& ctx, const std::vector<data_type>& arg_types, const std::vector<bytes_opt>& params, data_type return_type, bool allow_null_input);
|
|
|
|
}
|