gf/util/gconv/internal/converter/converter_slice_map.go

// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.

package converter

import "github.com/gogf/gf/v2/internal/json"

// SliceMapOption is the option for SliceMap function.
type SliceMapOption struct {
	SliceOption SliceOption
	MapOption   MapOption
}

func (c *Converter) getSliceMapOption(option ...SliceMapOption) SliceMapOption {
	if len(option) > 0 {
		return option[0]
	}
	return SliceMapOption{}
}

// SliceMap converts `value` to []map[string]any.
// Note that it automatically checks and converts json string to []map if `value` is string/[]byte.
func (c *Converter) SliceMap(value any, option ...SliceMapOption) ([]map[string]any, error) {
	if value == nil {
		return nil, nil
	}
	switch r := value.(type) {
	case string:
		list := make([]map[string]any, 0)
		if len(r) > 0 && r[0] == '[' && r[len(r)-1] == ']' {
			if err := json.UnmarshalUseNumber([]byte(r), &list); err != nil {
				return nil, err
			}
			return list, nil
		}
		return nil, nil

	case []byte:
		list := make([]map[string]any, 0)
		if len(r) > 0 && r[0] == '[' && r[len(r)-1] == ']' {
			if err := json.UnmarshalUseNumber(r, &list); err != nil {
				return nil, err
			}
			return list, nil
		}
		return nil, nil

	case []map[string]any:
		return r, nil

	default:
		sliceMapOption := c.getSliceMapOption(option...)
		array, err := c.SliceAny(value, sliceMapOption.SliceOption)
		if err != nil {
			return nil, err
		}
		if len(array) == 0 {
			return nil, nil
		}
		list := make([]map[string]any, len(array))
		for k, v := range array {
			m, err := c.Map(v, sliceMapOption.MapOption)
			if err != nil && !sliceMapOption.SliceOption.ContinueOnError {
				return nil, err
			}
			list[k] = m
		}
		return list, nil
	}
}