wp-go/helper/number/number.go
2023-01-28 21:38:34 +08:00

101 lines
1.4 KiB
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
}