wp-go/helper/number/number.go

package number

import (
	"fmt"
	"math/rand"
	"time"
)

var _ = func() any {
	rand.Seed(time.Now().UnixNano())
	return nil
}()

type IntNumber interface {
	~int | ~int64 | ~int32 | ~int8 | ~int16 |
		~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64
}

type Number interface {
	IntNumber | ~float64 | ~float32
}

func Range[T IntNumber](start, end, step T) []T {
	if step == 0 {
		panic("step can't be 0")
	}
	l := int((end-start+1)/step + 1)
	if l < 0 {
		l = 0 - l
	}
	r := make([]T, 0, l)
	for i := start; ; {
		r = append(r, i)
		i = i + step
		if (step > 0 && i > end) || (step < 0 && i < end) {
			break
		}
	}
	return r
}

// Rand 都为闭区间 [start,end]
func Rand[T IntNumber](start, end T) T {
	end++
	return T(rand.Int63n(int64(end-start))) + start
}

func Min[T Number](a ...T) T {
	min := a[0]
	for _, t := range a {
		if min > t {
			min = t
		}
	}
	return min
}

func Max[T Number](a ...T) T {
	max := a[0]
	for _, t := range a {
		if max < t {
			max = t
		}
	}
	return max
}

func Sum[T Number](a ...T) T {
	s := T(0)
	for _, t := range a {
		s += t
	}
	return s
}

func Add[T Number](i, j T) T {
	return i + j
}
func Sub[T Number](i, j T) T {
	return i - j
}

func ToString[T Number](n T) string {
	return fmt.Sprintf("%v", n)
}

func Abs[T Number](n T) T {
	if n >= 0 {
		return n
	}
	return -n
}

func Mul[T Number](i, j T) T {
	return i * j
}

func Divide[T Number](i, j T) T {
	return i / j
}