pbutil.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. // Copyright 2015 CoreOS, Inc.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package pbutil
  15. import "github.com/coreos/etcd/Godeps/_workspace/src/github.com/coreos/pkg/capnslog"
  16. var (
  17. plog = capnslog.NewPackageLogger("github.com/coreos/etcd/pkg", "flags")
  18. )
  19. type Marshaler interface {
  20. Marshal() (data []byte, err error)
  21. }
  22. type Unmarshaler interface {
  23. Unmarshal(data []byte) error
  24. }
  25. func MustMarshal(m Marshaler) []byte {
  26. d, err := m.Marshal()
  27. if err != nil {
  28. plog.Panicf("marshal should never fail (%v)", err)
  29. }
  30. return d
  31. }
  32. func MustUnmarshal(um Unmarshaler, data []byte) {
  33. if err := um.Unmarshal(data); err != nil {
  34. plog.Panicf("unmarshal should never fail (%v)", err)
  35. }
  36. }
  37. func GetBool(v *bool) (vv bool, set bool) {
  38. if v == nil {
  39. return false, false
  40. }
  41. return *v, true
  42. }
  43. func Boolp(b bool) *bool { return &b }