correctly allow Marshaler for UDT

moguchev · Apr 17, 2015 · e67c0fe · e67c0fe
1 parent c3ed1a1
commit e67c0fe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/marshal.go b/marshal.go
@@ -1225,6 +1225,8 @@ func marshalUDT(info TypeInfo, value interface{}) ([]byte, error) {
 	udt := info.(UDTTypeInfo)
 
 	switch v := value.(type) {
+	case Marshaler:
+		return v.MarshalCQL(info)
 	case UDTMarshaler:
 		var buf []byte
 		for _, e := range udt.Elements {