feat: 代码优化、机器计划任务完善

server/pkg/utils/collx/array.go

@@ -0,0 +1,81 @@
+package collx
+
+import "fmt"
+
+// 数组比较
+// 依次返回，新增值，删除值，以及不变值
+func ArrayCompare[T any](newArr []T, oldArr []T, compareFun func(T, T) bool) ([]T, []T, []T) {
+	var unmodifierValue []T
+	ni, oi := 0, 0
+	for {
+		if ni >= len(newArr) {
+			break
+		}
+		nv := newArr[ni]
+		for {
+			if oi >= len(oldArr) {
+				oi = 0
+				break
+			}
+			ov := oldArr[oi]
+			if compareFun(nv, ov) {
+				unmodifierValue = append(unmodifierValue, nv)
+				// 新数组移除该位置值
+				if len(newArr) > ni {
+					newArr = append(newArr[:ni], newArr[ni+1:]...)
+					ni = ni - 1
+				}
+				if len(oldArr) > oi {
+					oldArr = append(oldArr[:oi], oldArr[oi+1:]...)
+					oi = oi - 1
+				}
+			}
+			oi = oi + 1
+		}
+		ni = ni + 1
+	}
+
+	return newArr, oldArr, unmodifierValue
+}
+
+type NumT interface {
+	~int | ~int32 | ~uint64
+}
+
+func NumberArr2StrArr[T NumT](numberArr []T) []string {
+	strArr := make([]string, 0)
+	for _, v := range numberArr {
+		strArr = append(strArr, fmt.Sprintf("%d", v))
+	}
+	return strArr
+}
+
+// 判断数组中是否含有指定元素
+func ArrayContains[T comparable](arr []T, el T) bool {
+	for _, v := range arr {
+		if v == el {
+			return true
+		}
+	}
+	return false
+}
+
+// 数组转为map
+// @param keyFunc key的主键
+func ArrayToMap[T any, K comparable](arr []T, keyFunc func(val T) K) map[K]T {
+	res := make(map[K]T, len(arr))
+	for _, val := range arr {
+		key := keyFunc(val)
+		res[key] = val
+	}
+	return res
+}
+
+// 数组映射，即将一数组元素通过映射函数转换为另一数组
+func ArrayMap[T any, K comparable](arr []T, mapFunc func(val T) K) []K {
+	res := make([]K, len(arr))
+	for i, val := range arr {
+		res[i] = mapFunc(val)
+	}
+	return res
+}