3e0aacc8b5
Spans are more flexible and can be constructed from any contiguous container (such as small_vector), or a subrange of such a container. This can save allocations, so change the signature to accept a span. Spans cannot be constructed from std::initializer_list, so one such call site is changed to use construct a span directly from the single argument.
63 lines
1.7 KiB
C++
63 lines
1.7 KiB
C++
/*
|
|
* Copyright (C) 2021-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <span>
|
|
#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, std::span<const bytes_opt> params, data_type return_type, bool allow_null_input);
|
|
|
|
}
|