protobuf-go/proto/encode_gen.go
Damien Neil cedb595154 proto, internal/impl: avoid string->[]byte conversions
We trust the compiler to optimize away the string->[]byte conversion in
code like:

	b = wire.AppendBytes(b, []byte(s))

In testing (go 1.12.5 linux/amd64), this optimization is not happening.
Perhaps newer versions of the compiler will optimize this, but we
shouldn't rely on it; avoid unnecessary conversions.

Benchmark differences vs https://golang.org/cl/171462:

  name                                   old time/op    new time/op    delta
  Wire/Marshal/google_message1_proto2-6     310ns ± 2%     189ns ± 3%  -39.20%  (p=0.000 n=8+8)
  Wire/Marshal/google_message1_proto3-6     389ns ± 8%     261ns ± 2%  -33.03%  (p=0.000 n=8+8)
  Wire/Marshal/google_message2-6            103µs ±11%      59µs ± 4%  -42.17%  (p=0.000 n=8+8)

  name                                   old alloc/op   new alloc/op   delta
  Wire/Marshal/google_message1_proto2-6      592B ± 0%      240B ± 0%  -59.46%  (p=0.000 n=8+8)
  Wire/Marshal/google_message1_proto3-6      576B ± 0%      224B ± 0%  -61.11%  (p=0.000 n=8+8)
  Wire/Marshal/google_message2-6            196kB ± 0%      90kB ± 0%  -54.05%  (p=0.000 n=8+8)

  name                                   old allocs/op  new allocs/op  delta
  Wire/Marshal/google_message1_proto2-6      5.00 ± 0%      1.00 ± 0%  -80.00%  (p=0.000 n=8+8)
  Wire/Marshal/google_message1_proto3-6      5.00 ± 0%      1.00 ± 0%  -80.00%  (p=0.000 n=8+8)
  Wire/Marshal/google_message2-6            1.66k ± 0%     0.00k ± 0%  -99.94%  (p=0.000 n=8+8)

Change-Id: Idab7634b8c86604dffa46895ba2e61be38c9bd9c
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/183380
Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
2019-06-24 17:41:45 +00:00

97 lines
3.3 KiB
Go

// Copyright 2018 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style.
// license that can be found in the LICENSE file.
// Code generated by generate-types. DO NOT EDIT.
package proto
import (
"math"
"unicode/utf8"
"google.golang.org/protobuf/internal/encoding/wire"
"google.golang.org/protobuf/internal/errors"
"google.golang.org/protobuf/reflect/protoreflect"
)
var wireTypes = map[protoreflect.Kind]wire.Type{
protoreflect.BoolKind: wire.VarintType,
protoreflect.EnumKind: wire.VarintType,
protoreflect.Int32Kind: wire.VarintType,
protoreflect.Sint32Kind: wire.VarintType,
protoreflect.Uint32Kind: wire.VarintType,
protoreflect.Int64Kind: wire.VarintType,
protoreflect.Sint64Kind: wire.VarintType,
protoreflect.Uint64Kind: wire.VarintType,
protoreflect.Sfixed32Kind: wire.Fixed32Type,
protoreflect.Fixed32Kind: wire.Fixed32Type,
protoreflect.FloatKind: wire.Fixed32Type,
protoreflect.Sfixed64Kind: wire.Fixed64Type,
protoreflect.Fixed64Kind: wire.Fixed64Type,
protoreflect.DoubleKind: wire.Fixed64Type,
protoreflect.StringKind: wire.BytesType,
protoreflect.BytesKind: wire.BytesType,
protoreflect.MessageKind: wire.BytesType,
protoreflect.GroupKind: wire.StartGroupType,
}
func (o MarshalOptions) marshalSingular(b []byte, fd protoreflect.FieldDescriptor, v protoreflect.Value) ([]byte, error) {
switch fd.Kind() {
case protoreflect.BoolKind:
b = wire.AppendVarint(b, wire.EncodeBool(v.Bool()))
case protoreflect.EnumKind:
b = wire.AppendVarint(b, uint64(v.Enum()))
case protoreflect.Int32Kind:
b = wire.AppendVarint(b, uint64(int32(v.Int())))
case protoreflect.Sint32Kind:
b = wire.AppendVarint(b, wire.EncodeZigZag(int64(int32(v.Int()))))
case protoreflect.Uint32Kind:
b = wire.AppendVarint(b, uint64(uint32(v.Uint())))
case protoreflect.Int64Kind:
b = wire.AppendVarint(b, uint64(v.Int()))
case protoreflect.Sint64Kind:
b = wire.AppendVarint(b, wire.EncodeZigZag(v.Int()))
case protoreflect.Uint64Kind:
b = wire.AppendVarint(b, v.Uint())
case protoreflect.Sfixed32Kind:
b = wire.AppendFixed32(b, uint32(v.Int()))
case protoreflect.Fixed32Kind:
b = wire.AppendFixed32(b, uint32(v.Uint()))
case protoreflect.FloatKind:
b = wire.AppendFixed32(b, math.Float32bits(float32(v.Float())))
case protoreflect.Sfixed64Kind:
b = wire.AppendFixed64(b, uint64(v.Int()))
case protoreflect.Fixed64Kind:
b = wire.AppendFixed64(b, v.Uint())
case protoreflect.DoubleKind:
b = wire.AppendFixed64(b, math.Float64bits(v.Float()))
case protoreflect.StringKind:
if fd.Syntax() == protoreflect.Proto3 && !utf8.ValidString(v.String()) {
return b, errors.InvalidUTF8(string(fd.FullName()))
}
b = wire.AppendString(b, v.String())
case protoreflect.BytesKind:
b = wire.AppendBytes(b, v.Bytes())
case protoreflect.MessageKind:
var pos int
var err error
b, pos = appendSpeculativeLength(b)
b, err = o.marshalMessage(b, v.Message())
if err != nil {
return b, err
}
b = finishSpeculativeLength(b, pos)
case protoreflect.GroupKind:
var err error
b, err = o.marshalMessage(b, v.Message())
if err != nil {
return b, err
}
b = wire.AppendVarint(b, wire.EncodeTag(fd.Number(), wire.EndGroupType))
default:
return b, errors.New("invalid kind %v", fd.Kind())
}
return b, nil
}