mapo-core/MP.IOC/Services/RedisWeightProvider.cs

using StackExchange.Redis;

namespace MP.IOC.Services
{


    public class RedisWeightProvider : IWeightProvider
    {
        private readonly IDatabase _db;
        private readonly int _defaultOld;
        private readonly int _defaultNew;
        private const string KeyPrefix = "route_weight:"; // key per metodo: route_weight:enabled

        public RedisWeightProvider(IConnectionMultiplexer mux, IConfiguration config)
        {
            _db = mux.GetDatabase();
            _defaultOld = config.GetValue<int>("RouteMan:DefaultWeightOld", 100);
            _defaultNew = config.GetValue<int>("RouteMan:DefaultWeightNew", 0);
        }

        /// <summary>
        /// Ritorna (oldWeight, newWeight) per il metodo. Se non esiste, crea la chiave con i default.
        /// </summary>
        public (int oldWeight, int newWeight) GetWeightsFor(string method)
        {
            if (string.IsNullOrEmpty(method)) method = "unknown";
            var key = KeyPrefix + method;

            // Leggi entrambi i campi
            var oldVal = _db.HashGet(key, "old");
            var newVal = _db.HashGet(key, "new");

            // Se entrambi mancanti, inizializza con default (usando HSet con When.NotExists per evitare overwrite)
            if (oldVal.IsNull && newVal.IsNull)
            {
                // Imposta i campi singolarmente con When.NotExists per evitare overwrite
                _db.HashSet(key, "old", _defaultOld, When.NotExists);
                _db.HashSet(key, "new", _defaultNew, When.NotExists);

                // Rileggi per essere sicuri
                oldVal = _db.HashGet(key, "old");
                newVal = _db.HashGet(key, "new");
            }

            // Se uno dei due manca, impostalo al default (non sovrascrive l'altro)
            if (oldVal.IsNull)
            {
                _db.HashSet(key, "old", _defaultOld, When.NotExists);
                oldVal = _defaultOld;
            }
            if (newVal.IsNull)
            {
                _db.HashSet(key, "new", _defaultNew, When.NotExists);
                newVal = _defaultNew;
            }

            if (!int.TryParse(oldVal.ToString(), out var oldW)) oldW = _defaultOld;
            if (!int.TryParse(newVal.ToString(), out var newW)) newW = _defaultNew;

            // clamp 0..100
            oldW = Math.Clamp(oldW, 0, 100);
            newW = Math.Clamp(newW, 0, 100);

            return (oldW, newW);
        }

        // API per aggiornare i pesi a runtime (opzionale)
        public void SetWeights(string method, int oldWeight, int newWeight)
        {
            var key = KeyPrefix + (string.IsNullOrEmpty(method) ? "unknown" : method);
            _db.HashSet(key, new HashEntry[] {
            new HashEntry("old", Math.Clamp(oldWeight,0,100)),
            new HashEntry("new", Math.Clamp(newWeight,0,100))
        });
        }
    }

}