pointer_reflect.go 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. // Copyright 2018 The Go Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. // +build purego appengine
  5. package impl
  6. import (
  7. "fmt"
  8. "reflect"
  9. )
  10. const UnsafeEnabled = false
  11. // offset represents the offset to a struct field, accessible from a pointer.
  12. // The offset is the field index into a struct.
  13. type offset struct {
  14. index int
  15. export exporter
  16. }
  17. // offsetOf returns a field offset for the struct field.
  18. func offsetOf(f reflect.StructField, x exporter) offset {
  19. if len(f.Index) != 1 {
  20. panic("embedded structs are not supported")
  21. }
  22. if f.PkgPath == "" {
  23. return offset{index: f.Index[0]} // field is already exported
  24. }
  25. if x == nil {
  26. panic("exporter must be provided for unexported field")
  27. }
  28. return offset{index: f.Index[0], export: x}
  29. }
  30. // IsValid reports whether the offset is valid.
  31. func (f offset) IsValid() bool { return f.index >= 0 }
  32. // invalidOffset is an invalid field offset.
  33. var invalidOffset = offset{index: -1}
  34. // zeroOffset is a noop when calling pointer.Apply.
  35. var zeroOffset = offset{index: 0}
  36. // pointer is an abstract representation of a pointer to a struct or field.
  37. type pointer struct{ v reflect.Value }
  38. // pointerOfValue returns v as a pointer.
  39. func pointerOfValue(v reflect.Value) pointer {
  40. return pointer{v: v}
  41. }
  42. // pointerOfIface returns the pointer portion of an interface.
  43. func pointerOfIface(v interface{}) pointer {
  44. return pointer{v: reflect.ValueOf(v)}
  45. }
  46. // IsNil reports whether the pointer is nil.
  47. func (p pointer) IsNil() bool {
  48. return p.v.IsNil()
  49. }
  50. // Apply adds an offset to the pointer to derive a new pointer
  51. // to a specified field. The current pointer must be pointing at a struct.
  52. func (p pointer) Apply(f offset) pointer {
  53. if f.export != nil {
  54. if v := reflect.ValueOf(f.export(p.v.Interface(), f.index)); v.IsValid() {
  55. return pointer{v: v}
  56. }
  57. }
  58. return pointer{v: p.v.Elem().Field(f.index).Addr()}
  59. }
  60. // AsValueOf treats p as a pointer to an object of type t and returns the value.
  61. // It is equivalent to reflect.ValueOf(p.AsIfaceOf(t))
  62. func (p pointer) AsValueOf(t reflect.Type) reflect.Value {
  63. if got := p.v.Type().Elem(); got != t {
  64. panic(fmt.Sprintf("invalid type: got %v, want %v", got, t))
  65. }
  66. return p.v
  67. }
  68. // AsIfaceOf treats p as a pointer to an object of type t and returns the value.
  69. // It is equivalent to p.AsValueOf(t).Interface()
  70. func (p pointer) AsIfaceOf(t reflect.Type) interface{} {
  71. return p.AsValueOf(t).Interface()
  72. }
  73. func (p pointer) Bool() *bool { return p.v.Interface().(*bool) }
  74. func (p pointer) BoolPtr() **bool { return p.v.Interface().(**bool) }
  75. func (p pointer) BoolSlice() *[]bool { return p.v.Interface().(*[]bool) }
  76. func (p pointer) Int32() *int32 { return p.v.Interface().(*int32) }
  77. func (p pointer) Int32Ptr() **int32 { return p.v.Interface().(**int32) }
  78. func (p pointer) Int32Slice() *[]int32 { return p.v.Interface().(*[]int32) }
  79. func (p pointer) Int64() *int64 { return p.v.Interface().(*int64) }
  80. func (p pointer) Int64Ptr() **int64 { return p.v.Interface().(**int64) }
  81. func (p pointer) Int64Slice() *[]int64 { return p.v.Interface().(*[]int64) }
  82. func (p pointer) Uint32() *uint32 { return p.v.Interface().(*uint32) }
  83. func (p pointer) Uint32Ptr() **uint32 { return p.v.Interface().(**uint32) }
  84. func (p pointer) Uint32Slice() *[]uint32 { return p.v.Interface().(*[]uint32) }
  85. func (p pointer) Uint64() *uint64 { return p.v.Interface().(*uint64) }
  86. func (p pointer) Uint64Ptr() **uint64 { return p.v.Interface().(**uint64) }
  87. func (p pointer) Uint64Slice() *[]uint64 { return p.v.Interface().(*[]uint64) }
  88. func (p pointer) Float32() *float32 { return p.v.Interface().(*float32) }
  89. func (p pointer) Float32Ptr() **float32 { return p.v.Interface().(**float32) }
  90. func (p pointer) Float32Slice() *[]float32 { return p.v.Interface().(*[]float32) }
  91. func (p pointer) Float64() *float64 { return p.v.Interface().(*float64) }
  92. func (p pointer) Float64Ptr() **float64 { return p.v.Interface().(**float64) }
  93. func (p pointer) Float64Slice() *[]float64 { return p.v.Interface().(*[]float64) }
  94. func (p pointer) String() *string { return p.v.Interface().(*string) }
  95. func (p pointer) StringPtr() **string { return p.v.Interface().(**string) }
  96. func (p pointer) StringSlice() *[]string { return p.v.Interface().(*[]string) }
  97. func (p pointer) Bytes() *[]byte { return p.v.Interface().(*[]byte) }
  98. func (p pointer) BytesSlice() *[][]byte { return p.v.Interface().(*[][]byte) }
  99. func (p pointer) Extensions() *map[int32]ExtensionField {
  100. return p.v.Interface().(*map[int32]ExtensionField)
  101. }
  102. func (p pointer) Elem() pointer {
  103. return pointer{v: p.v.Elem()}
  104. }
  105. // PointerSlice copies []*T from p as a new []pointer.
  106. // This behavior differs from the implementation in pointer_unsafe.go.
  107. func (p pointer) PointerSlice() []pointer {
  108. // TODO: reconsider this
  109. if p.v.IsNil() {
  110. return nil
  111. }
  112. n := p.v.Elem().Len()
  113. s := make([]pointer, n)
  114. for i := 0; i < n; i++ {
  115. s[i] = pointer{v: p.v.Elem().Index(i)}
  116. }
  117. return s
  118. }
  119. // AppendPointerSlice appends v to p, which must be a []*T.
  120. func (p pointer) AppendPointerSlice(v pointer) {
  121. sp := p.v.Elem()
  122. sp.Set(reflect.Append(sp, v.v))
  123. }
  124. // SetPointer sets *p to v.
  125. func (p pointer) SetPointer(v pointer) {
  126. p.v.Elem().Set(v.v)
  127. }