BotSharp/src/Infrastructure/BotSharp.Core/Statistics/Services/BotSharpStatService.cs

using BotSharp.Abstraction.Infrastructures;
using BotSharp.Abstraction.Statistics.Settings;

namespace BotSharp.Core.Statistics.Services;

public class BotSharpStatService : IBotSharpStatService
{
    private readonly IServiceProvider _services;
    private readonly ILogger<BotSharpStatService> _logger;
    private readonly StatisticsSettings _settings;

    private const string GLOBAL_LLM_COST = "global-llm-cost";
    private const string GLOBAL_AGENT_CALL = "global-agent-call";
    private const int TIMEOUT_SECONDS = 5;

    public BotSharpStatService(
        IServiceProvider services,
        ILogger<BotSharpStatService> logger,
        StatisticsSettings settings)
    {
        _services = services;
        _logger = logger;
        _settings = settings;
    }

    public bool UpdateLlmCost(BotSharpStats stats)
    {
        try
        {
            if (!_settings.Enabled) return false;

            var db = _services.GetRequiredService<IBotSharpRepository>();
            var locker = _services.GetRequiredService<IDistributedLocker>();

            var res = locker.Lock(GLOBAL_LLM_COST, () =>
            {
                var body = db.GetGlobalStats(stats.Category, stats.Group, stats.RecordDate);
                if (body == null)
                {
                    db.SaveGlobalStats(stats);
                    return;
                }

                foreach (var item in stats.Data)
                {
                    var value = item.Value;
                    if (body.Data.TryGetValue(item.Key, out var curValue) && curValue != null)
                    {
                        var str = curValue.ToString();
                        if (long.TryParse(str, out var count))
                        {
                            value = long.Parse(value?.ToString() ?? "0") + count;
                        }
                        else if (decimal.TryParse(str, out var num))
                        {
                            value = decimal.Parse(value?.ToString() ?? "0") + num;
                        }
                    }
                    body.Data[item.Key] = value;
                }

                db.SaveGlobalStats(body);
            }, TIMEOUT_SECONDS);
            return res;
        }
        catch (Exception ex)
        {
            _logger.LogError($"Error when updating global llm cost stats {stats}. {ex.Message}\r\n{ex.InnerException}");
            return false;
        }
    }

    public bool UpdateAgentCall(BotSharpStats stats)
    {
        try
        {
            if (!_settings.Enabled) return false;

            var db = _services.GetRequiredService<IBotSharpRepository>();
            var locker = _services.GetRequiredService<IDistributedLocker>();

            var res = locker.Lock(GLOBAL_AGENT_CALL, () =>
            {
                var body = db.GetGlobalStats(stats.Category, stats.Group, stats.RecordDate);
                if (body == null)
                {
                    db.SaveGlobalStats(stats);
                    return;
                }

                foreach (var item in stats.Data)
                {
                    var value = item.Value;
                    if (body.Data.TryGetValue(item.Key, out var curValue) && curValue != null)
                    {
                        var str = curValue.ToString();
                        if (long.TryParse(str, out var count))
                        {
                            value = long.Parse(value?.ToString() ?? "0") + count;
                        }
                    }
                    body.Data[item.Key] = value;
                }

                db.SaveGlobalStats(body);
            }, TIMEOUT_SECONDS);
            return res;
        }
        catch (Exception ex)
        {
            _logger.LogError($"Error when updating global agent call stats {stats}. {ex.Message}\r\n{ex.InnerException}");
            return false;
        }
    }
}