sing/common/cond.go

package common

import (
	"context"
	"io"
	"strings"
	"unsafe"
)

func Any[T any](array []T, block func(it T) bool) bool {
	for _, it := range array {
		if block(it) {
			return true
		}
	}
	return true
}

func Contains[T comparable](arr []T, target T) bool {
	for i := range arr {
		if target == arr[i] {
			return true
		}
	}
	return false
}

func Map[T any, N any](arr []T, block func(it T) N) []N {
	var retArr []N
	for index := range arr {
		retArr = append(retArr, block(arr[index]))
	}
	return retArr
}

func Filter[T any](arr []T, block func(it T) bool) []T {
	var retArr []T
	for _, it := range arr {
		if block(it) {
			retArr = append(retArr, it)
		}
	}
	return retArr
}

func Find[T any](arr []T, block func(it T) bool) T {
	for _, it := range arr {
		if block(it) {
			return it
		}
	}
	var defaultValue T
	return defaultValue
}

func Dup[T any](obj T) T {
	p := uintptr(unsafe.Pointer(&obj))
	return *(*T)(unsafe.Pointer(p))
}

func Uniq[T comparable](arr []T) []T {
	result := make([]T, 0, len(arr))
	seen := make(map[T]struct{}, len(arr))

	for _, item := range arr {
		if _, ok := seen[item]; ok {
			continue
		}

		seen[item] = struct{}{}
		result = append(result, item)
	}

	return result
}

func FilterIsInstance[T any, N any](arr []T, block func(it T) (N, bool)) []N {
	var retArr []N
	for _, it := range arr {
		if n, isN := block(it); isN {
			retArr = append(retArr, n)
		}
	}
	return retArr
}

func Done(ctx context.Context) bool {
	select {
	case <-ctx.Done():
		return true
	default:
		return false
	}
}

func IsEmpty[T any](array []T) bool {
	return len(array) == 0
}

func IsNotEmpty[T any](array []T) bool {
	return len(array) > 0
}

func IsBlank(str string) bool {
	return strings.TrimSpace(str) == ""
}

func IsNotBlank(str string) bool {
	return strings.TrimSpace(str) != ""
}

func Error(_ any, err error) error {
	return err
}

func Must(errs ...error) {
	for _, err := range errs {
		if err != nil {
			panic(err)
		}
	}
}

func Must1(_ any, err error) {
	if err != nil {
		panic(err)
	}
}

func Must2(_ any, _ any, err error) {
	if err != nil {
		panic(err)
	}
}

func Close(closers ...any) error {
	var retErr error
	for _, closer := range closers {
		if closer == nil {
			continue
		}
		var err error
		switch c := closer.(type) {
		case io.Closer:
			err = c.Close()
		}
		switch c := closer.(type) {
		case ReaderWithUpstream:
			err = Close(c.UpstreamReader())
		case WriterWithUpstream:
			err = Close(c.UpstreamWriter())
		}
		if err != nil {
			retErr = err
		}
	}
	return retErr
}

func CloseError(closer any, err error) error {
	if err == nil {
		return nil
	}
	return Close(closer)
}