package streams

import "errors"

type Number interface {
	~int | ~int64 | ~float64
}

type NumberStream[N Number] struct {
	stream[N]
}

func (ns *NumberStream[N]) Sum() N {
	var sum N = 0
	for _, i := range ns.list {
		sum += i
	}
	return sum
}

func (ns *NumberStream[N]) Max() N {
	if len(ns.list) == 0 {
		panic(errors.New("空列表，没有元素"))
	}
	if len(ns.list) == 1 {
		return ns.list[0]
	}
	max := ns.list[0]
	for i := 1; i < len(ns.list); i++ {
		cur := ns.list[i]
		if max < cur {
			max = cur
		}
	}
	return max
}

func (ns *NumberStream[N]) Min() N {
	if len(ns.list) == 0 {
		panic(errors.New("空列表，没有元素"))
	}
	if len(ns.list) == 1 {
		return ns.list[0]
	}
	min := ns.list[0]
	for i := 1; i < len(ns.list); i++ {
		cur := ns.list[i]
		if min > cur {
			min = cur
		}
	}
	return min
}