Method implForPrimitiveKey

bolt/functions/sparksql/Map.cpp:204–287

 private:
  template <TypeKind Kind>
  void implForPrimitiveKey(
      const SelectivityVector& rows,
      vector_size_t mapSize,
      std::vector<VectorPtr>& args,
      exec::DecodedArgs& decodedArgs,
      VectorPtr& result) const {
    using NativeType = typename TypeTraits<Kind>::NativeType;
    auto mapResult = result->as<MapVector>();
    auto rawSizes = mapResult->mutableSizes(rows.end())->asMutable<int32_t>();
    auto rawOffsets =
        mapResult->mutableOffsets(rows.end())->asMutable<int32_t>();

    auto& keysResult = mapResult->mapKeys();
    auto& valuesResult = mapResult->mapValues();
    vector_size_t offset = keysResult->size();

    std::vector<const SimpleVector<NativeType>*> argsKeySimpleVector;
    argsKeySimpleVector.reserve(mapSize);
    for (vector_size_t i = 0; i < mapSize; i++) {
      argsKeySimpleVector.emplace_back(
          decodedArgs.at(i * 2)->base()->as<SimpleVector<NativeType>>());
    }
    std::vector<SelectivityVector> argsSelectivity(
        rows.end(), SelectivityVector(mapSize, false));
    folly::F14FastMap<NativeType, vector_size_t> keyToArgsIndex;
    keyToArgsIndex.reserve(mapSize);

    rows.applyToSelected([&](vector_size_t row) {
      for (vector_size_t i = 0; i < mapSize; i++) {
        auto decodedkey = decodedArgs.at(i * 2);
        BOLT_USER_CHECK(
            !decodedkey->isNullAt(row), "Cannot use null as map key!");
        auto key = argsKeySimpleVector[i];
        auto keyRow = decodedkey->index(row);
        if constexpr (Policy == EXCEPTION) {
          BOLT_USER_CHECK(
              keyToArgsIndex.insert({key->valueAt(keyRow), i}).second,
              DuplicateKeyExceptionInfo);
        } else if constexpr (Policy == FIRST_WIN) {
          keyToArgsIndex.insert({key->valueAt(keyRow), i});
        } else if constexpr (Policy == LAST_WIN) {
          keyToArgsIndex.insert_or_assign(key->valueAt(keyRow), i);
        } else {
          BOLT_UNREACHABLE();
        }
      }

      rawSizes[row] = keyToArgsIndex.size();
      rawOffsets[row] = offset;
      offset += keyToArgsIndex.size();
      auto& argsSelec = argsSelectivity[row];
      for (const auto& [_, argsIndex] : keyToArgsIndex) {
        argsSelec.setValid(argsIndex, true);
      }
      argsSelec.updateBounds();
      keyToArgsIndex.clear();
    });