1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
|
package proto
import (
"io"
"unsafe"
)
var stringCodec = codec{
wire: varlen,
size: sizeOfString,
encode: encodeString,
decode: decodeString,
}
func sizeOfString(p unsafe.Pointer, flags flags) int {
if p != nil {
if v := *(*string)(p); v != "" || flags.has(wantzero) {
return sizeOfVarlen(len(v))
}
}
return 0
}
func encodeString(b []byte, p unsafe.Pointer, flags flags) (int, error) {
if p != nil {
if v := *(*string)(p); v != "" || flags.has(wantzero) {
n, err := encodeVarint(b, uint64(len(v)))
if err != nil {
return n, err
}
c := copy(b[n:], v)
n += c
if c < len(v) {
err = io.ErrShortBuffer
}
return n, err
}
}
return 0, nil
}
func decodeString(b []byte, p unsafe.Pointer, _ flags) (int, error) {
v, n, err := decodeVarlen(b)
*(*string)(p) = string(v)
return n, err
}
|