using Microsoft.AspNetCore.Mvc.Filters;
using System.Collections.Concurrent;
using System.Text.Json;

public class ThrottleFilter : IAsyncActionFilter
{
    private static readonly ConcurrentDictionary<string, DateTime> _lastExecutionTimes = new ConcurrentDictionary<string, DateTime>();
    private readonly int _intervalInSeconds;

    public ThrottleFilter(int intervalInSeconds)
    {
        _intervalInSeconds = intervalInSeconds;
    }

    public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
    {
        var actionName = context.ActionDescriptor.DisplayName;

        var parameterString = GenerateParameterString(context.ActionArguments);
        var key = $"{actionName}_{parameterString}";

        if (_lastExecutionTimes.TryGetValue(key, out var lastExecutionTime))
        {
            var elapsedTime = DateTime.Now - lastExecutionTime;
            if (elapsedTime.TotalSeconds < _intervalInSeconds)
            {
                context.Result = new OkObjectResult(new WebResponseContent().Error("请求过于频繁，请稍后再试"));
                return;
            }
        }

        _lastExecutionTimes[key] = DateTime.Now;
        await next();
    }

    private string GenerateParameterString(IDictionary<string, object> arguments)
    {
        if (arguments == null || arguments.Count == 0)
        {
            return "";
        }

        var paramStrings = new List<string>();
        foreach (var argument in arguments)
        {
            var key = argument.Key;
            var value = argument.Value;
            string valueString;

            if (value == null)
            {
                valueString = "null";
            }
            else if (IsSimpleType(value.GetType()))
            {
                valueString = value.ToString();
            }
            else
            {
                valueString = JsonSerializer.Serialize(value);
            }

            paramStrings.Add($"{key}={valueString}");
        }

        return string.Join("&", paramStrings);
    }

    // 判断类型是否为简单类型
    private bool IsSimpleType(Type type)
    {
        return type.IsPrimitive || type == typeof(string) || type == typeof(decimal);
    }
}