使用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的数据。

发表评论

电子邮件地址不会被公开。 必填项已用*标注