package basic

import (
	"math/big"
)

// NodeSizeTable is not thread safe, make sure it is accessed with external
// locks or in single routine.
type NodeSizeTable struct {
	prices map[string]uint64
	total  uint64
}

func (t *NodeSizeTable) Put(id []byte, avg uint64) {
	t.prices[string(id)] += avg
	t.total += avg
}

func (t *NodeSizeTable) Total() *big.Int {
	return new(big.Int).SetUint64(t.total)
}

func (t *NodeSizeTable) Iterate(f func([]byte, *big.Int)) {
	for k, v := range t.prices {
		n := new(big.Int).SetUint64(v)
		f([]byte(k), n)
	}
}

func NewNodeSizeTable() *NodeSizeTable {
	return &NodeSizeTable{
		prices: make(map[string]uint64),
	}
}

type nodeInfoWrapper []byte

func (nodeInfoWrapper) Price() *big.Int {
	panic("should not be used")
}

func (n nodeInfoWrapper) PublicKey() []byte {
	return n
}