Method randomNextType

src/Client/BuzzHouse/Generator/SQLTypes.cpp:2146–2296

}

std::unique_ptr<SQLType>
StatementGenerator::randomNextType(RandomGenerator & rg, const uint64_t allowed_types, uint32_t & col_counter, TopTypeName * tp)
{
    const uint32_t non_nullable_type = 70;
    const uint32_t nullable_type = 35 * static_cast<uint32_t>((allowed_types & allow_nullable) != 0);
    const uint32_t array_type = 10 * static_cast<uint32_t>((allowed_types & allow_array) != 0 && this->depth < this->fc.max_depth);
    const uint32_t map_type = 10
        * static_cast<uint32_t>((allowed_types & allow_map) != 0 && this->depth < this->fc.max_depth && this->width < this->fc.max_width);
    const uint32_t tuple_type = 10 * static_cast<uint32_t>((allowed_types & allow_tuple) != 0 && this->depth < this->fc.max_depth);
    const uint32_t variant_type = 10 * static_cast<uint32_t>((allowed_types & allow_variant) != 0 && this->depth < this->fc.max_depth);
    const uint32_t nested_type = 10
        * static_cast<uint32_t>((allowed_types & allow_nested) != 0 && this->depth < this->fc.max_depth
                                && this->width < this->fc.max_width);
    std::unique_ptr<SQLType> result;

    rg.pickWeighted(
        {{non_nullable_type,
          [&]
          {
              /// Non nullable
              const bool lcard = (allowed_types & allow_low_cardinality) != 0 && rg.nextMediumNumber() < 18;
              auto res = bottomType(
                  rg, allowed_types, lcard, tp ? (lcard ? tp->mutable_non_nullable_lcard() : tp->mutable_non_nullable()) : nullptr);
              result = lcard ? std::make_unique<LowCardinality>(std::move(res)) : std::move(res);
          }},
         {nullable_type,
          [&]
          {
              /// Nullable
              const bool lcard = (allowed_types & allow_low_cardinality) != 0 && rg.nextMediumNumber() < 18;
              std::unique_ptr<SQLType> res = std::make_unique<Nullable>(bottomType(
                  rg,
                  allowed_types & ~(allow_dynamic | allow_aggregate),
                  lcard,
                  tp ? (lcard ? tp->mutable_nullable_lcard() : tp->mutable_nullable()) : nullptr));
              result = lcard ? std::make_unique<LowCardinality>(std::move(res)) : std::move(res);
          }},
         {array_type,
          [&]
          {
              /// Array
              TopTypeName * arr = tp ? tp->mutable_array() : nullptr;

              this->depth++;
              auto k = this->randomNextType(rg, this->next_type_mask & ~(allow_nested), col_counter, arr);
              this->depth--;
              result = std::make_unique<ArrayType>(std::move(k));
          }},
         {map_type,
          [&]
          {
              /// Map
              MapTypeDef * mt = tp ? tp->mutable_map() : nullptr;

              this->depth++;
              auto k = this->randomNextType(
                  rg, this->next_type_mask & ~(allow_nullable | allow_nested), col_counter, mt ? mt->mutable_key() : nullptr);
              this->width++;