using System;
using Microsoft.Extensions.Caching.Memory;

namespace IStation
{
    internal sealed partial class SharedMemoryCache
    {
        /// <summary>
        /// If key not found, sets a new cache item by key, function and optional eviction
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="key">A unique identifier for the cache entry to insert</param>
        /// <param name="valueFunction">A function to execute to get the value for the cache entry</param>
        /// <param name="policy">(Optional) An object that contains eviction details for the cache entry</param>
        /// <returns>An object of type T, or default T</returns>
        public T AddOrGetExisting<T>(string key, Func<T> valueFunction, int? seconds)
        {
            if (!HasKey(key))
            {
                // GetSet calls can be cancelled, where as Set calls expect to always run
                return this.GetSet<T>(key, valueFunction, seconds);
            }

            return this.Get<T>(key);
        }

        /// <summary>
        /// If key not found, sets a new cache item by key, value and optional eviction
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="key">A unique identifier for the cache entry to insert</param>
        /// <param name="value">A value for the cache entry</param>
        /// <param name="policy">(Optional) An object that contains eviction details for the cache entry</param>
        /// <returns>An object of type T, or default T</returns>
        public T AddOrGetExisting<T>(string key, T value, int? seconds)
        {
            if (!HasKey(key))
            {
                return GetSet<T>(key, value, seconds);
            }

            return this.Get<T>(key);
        }
    }
}