您的位置:首页 > 数据库 > Redis

c#中用lua脚本执行redis命令

2018-01-09 10:17 447 查看
直接贴出代码,实现执行lua脚本的方法,用到的第三方类库是 StackExchange.Redis(nuget上有)
注:下面的代码是简化后的,实际使用要修改,

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Threading.Tasks;
using StackExchange.Redis;

namespace TL.Cloud.KV
{
public class Kv:IKv
{

private  ConfigurationOptions _redisConfig;// = new ConfigurationOptions
private  IDatabaseAsync _db;//=Connection.GetDatabase();
private  IServer _server;// = GetServer();
private  string _keyPrefix;// 区分前缀

/// <summary>
/// 每个微服务一个kv云实例,或共享云实例中(无独立前缀,服务间Key取名需要自行防止冲突)
/// </summary>
/// <param name="hostServiceName">宿主微服务</param>
public Kv(PublicCloudKvConfig config)
{
Init(config);
}

private void Init(PublicCloudKvConfig config)
{
//独特的区分dev/100前缀,online可以没有
_keyPrefix = string.IsNullOrWhiteSpace(config.KeyPrefix)
?null
:config.KeyPrefix.Trim().ToLower()+':';//小写带冒号分割
//ali kv
var kvUrl = config.KvUrl;

_redisConfig = ConfigurationOptions.Parse(kvUrl);

_redisConfig.Password = config.KvPassword;
_redisConfig.SetDefaultPorts(); //自动填充默认端口
var connection = ConnectionMultiplexer.Connect(_redisConfig);
_db = connection.GetDatabase();
_server = GetServer();

}

private  IServer GetServer()
{
var config = new ConfigurationOptions
{
KeepAlive = 0,
EndPoints = { _redisConfig.EndPoints[0]},
AbortOnConnectFail = false,
AllowAdmin = true
};
var conn = ConnectionMultiplexer.Connect(config);
return conn.GetServer(config.EndPoints[0]);
}

public Task<RedisResult> EvalLua(string lua, IList<RedisKey> keys, IList<RedisValue> values)
{
if (_keyPrefix != null)
keys = keys.Select(p => p.Prepend(_keyPrefix)).ToList();//加上前缀
return _db.ScriptEvaluateAsync(lua, keys.ToArray(), values.ToArray());
}

public async Task<RedisResult> EvalLua(byte[] luaSha1, IList<RedisKey> keys, IList<RedisValue> values)
{
if (_keyPrefix != null)
keys = keys.Select(p => p.Prepend(_keyPrefix)).ToList();//加上前缀
return await _db.ScriptEvaluateAsync(luaSha1, keys.ToArray(), values.ToArray());
}

public async Task<byte[]> LoadLuaToServerAsync(string lua)
{
var sha1 = lua.CalcLuaSha1();//本地计算
if(!await _server.ScriptExistsAsync(sha1))//服务器上不存在
sha1 = await _server.ScriptLoadAsync(lua);//应该和计算的相同

return sha1;
}
}
}


下面是测试代码段

//下面是测试代码
[Test]
public async Task LoadLuaToServerAsync()
{
var key = "TestEvalLua10010";
var fieldContent = "testlua10011";
const string lua =
"redis.call('SET', KEYS[1], ARGV[1])\n" +
"return redis.call('GET', KEYS[1])\n";

var sha2 = lua.CalcLuaSha1();
var sha1 = await _kv.LoadLuaToServerAsync(lua);
Assert.AreEqual(20, sha1.Length);
for (var i = 0; i < 20; i++)
Assert.AreEqual(sha1[i],sha2[i]);

var keys = new List<RedisKey> { key };
var values = new List<RedisValue> { fieldContent };
var result = await _kv.EvalLua(sha1, keys, values);
Assert.AreEqual(fieldContent, (string)result);
}


计算sha1用到的方法

//计算lua的sha1结果,作为执行lua的参数
public  static byte[] CalcLuaSha1(this string lua)
{
SHA1 sha1 = new SHA1CryptoServiceProvider();
var bytesSha1In = Encoding.Default.GetBytes(lua);
return sha1.ComputeHash(bytesSha1In);
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: