73 lines
2.3 KiB
C#
73 lines
2.3 KiB
C#
|
using log4net.Util;
|
|||
|
|
using Newtonsoft.Json;
|
|||
|
|
using StackExchange.Redis;
|
|||
|
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Linq;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Threading.Tasks;
|
|||
|
|
|
|||
|
|
namespace Hopetry.Services
|
|||
|
|
{
|
|||
|
|
public class RedisService : IDisposable
|
|||
|
|
{
|
|||
|
|
private readonly ConnectionMultiplexer _redis;
|
|||
|
|
private readonly IDatabase _db;
|
|||
|
|
private const string Prefix = "client:";
|
|||
|
|
|
|||
|
|
public RedisService(string connectionString)
|
|||
|
|
{
|
|||
|
|
var config = ConfigurationOptions.Parse(connectionString);
|
|||
|
|
config.AbortOnConnectFail = false;
|
|||
|
|
config.ConnectTimeout = 5000;
|
|||
|
|
_redis = ConnectionMultiplexer.Connect(config);
|
|||
|
|
_db = _redis.GetDatabase();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 存储客户端信息（自动过期30天）
|
|||
|
|
public async Task StoreClientInfoAsync(Hopetry.Provider.SystemInfo info)
|
|||
|
|
{
|
|||
|
|
var key = $"{Prefix}{info.MachineId}";
|
|||
|
|
var json = JsonConvert.SerializeObject(info);
|
|||
|
|
await _db.StringSetAsync(key, json, TimeSpan.FromDays(30));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取单个客户端信息
|
|||
|
|
public async Task<SystemInfo> GetClientInfoAsync(string machineId)
|
|||
|
|
{
|
|||
|
|
var json = await _db.StringGetAsync($"{Prefix}{machineId}");
|
|||
|
|
return json.HasValue ? JsonConvert.DeserializeObject<SystemInfo>(json) : null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取所有客户端信息
|
|||
|
|
public async Task<List<SystemInfo>> GetAllClientsAsync()
|
|||
|
|
{
|
|||
|
|
var endpoints = _redis.GetEndPoints();
|
|||
|
|
var server = _redis.GetServer(endpoints.First());
|
|||
|
|
var keys = server.Keys(pattern: $"{Prefix}*").ToArray();
|
|||
|
|
|
|||
|
|
var results = new List<SystemInfo>();
|
|||
|
|
foreach (var key in keys)
|
|||
|
|
{
|
|||
|
|
var json = await _db.StringGetAsync(key);
|
|||
|
|
if (json.HasValue)
|
|||
|
|
{
|
|||
|
|
results.Add(JsonConvert.DeserializeObject<SystemInfo>(json));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return results;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 删除客户端信息
|
|||
|
|
public async Task<bool> DeleteClientAsync(string machineId)
|
|||
|
|
{
|
|||
|
|
return await _db.KeyDeleteAsync($"{Prefix}{machineId}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Dispose()
|
|||
|
|
{
|
|||
|
|
_redis?.Dispose();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|