a93ad3838f
We don't need a special guard value, it's only being filled for batch statements for which we can simply ignore the value. Not having special value allows us to return fast when audit is not enabled.
62 lines
1.6 KiB
C++
62 lines
1.6 KiB
C++
/*
|
|
* Modified by ScyllaDB
|
|
* Copyright (C) 2015-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: (LicenseRef-ScyllaDB-Source-Available-1.0 and Apache-2.0)
|
|
*/
|
|
#pragma once
|
|
|
|
#include "cql3/statements/raw/cf_statement.hh"
|
|
#include "cql3/statements/raw/modification_statement.hh"
|
|
#include "service/client_state.hh"
|
|
|
|
namespace cql3 {
|
|
|
|
namespace statements {
|
|
|
|
namespace raw {
|
|
|
|
class modification_statement;
|
|
|
|
class batch_statement : public raw::cf_statement {
|
|
public:
|
|
enum class type {
|
|
LOGGED, UNLOGGED, COUNTER
|
|
};
|
|
private:
|
|
type _type;
|
|
std::unique_ptr<attributes::raw> _attrs;
|
|
std::vector<std::unique_ptr<raw::modification_statement>> _parsed_statements;
|
|
public:
|
|
batch_statement(
|
|
type type_,
|
|
std::unique_ptr<attributes::raw> attrs,
|
|
std::vector<std::unique_ptr<raw::modification_statement>> parsed_statements)
|
|
: cf_statement(std::nullopt)
|
|
, _type(type_)
|
|
, _attrs(std::move(attrs))
|
|
, _parsed_statements(std::move(parsed_statements)) {
|
|
}
|
|
|
|
virtual void prepare_keyspace(const service::client_state& state) override {
|
|
for (auto&& s : _parsed_statements) {
|
|
s->prepare_keyspace(state);
|
|
}
|
|
}
|
|
|
|
virtual std::unique_ptr<prepared_statement> prepare(data_dictionary::database db, cql_stats& stats) override;
|
|
protected:
|
|
virtual audit::statement_category category() const override;
|
|
virtual audit::audit_info_ptr audit_info() const override {
|
|
constexpr bool batch = true;
|
|
return audit::audit::create_audit_info(category(), sstring(), sstring(), batch);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
}
|