用户金额的终极解决方案--Redis Lua

最新推荐文章于 2023-03-06 09:52:16 发布

weixin_34124577

最新推荐文章于 2023-03-06 09:52:16 发布

阅读量505

点赞数

文章标签：数据库 lua python

原文链接：https://my.oschina.net/u/3768341/blog/2991501

版权

2019独角兽企业重金招聘Python工程师标准>>>

我们在开发各种订单系统的时候都会碰到一个必须面对的问题，用户金额的最终结算。一般我们会把金额存在数据库中，采用行级锁的方式来对金额的变动来进行处理，但是在高并发下，对于结算时间有要求的情况下，这种方式是极为低效的。

所以我们打算将用户的金额放在Redis中进行处理，无论用户的查询，充值，付账，开奖增加都会有一个高效的返回。但是我们遇到的问题依然是是否要用锁的问题。因为Redis是单线程的，在多个动作同时执行的时候，要保证我们代码执行到每一步，他的金额都不会错乱，比如不会小于0的情况下依然可以付账成功。如果使用分布式锁来锁定用户每次操作，那么大量的用户进行操作时对性能的影响是非常大的，这个时候我们的主角登场了——Redis Lua，它可以将多条Redis命令合成一条原子操作命令，避免在判断金额大小的时候，其他线程修改了金额，在非加锁的情况下后续的操作却不知道的尴尬。当然最后我们可以通过异步的方式将数据更新到数据库中。再存储每一条金额的变动流水日志，以方便对账和校准。

具体实现为

在接口中添加两个方法

/**
 * 支付金额
 * @param key
 * @param amount
 * @return
 */
Object payMonery(String key,String amount);

/**
 * 增加金额
 * @param key
 * @param amount
 * @return
 */
Object addMonery(String key,String amount);

集群实现类方法

@Override
public Object payMonery(String key, String amount) {
    String script = "if redis.call('exists',KEYS[1]) == 1 then if (redis.call('get',KEYS[1]) + ARGV[1]) >= 0 then" +
            " redis.call('incrbyfloat',KEYS[1],ARGV[1]) return redis.call('set',KEYS[1],string.format('%0.2f',redis.call('get',KEYS[1])))" +
            "  else return 0 end else return 0  end";
    return jedisCluster.eval(script,1,key,"-" + amount);
}

@Override
public Object addMonery(String key, String amount) {
    String script = "if redis.call('exists',KEYS[1]) == 1 then  redis.call('incrbyfloat',KEYS[1],ARGV[1])" +
            " return redis.call('set',KEYS[1],string.format('%0.2f',redis.call('get',KEYS[1]))) " +
            " else return 0  end";
    return jedisCluster.eval(script,1,key, amount);
}

单机实现类方法

@Override
public Object payMonery(String key, String amount) {
    String script = "if redis.call('exists',KEYS[1]) == 1 then if (redis.call('get',KEYS[1]) + ARGV[1]) >= 0 then" +
            " redis.call('incrbyfloat',KEYS[1],ARGV[1]) return redis.call('set',KEYS[1],string.format('%0.2f',redis.call('get',KEYS[1])))" +
            "  else return 0 end else return 0  end";
    return execute((jedis -> jedis.eval(script,1,key,"-" + amount)));
}

@Override
public Object addMonery(String key, String amount) {
    String script = "if redis.call('exists',KEYS[1]) == 1 then  redis.call('incrbyfloat',KEYS[1],ARGV[1])" +
            " return redis.call('set',KEYS[1],string.format('%0.2f',redis.call('get',KEYS[1]))) " +
            " else return 0  end";
    return execute((jedis -> jedis.eval(script,1,key, amount)));
}

转载于:https://my.oschina.net/u/3768341/blog/2991501