package snowflake_proxy

import (
	"sync/atomic"
)

type tokens_t struct {
	ch       chan struct{}
	capacity uint
	clients  atomic.Int64
}

func newTokens(capacity uint) *tokens_t {
	var ch chan struct{}
	if capacity != 0 {
		ch = make(chan struct{}, capacity)
	}

	return &tokens_t{
		ch:       ch,
		capacity: capacity,
		clients:  atomic.Int64{},
	}
}

func (t *tokens_t) get() {
	t.clients.Add(1)

	if t.capacity != 0 {
		t.ch <- struct{}{}
	}
}

func (t *tokens_t) ret() {
	t.clients.Add(-1)

	if t.capacity != 0 {
		<-t.ch
	}
}

func (t *tokens_t) count() int64 {
	return t.clients.Load()
}