| 36 | type partialQueryCodec struct{} |
| 37 | |
| 38 | func (c *partialQueryCodec) Encode(msg message.Message, dest io.Writer, version primitive.ProtocolVersion) error { |
| 39 | switch query := msg.(type) { |
| 40 | case *PartialQuery: |
| 41 | if err := primitive.WriteLongString(query.Query, dest); err != nil { |
| 42 | return fmt.Errorf("cannot write QUERY query string: %w", err) |
| 43 | } |
| 44 | if err := primitive.WriteShort(uint16(query.Consistency), dest); err != nil { |
| 45 | return fmt.Errorf("cannot write QUERY consistency level: %w", err) |
| 46 | } |
| 47 | if _, err := dest.Write(query.Parameters); err != nil { |
| 48 | return fmt.Errorf("cannot write QUERY parameters: %w", err) |
| 49 | } |
| 50 | return nil |
| 51 | default: |
| 52 | return builtinQueryCodec.Encode(msg, dest, version) |
| 53 | } |
| 54 | } |
| 55 | |
| 56 | func (c *partialQueryCodec) EncodedLength(msg message.Message, version primitive.ProtocolVersion) (int, error) { |
| 57 | switch query := msg.(type) { |