pbutil.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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 MaybeUnmarshal(um Unmarshaler, data []byte) bool {
  38. if err := um.Unmarshal(data); err != nil {
  39. return false
  40. }
  41. return true
  42. }
  43. func GetBool(v *bool) (vv bool, set bool) {
  44. if v == nil {
  45. return false, false
  46. }
  47. return *v, true
  48. }
  49. func Boolp(b bool) *bool { return &b }