Function toCacheIndex

bolt/vector/VectorPool.cpp:39–63

Checks if specified type is supported and returns an index of the corresponding TypePool cache in VectorPool::vectors_ array. Return -1 if type is not supported, i.e. not a built-in singleton type.

/// corresponding TypePool cache in VectorPool::vectors_ array. Return -1 if
/// type is not supported, i.e. not a built-in singleton type.
FOLLY_ALWAYS_INLINE int32_t toCacheIndex(const TypePtr& type) {
  static constexpr int32_t kNumCachedVectorTypes =
      static_cast<int32_t>(TypeKind::HUGEINT) + 1;

  static std::array<const Type*, kNumCachedVectorTypes> kSupportedTypes = {
      BOOLEAN().get(),
      TINYINT().get(),
      SMALLINT().get(),
      INTEGER().get(),
      BIGINT().get(),
      REAL().get(),
      DOUBLE().get(),
      VARCHAR().get(),
      VARBINARY().get(),
      TIMESTAMP().get(),
      DATE().get(),
  };

  auto index = static_cast<int32_t>(type->kind());
  if (index < kNumCachedVectorTypes && kSupportedTypes[index] == type.get()) {
    return index;
  }

  return -1;
}
} // namespace

VectorPtr VectorPool::get(const TypePtr& type, vector_size_t size) {