store_manip.hpp

Go to the documentation of this file.

#pragma once
 
#include "storage/ColumnStore.hpp"
#include <mutable/catalog/Schema.hpp>
#include <random>
#include <utility>
#include <vector>
 
 
namespace m {
 
void set_all_null(uint8_t *column_ptr, std::size_t num_attrs, std::size_t begin, std::size_t end);
 
void set_all_not_null(uint8_t *column_ptr, std::size_t num_attrs, std::size_t begin, std::size_t end);
 
void generate_primary_keys(void *column_ptr, const Type &type, std::size_t begin, std::size_t end);
 
template<typename T, typename Generator = std::mt19937_64>
std::enable_if_t<std::is_arithmetic_v<T>, void>
M_EXPORT
fill_uniform(T *column_ptr, std::vector<T> values, std::size_t begin, std::size_t end, Generator &&g = Generator()) {
    M_insist(begin < end, "must set at least one row");
 
    const auto count = end - begin;
 
    /* Write distinct values repeatedly in arbitrary order to column. */
    auto ptr = column_ptr + begin;
    for (std::size_t i = 0; i != count; ) {
        /* Shuffle the vector before writing its values to the column. */
        std::shuffle(values.begin(), values.end(), std::forward<Generator>(g));
        for (auto v : values) {
            *ptr++ = v;
            ++i;
            if (i == count)
                goto exit;
        }
    }
exit:
    M_insist(ptr - column_ptr == long(count), "incorrect number of elements written");
}
 
void generate_column_data(void *column_ptr, const Attribute &attr, std::size_t num_distinct_values, std::size_t begin,
                          std::size_t end);
 
void generate_correlated_column_data(void *left_ptr, void *right_ptr, const Attribute &attr,
                                     std::size_t num_distinct_values_left, std::size_t num_distinct_values_right,
                                     std::size_t count_left, std::size_t count_right,
                                     std::size_t num_distinct_values_matching);
 
}