Merge pull request #1936 from kuzudb/union-fix

Fix union tag bug
kuzudb · Aug 16, 2023 · d71c54f · d71c54f
2 parents c90fb84 + c31422c
commit d71c54f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/common/vector/value_vector.cpp b/src/common/vector/value_vector.cpp
@@ -17,7 +17,7 @@ ValueVector::ValueVector(LogicalType dataType, storage::MemoryManager* memoryMan
 
 void ValueVector::setState(std::shared_ptr<DataChunkState> state) {
     this->state = state;
-    if (dataType.getLogicalTypeID() == LogicalTypeID::STRUCT) {
+    if (dataType.getPhysicalType() == PhysicalTypeID::STRUCT) {
         auto childrenVectors = StructVector::getFieldVectors(this);
         for (auto& childVector : childrenVectors) {
             childVector->setState(state);

diff --git a/test/test_files/tinysnb/projection/single_label.test b/test/test_files/tinysnb/projection/single_label.test
@@ -663,6 +663,13 @@ NO hex code
 ---- 1
 hEl '"\
 
+-LOG ReturnUnionTag
+-STATEMENT MATCH (m:movies) RETURN union_tag(m.grade)
+---- 3
+credit
+grade1
+grade1
+
 -CASE LargeListOfStruct
 -DEFINE STRUCT_VAL REPEAT 2400 "{a: ${count}},"
 -STATEMENT RETURN list_append([${STRUCT_VAL}{a: 99}], {a: 200})