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

算法之名

发布于 2019-08-20 09:59:01

1.2K0

发布于 2019-08-20 09:59:01

文章被收录于专栏：算法之名

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

所以我们打算将用户的金额放在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)));
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

云数据库 Redis