Files
scylladb/types/user.hh
Kamil Braun d8f8908d34 types: introduce user_type_impl::idx_of_field method.
Each field of a user type has its index inside the type.
This method allows to find it easily, which is needed in a bunch of
places.
2019-10-25 10:42:58 +02:00

68 lines
2.8 KiB
C++

/*
* Copyright (C) 2014 ScyllaDB
*/
/*
* This file is part of Scylla.
*
* Scylla is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Scylla is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include "types.hh"
#include "types/tuple.hh"
class user_type_impl : public tuple_type_impl {
using intern = type_interning_helper<user_type_impl, sstring, bytes, std::vector<bytes>, std::vector<data_type>>;
public:
const sstring _keyspace;
const bytes _name;
private:
std::vector<bytes> _field_names;
std::vector<sstring> _string_field_names;
public:
using native_type = std::vector<data_value>;
user_type_impl(sstring keyspace, bytes name, std::vector<bytes> field_names, std::vector<data_type> field_types)
: tuple_type_impl(kind::user, make_name(keyspace, name, field_names, field_types, false /* frozen */), field_types)
, _keyspace(keyspace)
, _name(name)
, _field_names(field_names) {
for (const auto& field_name : _field_names) {
_string_field_names.emplace_back(utf8_type->to_string(field_name));
}
}
static shared_ptr<const user_type_impl> get_instance(sstring keyspace, bytes name, std::vector<bytes> field_names, std::vector<data_type> field_types) {
return intern::get_instance(std::move(keyspace), std::move(name), std::move(field_names), std::move(field_types));
}
data_type field_type(size_t i) const { return type(i); }
const std::vector<data_type>& field_types() const { return _types; }
bytes_view field_name(size_t i) const { return _field_names[i]; }
sstring field_name_as_string(size_t i) const { return _string_field_names[i]; }
const std::vector<bytes>& field_names() const { return _field_names; }
const std::vector<sstring>& string_field_names() const { return _string_field_names; }
std::optional<size_t> idx_of_field(const bytes& name) const;
sstring get_name_as_string() const;
private:
static sstring make_name(sstring keyspace,
bytes name,
std::vector<bytes> field_names,
std::vector<data_type> field_types,
bool is_multi_cell);
};
data_value make_user_value(data_type tuple_type, user_type_impl::native_type value);