Method DsThetaExclude

be/src/exprs/datasketches-functions-ir.cc:176–203

}

StringVal DataSketchesFunctions::DsThetaExclude(FunctionContext* ctx,
    const StringVal& serialized_sketch1, const StringVal& serialized_sketch2) {
  // Note, A and B refer to the two input sketches in the order A-not-B.
  // if A is null return null.
  // if A is not null, B is null return copyA.
  // other return A-not-B.
  if (!serialized_sketch1.is_null && serialized_sketch1.len > 0) {
    if (serialized_sketch2.is_null || serialized_sketch2.len == 0) {
      return StringVal::CopyFrom(ctx, serialized_sketch1.ptr, serialized_sketch1.len);
    }
    // A and B are not null, call a_not_b.compute()
    datasketches::theta_a_not_b a_not_b;
    try {
      auto bytes = a_not_b.compute(
        datasketches::compact_theta_sketch::deserialize(serialized_sketch1.ptr,
            serialized_sketch1.len),
        datasketches::compact_theta_sketch::deserialize(serialized_sketch2.ptr,
            serialized_sketch2.len)
      ).serialize();
      StringVal result(ctx, bytes.size());
      memcpy(result.ptr, bytes.data(), bytes.size());
      return result;
    } catch (const std::exception& e) {
      LogSketchDeserializationError(ctx, e);
    }
  }
  return StringVal::null();
}

StringVal DataSketchesFunctions::DsThetaUnionF(FunctionContext* ctx,
    const StringVal& first_serialized_sketch, const StringVal& second_serialized_sketch) {