使用redis来做计数器完善投票系统

首先看应用场景：几年前，微信里许多投票系统。很多人都找好友或朋友圈找人拉票。当时，有一个比较大的乐园也做了这个投票活动，好像是超过一定票数就可以免费玩。我也被好友要求给他投票，一天好像可以投3票还是5票，具体的记不太清了。我当时进去该系统后，给朋友点了投票按钮，发现没有任何反应。我以为是我网络的问题，后来发现是程序的问题。我当时猜测，它应该是投票的人太多了，系统一时没抗住。

因为当时，我刚好在学redis。所以，我想可以通过redis的计数器来改善该功能。

这里，我们只统计用户总的投票数，而不考虑用户具体的投票情况，像A什么时候给B投票的记录我们不做考虑。这种情况如果考虑的话，就要使用其他的解决方案了（可以用消息队列）。

校验

首先，要对每个用户的投票数进行校验。用户每天都有3次投票机会，所以，我们可建立个键，将他的有效期设为明天凌晨。用户每投一票就加1等到了3票后，就提示今日次数已用完。

// 投票次数校验
function checkVote($uid)
{
    $key = "uid:$uid:cnt";
    $tomrTime = mktime(0,0,0, date('m'), date('d')+1, date('Y'));
    
    if (!$redis->exsits($key)) {
        $redis->set($key, 0, ['ex' => $tomrTime - time()]);
        return true;
    } else if ( ($cnt = $redis->get($key)) < 3 ) {
        return true;
    } else {
        return false;
    }
}

统计投票

次数校验通过后，就需要统计用户的得票数了。

// 得票数计数
// uid表示投票人id
// touid表示得票人id
function vote ($uid, $toUid)
{
    $key = "uid:$toUid:vote";
    
    // 投票数校验
    if (checkVote($uid)) {
        // 统计投票数及参选人得票数
        $redis->incr($key);
        $redis->incr("uid:$uid:cnt");
        
        return true;
    } else {
        return false;
    }
}

我自己的阿里云主机配置只有1核cpu1G内存的配置，每秒incr性能能达到10万多次，redis性能真恐怖。

# redis-benchmark -t incr -q
INCR: 105708.25 requests per second

Mysql异步更新用户得票数

最后，我们只需要做一个定时任务了，让mysql每隔一定时间就去同步用户的得票数。

方法是做一个死循环，每次循环都获取50个用户得票数，直到遍历完所有用户后，就退出循环，结束脚本。

$start = 0;
while (true) {
    // 每次循环取50个用户id
    $users = $DB->query("SELECT uid,votes FROM users LIMIT $start, 50");
    if (!$users) {
        exit();
    }
    $keys = [];
    foreach ($users as $userinfo) {
        $keys[] = "uid:{$userinfo['uid']}:vote";
    }
    $votes = $redis->mget($keys);
    foreach ($votes as $index => $vote) {
        if ($vote != $users[$index]['votes']) {
            $DB->query("UPDATE users SET votes = '$vote' WHERE uid='{$users[$index]['uid']}");
        }
    }
    $start += 50;
}

Redis的mget命令的是一次获取多个key的值。获取了redis里存放的redis得票数后，要先和Mysql里的得票数做比对。不一样的时候才去更新Mysql的数据。

以上就是使用redis来做计数器完善投票系统的详细内容，更多请关注其它相关文章！