49f9794b1b
It's more flexible than schema's constructor. Primarily meant to be used from tests, but not necessarily only there.
53 lines
1.5 KiB
C++
53 lines
1.5 KiB
C++
/*
|
|
* Copyright 2015 Cloudius Systems
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "schema.hh"
|
|
|
|
enum class column_kind { partition_key, clustering_key, static_column, regular_column };
|
|
|
|
struct schema_builder {
|
|
sstring _ks_name;
|
|
sstring _cf_name;
|
|
std::vector<schema::column> _partition_key;
|
|
std::vector<schema::column> _clustering_key;
|
|
std::vector<schema::column> _static_columns;
|
|
std::vector<schema::column> _regular_columns;
|
|
public:
|
|
schema_builder(const sstring& ks_name, const sstring& cf_name)
|
|
: _ks_name(ks_name)
|
|
, _cf_name(cf_name)
|
|
{ }
|
|
|
|
schema_builder& with_column(bytes name, data_type type, column_kind kind = column_kind::regular_column) {
|
|
switch (kind) {
|
|
case column_kind::partition_key:
|
|
_partition_key.emplace_back(schema::column{name, type});
|
|
break;
|
|
case column_kind::clustering_key:
|
|
_clustering_key.emplace_back(schema::column{name, type});
|
|
break;
|
|
case column_kind::static_column:
|
|
_static_columns.emplace_back(schema::column{name, type});
|
|
break;
|
|
case column_kind::regular_column:
|
|
_regular_columns.emplace_back(schema::column{name, type});
|
|
break;
|
|
};
|
|
return *this;
|
|
}
|
|
|
|
schema_ptr build() {
|
|
return make_lw_shared<schema>(schema({},
|
|
_ks_name,
|
|
_cf_name,
|
|
_partition_key,
|
|
_clustering_key,
|
|
_regular_columns,
|
|
_static_columns,
|
|
utf8_type));
|
|
}
|
|
};
|