//
// option.go
// Copyright (C) 2023 tiglog <me@tiglog.com>
//
// Distributed under terms of the MIT license.
//

package gcache

import "time"

// SetIOption The option used to cache set
type SetIOption func(ICache, string, IItem) bool

// WithEx Set the specified expire time, in time.Duration.
func WithEx(d time.Duration) SetIOption {
	return func(c ICache, k string, v IItem) bool {
		v.SetExpireAt(time.Now().Add(d))
		return true
	}
}

// WithExAt Set the specified expire deadline, in time.Time.
func WithExAt(t time.Time) SetIOption {
	return func(c ICache, k string, v IItem) bool {
		v.SetExpireAt(t)
		return true
	}
}

// ICacheOption The option used to create the cache object
type ICacheOption func(conf *Config)

// WithShards set custom size of sharding. Default is 1024
// The larger the size, the smaller the lock force, the higher the concurrency performance,
// and the higher the memory footprint, so try to choose a size that fits your business scenario
func WithShards(shards int) ICacheOption {
	if shards <= 0 {
		panic("Invalid shards")
	}
	return func(conf *Config) {
		conf.shards = shards
	}
}

// WithExpiredCallback set custom expired callback function
// This callback function is called when the key-value pair expires
func WithExpiredCallback(ec ExpiredCallback) ICacheOption {
	return func(conf *Config) {
		conf.expiredCallback = ec
	}
}

// WithHash set custom hash key function
func WithHash(hash IHash) ICacheOption {
	return func(conf *Config) {
		conf.hash = hash
	}
}

// WithClearInterval set custom clear interval.
// Interval for clearing expired key-value pairs. The default value is 1 second
// If the d is 0, the periodic clearing function is disabled
func WithClearInterval(d time.Duration) ICacheOption {
	return func(conf *Config) {
		conf.clearInterval = d
	}
}