package pool_test

import (
	"code.squareroundforest.org/arpio/pool"
	"time"
)

type maxAlgo int

type timeoutAlgo struct {
	to    time.Duration
	items []time.Time
}

func (a maxAlgo) Target(s pool.Stats) (int, time.Duration) {
	if s.Idle <= int(a) {
		return s.Idle, 0
	}

	return int(a), 0
}

func (a *timeoutAlgo) Target(s pool.Stats) (int, time.Duration) {
	now := time.Now()
	for len(a.items) < s.Idle {
		a.items = append(a.items, now)
	}

	t := s.Idle
	for len(a.items) > 0 && a.items[0].Add(a.to).Before(now) {
		t--
		a.items = a.items[1:]
	}

	if t < 0 {
		t = 0
	}

	if len(a.items) == 0 {
		return t, 0
	}

	return t, a.items[0].Add(a.to).Sub(now)
}