48 lines
968 B
Go
48 lines
968 B
Go
package utils
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/mozillazg/go-pinyin"
|
|
)
|
|
|
|
func Contains[T string | int | uint](element T, slice []T) bool {
|
|
for _, v := range slice {
|
|
if v == element {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func ContainsInsensitive(element string, slice []string) bool {
|
|
lowercasedElement := strings.TrimSpace(strings.ToLower(element))
|
|
return Contains(lowercasedElement, slice)
|
|
}
|
|
|
|
func PinyinAbbr(source string) string {
|
|
abbr := pinyin.Pinyin(source, pinyin.NewArgs())
|
|
var abbrCollect = make([]string, 0)
|
|
for _, a := range abbr {
|
|
abbrCollect = append(abbrCollect, a[0][0:1])
|
|
}
|
|
finalAbbr := strings.Join(abbrCollect, "")
|
|
return finalAbbr
|
|
}
|
|
|
|
func Keys[T comparable, V any](m map[T]V) []T {
|
|
keys := make([]T, 0, len(m))
|
|
for key := range m {
|
|
keys = append(keys, key)
|
|
}
|
|
return keys
|
|
}
|
|
|
|
func Values[K comparable, V any](m map[K]V) []V {
|
|
values := make([]V, 0, len(m))
|
|
for _, v := range m {
|
|
values = append(values, v)
|
|
}
|
|
return values
|
|
}
|