// ------------------------------------------------------------------------
//   Project atila
//   Active Thing (activething.com) git.activething.com/go
//
//   File name  pool_ops.go
//   Created by DEV
//   Modified   17/01/2024
//
//   Copyright 2024 activething.com
// ------------------------------------------------------------------------
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ------------------------------------------------------------------------

package syncs

import (
	"opts"
	"sync/atomic"
)

func PoolWithNew[T any](nfn func() *T) opts.GOptionFnc[Pool[T]] {
	return func(p *Pool[T]) {
		p.createFn = nfn
	}
}


func PoolWithCap[T any](cap int32) opts.GOptionFnc[Pool[T]] {
	return func(p *Pool[T]) {
		atomic.StoreInt32(&p.cap, cap)
	}
}


func PoolWithLen[T any](len int32) opts.GOptionFnc[Pool[T]] {
	return func(p *Pool[T]) {
		if p.createFn == nil {
			return
		}
		c := p.Cap()
		if c != -1 && len > c {
			len = c
		}
		for i := int32(0); i < len; i++ {
			p.Put(p.createFn())
		}
	}
}


func PoolWithSet[T any](els ...*T) opts.GOptionFnc[Pool[T]] {
	return func(p *Pool[T]) {
		for _, e := range els {
			if e != nil {
				p.Put(e)
			}
		}
	}
}