go-utils/utils/array.go

package utils

import (
	"github.com/spf13/cast"
	"strings"
)

// ArrayFilter - 数组过滤
func ArrayFilter(array []string) (slice []string) {
	for key, val := range array {
		if (key > 0 && array[key-1] == val) || len(val) == 0 {
			continue
		}
		slice = append(slice, val)
	}
	return
}

// ArrayRemove - 数组删除
func ArrayRemove(array []string, args ...string) []string {
	for _, value := range args {
		for key, val := range array {
			// 去除空格
			val = strings.TrimSpace(val)
			// 根据索引删除
			if val == value {
				array = append(array[:key], array[key+1:]...)
			}
		}
	}
	return array
}

// ArrayUnique - 数组去重
func ArrayUnique[T any](array []T) (slice []T) {
	list := make(map[any]bool)
	for _, item := range array {
		if !list[item] {
			list[item] = true
			slice = append(slice, item)
		}
	}
	return slice
}

// ArrayEmpty - 数组去空
func ArrayEmpty[T any](array []T) (slice []T) {
	for _, item := range array {
		if !Is.Empty(item) {
			slice = append(slice, item)
		}
	}
	return slice
}

// ArrayMapWithField - 数组Map保留指定字段
func ArrayMapWithField(array []map[string]any, field any) (slice []any) {

	// 获取字段
	keys := cast.ToStringSlice(Unity.Keys(field))

	if Is.Empty(keys) {
		return cast.ToSlice(array)
	}

	for _, item := range array {
		val := Map.WithField(cast.ToStringMap(item), keys)
		slice = append(cast.ToSlice(slice), val)
	}

	return slice
}

// ArrayReverse - 数组反转
func ArrayReverse[T any](array []T) (slice []T) {
	for i, j := 0, len(array)-1; i < j; i, j = i+1, j-1 {
		array[i], array[j] = array[j], array[i]
	}
	return array
}

// ArrayDiff - 数组差集
func ArrayDiff[T any](array1, array2 []T) (slice []T) {
	for _, item := range array1 {
		if !InArray(item, array2) {
			slice = append(slice, item)
		}
	}
	return slice
}

// ArrayIntersect - 交集
func ArrayIntersect[T any](array1, array2 []T) (slice []T) {
	for _, item := range array1 {
		if InArray(item, array2) {
			slice = append(slice, item)
		}
	}
	return slice
}