mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-12-05 01:52:46 +00:00
279 lines
4.7 KiB
Go
279 lines
4.7 KiB
Go
|
package jsoniter
|
||
|
|
||
|
import (
|
||
|
"reflect"
|
||
|
"unsafe"
|
||
|
)
|
||
|
|
||
|
type arrayLazyAny struct {
|
||
|
baseAny
|
||
|
cfg *frozenConfig
|
||
|
buf []byte
|
||
|
err error
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ValueType() ValueType {
|
||
|
return ArrayValue
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) MustBeValid() Any {
|
||
|
return any
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) LastError() error {
|
||
|
return any.err
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToBool() bool {
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
return iter.ReadArray()
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToInt() int {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToInt32() int32 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToInt64() int64 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToUint() uint {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToUint32() uint32 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToUint64() uint64 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToFloat32() float32 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToFloat64() float64 {
|
||
|
if any.ToBool() {
|
||
|
return 1
|
||
|
}
|
||
|
return 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToString() string {
|
||
|
return *(*string)(unsafe.Pointer(&any.buf))
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) ToVal(val interface{}) {
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
iter.ReadVal(val)
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) Get(path ...interface{}) Any {
|
||
|
if len(path) == 0 {
|
||
|
return any
|
||
|
}
|
||
|
switch firstPath := path[0].(type) {
|
||
|
case int:
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
valueBytes := locateArrayElement(iter, firstPath)
|
||
|
if valueBytes == nil {
|
||
|
return newInvalidAny(path)
|
||
|
}
|
||
|
iter.ResetBytes(valueBytes)
|
||
|
return locatePath(iter, path[1:])
|
||
|
case int32:
|
||
|
if '*' == firstPath {
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
arr := make([]Any, 0)
|
||
|
iter.ReadArrayCB(func(iter *Iterator) bool {
|
||
|
found := iter.readAny().Get(path[1:]...)
|
||
|
if found.ValueType() != InvalidValue {
|
||
|
arr = append(arr, found)
|
||
|
}
|
||
|
return true
|
||
|
})
|
||
|
return wrapArray(arr)
|
||
|
}
|
||
|
return newInvalidAny(path)
|
||
|
default:
|
||
|
return newInvalidAny(path)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) Size() int {
|
||
|
size := 0
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
iter.ReadArrayCB(func(iter *Iterator) bool {
|
||
|
size++
|
||
|
iter.Skip()
|
||
|
return true
|
||
|
})
|
||
|
return size
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) WriteTo(stream *Stream) {
|
||
|
stream.Write(any.buf)
|
||
|
}
|
||
|
|
||
|
func (any *arrayLazyAny) GetInterface() interface{} {
|
||
|
iter := any.cfg.BorrowIterator(any.buf)
|
||
|
defer any.cfg.ReturnIterator(iter)
|
||
|
return iter.Read()
|
||
|
}
|
||
|
|
||
|
type arrayAny struct {
|
||
|
baseAny
|
||
|
val reflect.Value
|
||
|
}
|
||
|
|
||
|
func wrapArray(val interface{}) *arrayAny {
|
||
|
return &arrayAny{baseAny{}, reflect.ValueOf(val)}
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ValueType() ValueType {
|
||
|
return ArrayValue
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) MustBeValid() Any {
|
||
|
return any
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) LastError() error {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToBool() bool {
|
||
|
return any.val.Len() != 0
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToInt() int {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToInt32() int32 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToInt64() int64 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToUint() uint {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToUint32() uint32 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToUint64() uint64 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToFloat32() float32 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToFloat64() float64 {
|
||
|
if any.val.Len() == 0 {
|
||
|
return 0
|
||
|
}
|
||
|
return 1
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) ToString() string {
|
||
|
str, _ := MarshalToString(any.val.Interface())
|
||
|
return str
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) Get(path ...interface{}) Any {
|
||
|
if len(path) == 0 {
|
||
|
return any
|
||
|
}
|
||
|
switch firstPath := path[0].(type) {
|
||
|
case int:
|
||
|
if firstPath < 0 || firstPath >= any.val.Len() {
|
||
|
return newInvalidAny(path)
|
||
|
}
|
||
|
return Wrap(any.val.Index(firstPath).Interface())
|
||
|
case int32:
|
||
|
if '*' == firstPath {
|
||
|
mappedAll := make([]Any, 0)
|
||
|
for i := 0; i < any.val.Len(); i++ {
|
||
|
mapped := Wrap(any.val.Index(i).Interface()).Get(path[1:]...)
|
||
|
if mapped.ValueType() != InvalidValue {
|
||
|
mappedAll = append(mappedAll, mapped)
|
||
|
}
|
||
|
}
|
||
|
return wrapArray(mappedAll)
|
||
|
}
|
||
|
return newInvalidAny(path)
|
||
|
default:
|
||
|
return newInvalidAny(path)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) Size() int {
|
||
|
return any.val.Len()
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) WriteTo(stream *Stream) {
|
||
|
stream.WriteVal(any.val)
|
||
|
}
|
||
|
|
||
|
func (any *arrayAny) GetInterface() interface{} {
|
||
|
return any.val.Interface()
|
||
|
}
|