package sync

import (
	"sync"
	"time"
)

//经试验当前最高支持200wQPS

var lock sync.RWMutex

type Limit struct {
	times    int
	num      int
	now      int64
	interval int
	stop     bool
	timemap  map[int64]int64
}

func (l *Limit) Stop() {
	l.stop = true
}
func (l *Limit) GetStop() bool {
	return l.stop
}

func Limiter(times int) *Limit {
	if times < 0 {
		panic("限制出错,不能小于1")
	}
	if times == 0 {
		times = 1
	}
	l := &Limit{
		times:    times,
		now:      -1,
		interval: 1e7 / (times + 5),
		timemap:  make(map[int64]int64),
	}

	return l
}
func (l *Limit) TimeLine() map[int64]int64 {
	return l.timemap
}
func (l *Limit) Get() {

	lock.Lock()
	defer lock.Unlock()

	now := time.Now().Unix()
	if l.now == now {

		if l.num < l.times {
			l.num++
			time.Sleep(time.Duration(l.interval))
			l.timemap[l.now] += 1
			return
		} else {
			next := time.Unix(now+1, -100)
			time.Sleep(time.Until(next))
			now = now + 1
		}
	}
	l.now = now
	l.num = 1
	l.timemap[l.now] += 1
}