Function gatherFromTimestampBufferIPC

bolt/vector/arrow/Bridge.cpp:510–580

}

void gatherFromTimestampBufferIPC(
    const BaseVector& vec,
    const Selection& rows,
    TimestampUnit unit,
    Buffer& out) {
  auto src = (*vec.values()).as<Timestamp>();
  auto dst = out.asMutable<int64_t>();
  vector_size_t j = 0;

  auto safeConv = [&](const Timestamp& ts, int64_t* outVal) -> bool {
    const int64_t sec = ts.getSeconds();
    const int64_t nanos = ts.getNanos();

    if (UNLIKELY(nanos < 0 || nanos >= 1000000000LL)) {
      return false;
    }

    __int128 v = 0;
    switch (unit) {
      case TimestampUnit::kSecond:
        v = static_cast<__int128>(sec);
        break;
      case TimestampUnit::kMilli:
        // sec * 1'000 + nanos / 1'000'000
        v = static_cast<__int128>(sec) * 1000 + nanos / 1000000;
        break;
      case TimestampUnit::kMicro:
        // sec * 1'000'000 + nanos / 1'000
        v = static_cast<__int128>(sec) * 1000000 + nanos / 1000;
        break;
      case TimestampUnit::kNano:
        // sec * 1'000'000'000 + nanos
        v = static_cast<__int128>(sec) * 1000000000 + nanos;
        break;
      default:
        BOLT_UNREACHABLE();
    }

    if (UNLIKELY(
            v > std::numeric_limits<int64_t>::max() ||
            v < std::numeric_limits<int64_t>::min())) {
      return false;
    }

    *outVal = static_cast<int64_t>(v);
    return true;
  };

  auto writeOrZero = [&](vector_size_t i) {
    int64_t v;
    if (LIKELY(safeConv(src[i], &v))) {
      dst[j] = v;
    } else {
      dst[j] = 0;
    }
    ++j;
  };