瀏覽代碼

correctly allow Marshaler for UDT

Chris Bannister 10 年之前
父節點
當前提交
e67c0fe9b9
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      marshal.go

+ 2 - 0
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 {