Support vector of tuples

Author: lukasz-antoniak

marshal.go

@@ -1827,7 +1827,7 @@ func isVectorVariableLengthType(elemType TypeInfo) bool {
 		return true
 	case TypeInet:
 		return true
-	case TypeList, TypeSet, TypeMap, TypeUDT:
+	case TypeList, TypeSet, TypeMap, TypeUDT, TypeTuple:
 		return true
 	case TypeCustom:
 		switch elemType.(type) {

metadata.go

@@ -1211,6 +1211,7 @@ const (
 	SET_TYPE        = "org.apache.cassandra.db.marshal.SetType"
 	MAP_TYPE        = "org.apache.cassandra.db.marshal.MapType"
 	UDT_TYPE        = "org.apache.cassandra.db.marshal.UserType"
+	TUPLE_TYPE      = "org.apache.cassandra.db.marshal.TupleType"
 	VECTOR_TYPE     = "org.apache.cassandra.db.marshal.VectorType"
 )
 
@@ -1386,6 +1387,19 @@ func (class *typeParserClassNode) asTypeInfo() TypeInfo {
 			Elements: fields,
 		}
 	}
+	if strings.HasPrefix(class.name, TUPLE_TYPE) {
+		fields := make([]TypeInfo, len(class.params))
+		for i := 0; i < len(class.params); i++ {
+			fields[i] = class.params[i].class.asTypeInfo()
+		}
+		return TupleTypeInfo{
+			NativeType: NativeType{
+				typ:   TypeTuple,
+				proto: class.proto,
+			},
+			Elems: fields,
+		}
+	}
 	if strings.HasPrefix(class.name, VECTOR_TYPE) {
 		dim, _ := strconv.Atoi(class.params[1].class.name)
 		return VectorType{

vector_test.go

@@ -170,6 +170,8 @@ func TestVector_Types(t *testing.T) {
 			{{2, 3}, {2, -1}, {3}, {0}, {-1.3}},
 			{{1, 1000.0}, {0}, {}, {12, 14, 15, 16}, {-1.3}},
 		}},
+		{name: "vector_tuple_text_int_float", cqlType: "tuple<text, int, float>", value: [][]interface{}{{"a", 1, float32(0.5)}, {"b", 2, float32(-1.2)}, {"c", 3, float32(0)}}},
+		{name: "vector_tuple_text_list_text", cqlType: "tuple<text, list<text>>", value: [][]interface{}{{"a", []string{"b", "c"}}, {"d", []string{"e", "f", "g"}}, {"h", []string{"i"}}}},
 		{name: "vector_set_text", cqlType: "set<text>", value: [][]string{{"a", "b"}, {"c", "d"}, {"e", "f"}}},
 		{name: "vector_list_int", cqlType: "list<int>", value: [][]int32{{1, 2, 3}, {-1, -2, -3}, {0, 0, 0}}},
 		{name: "vector_map_text_int", cqlType: "map<text, int>", value: []map[string]int{map1, map2, map3}},
@@ -329,6 +331,18 @@ func TestVector_SubTypeParsing(t *testing.T) {
 				},
 			},
 		},
+		{
+			name:   "tuple",
+			custom: "org.apache.cassandra.db.marshal.TupleType(org.apache.cassandra.db.marshal.UTF8Type,org.apache.cassandra.db.marshal.Int32Type,org.apache.cassandra.db.marshal.UTF8Type)",
+			expected: TupleTypeInfo{
+				NativeType{typ: TypeTuple},
+				[]TypeInfo{
+					NativeType{typ: TypeVarchar},
+					NativeType{typ: TypeInt},
+					NativeType{typ: TypeVarchar},
+				},
+			},
+		},
 		{
 			name:   "vector_vector_inet",
 			custom: "org.apache.cassandra.db.marshal.VectorType(org.apache.cassandra.db.marshal.VectorType(org.apache.cassandra.db.marshal.InetAddressType, 2), 3)",