fix "Type mismatch: Varbinary vs Varchar" in VectorHasher

yma11 · Jun 26, 2024 · 91be2cc · 91be2cc
1 parent ba70a4d
commit 91be2cc
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/velox/exec/VectorHasher.h b/velox/exec/VectorHasher.h
@@ -161,15 +161,21 @@ class VectorHasher {
 
   static constexpr uint64_t kNullHash = BaseVector::kNullHash;
 
+  bool isVarbinaryOrVarchar(TypeKind type) {
+    return type == TypeKind::VARCHAR || type == TypeKind::VARBINARY;
+  }
+
   // Decodes the 'vector' in preparation for calling hash() or
   // computeValueIds(). The decoded vector can be accessed via decodedVector()
   // getter.
   void decode(const BaseVector& vector, const SelectivityVector& rows) {
-    VELOX_CHECK(
-        type_->kindEquals(vector.type()),
-        "Type mismatch: {} vs. {}",
-        type_->toString(),
-        vector.type()->toString());
+    if (!(isVarbinaryOrVarchar(type_->kind()) && isVarbinaryOrVarchar(vector.type()->kind()))) {
+      VELOX_CHECK(
+          type_->kindEquals(vector.type()),
+          "Type mismatch: {} vs. {}",
+          type_->toString(),
+          vector.type()->toString());
+    }
     decoded_.decode(vector, rows);
   }