110 lines
2.4 KiB
C#
110 lines
2.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text.Json;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace AxCopilot.Services;
|
|
|
|
internal static class UsageRankingService
|
|
{
|
|
private static readonly string _dataFile = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData), "AxCopilot", "usage.json");
|
|
|
|
private static Dictionary<string, int> _counts = new Dictionary<string, int>(StringComparer.OrdinalIgnoreCase);
|
|
|
|
private static bool _loaded = false;
|
|
|
|
private static readonly object _lock = new object();
|
|
|
|
public static void RecordExecution(string key)
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(key))
|
|
{
|
|
EnsureLoaded();
|
|
lock (_lock)
|
|
{
|
|
_counts.TryGetValue(key, out var value);
|
|
_counts[key] = value + 1;
|
|
}
|
|
SaveAsync();
|
|
}
|
|
}
|
|
|
|
public static int GetScore(string key)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(key))
|
|
{
|
|
return 0;
|
|
}
|
|
EnsureLoaded();
|
|
lock (_lock)
|
|
{
|
|
int value;
|
|
return _counts.TryGetValue(key, out value) ? value : 0;
|
|
}
|
|
}
|
|
|
|
public static IEnumerable<T> SortByUsage<T>(IEnumerable<T> items, Func<T, string?> keySelector)
|
|
{
|
|
EnsureLoaded();
|
|
return from x in items.Select((T item, int idx) => (item: item, idx: idx, score: GetScore(keySelector(item) ?? "")))
|
|
orderby x.score descending, x.idx
|
|
select x.item;
|
|
}
|
|
|
|
private static void EnsureLoaded()
|
|
{
|
|
if (_loaded)
|
|
{
|
|
return;
|
|
}
|
|
lock (_lock)
|
|
{
|
|
if (_loaded)
|
|
{
|
|
return;
|
|
}
|
|
try
|
|
{
|
|
if (File.Exists(_dataFile))
|
|
{
|
|
string json = File.ReadAllText(_dataFile);
|
|
Dictionary<string, int> dictionary = JsonSerializer.Deserialize<Dictionary<string, int>>(json);
|
|
if (dictionary != null)
|
|
{
|
|
_counts = new Dictionary<string, int>(dictionary, StringComparer.OrdinalIgnoreCase);
|
|
}
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
LogService.Warn("usage.json 로드 실패: " + ex.Message);
|
|
}
|
|
_loaded = true;
|
|
}
|
|
}
|
|
|
|
private static async Task SaveAsync()
|
|
{
|
|
try
|
|
{
|
|
Dictionary<string, int> snapshot;
|
|
lock (_lock)
|
|
{
|
|
snapshot = new Dictionary<string, int>(_counts);
|
|
}
|
|
Directory.CreateDirectory(Path.GetDirectoryName(_dataFile));
|
|
await File.WriteAllTextAsync(contents: JsonSerializer.Serialize(snapshot, new JsonSerializerOptions
|
|
{
|
|
WriteIndented = false
|
|
}), path: _dataFile);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Exception ex2 = ex;
|
|
LogService.Warn("usage.json 저장 실패: " + ex2.Message);
|
|
}
|
|
}
|
|
}
|