ICode9

精准搜索请尝试: 精确搜索
首页 > 数据库> 文章详细

c++ 语言redis分布式锁的实现

2022-01-09 14:34:50  阅读:249  来源: 互联网

标签:int redis Redis ret tv c++ include 分布式


main.cpp

#include "redis.h"
#include <unistd.h>
int main()
{
	Redis *r = new Redis();
    r->connectRedis("127.0.0.1",6379,5);
    while (r->tryLock(40) != 0)
    {
        /* code */
        sleep(10);
    }
	return 0;
}

redis.h

#ifndef _REDIS_H_
#define _REDIS_H_

#include <iostream>
#include <string.h>
#include <string>
#include <stdio.h>

#include <hiredis/hiredis.h>

class Redis
{
public:
    Redis();
    ~Redis();
    int connectRedis(std::string addr, int port, int time_out);
    int tryLock(int duration);
    redisReply *ExecuteCmd(const char *cmd, char *data = NULL, int dataSize = 0);
private:
    redisContext *_connect;
    redisReply *_reply;

private:
    int m_timeOutConnect;
    struct timeval m_tv; //redis  命令超时时间
    redisContext *m_redisContent;
    std::string m_redisIp;
    int m_redisPort;
};

#endif //_REDIS_H_

redis.cpp

#include "redis.h"

Redis::Redis() {}

Redis::~Redis()
{
    this->_connect = NULL;
    this->_reply = NULL;
}

int Redis::connectRedis(std::string addr, int port, int time_out)
{
    m_redisIp = addr;
    m_redisPort = port;

    int ret = -1;
    //struct timeval tv;
    m_tv.tv_sec = time_out;
    m_tv.tv_usec = 0;
    m_timeOutConnect = time_out;

    m_redisContent = redisConnectWithTimeout(m_redisIp.c_str(), m_redisPort, m_tv);
    if (m_redisContent == NULL)
    {
        ret = -1;
        return ret;
    }
    else
    {
        if (m_redisContent->err != 0)
        {
            std::cout << "connect redis ip:" << m_redisIp << ",port:" << m_redisPort << ",error:" << m_redisContent->errstr;
            ret = -1;
            redisFree(m_redisContent);
            m_redisContent = NULL;
            return ret;
        }
    }
    // ret = (int)authRedis();
    redisSetTimeout(m_redisContent, m_tv);
    redisEnableKeepAlive(m_redisContent);
}

int Redis::tryLock(int duration)
{
    int ret = -1;
    char cmdBuf[256] = {0};

    sprintf(cmdBuf, "set %s %s ex %d nx", "gbdlock", "0.0.0.0", duration); //ex 设置键的过期时间为 second 秒。NX :只在键不存在时,才对键进行设置操作。

    redisReply *reply = ExecuteCmd(cmdBuf);
    if (reply && reply->type != REDIS_REPLY_NIL && strcmp(reply->str, "OK") == 0)
    {
        ret = 0;
        std::cout << "redis set lock OK!!!"
                  << "0.0.0.0" << std::endl;
    }
    else
    {
        char cmdBuf1[256] = {0};

        /* get lock */
        sprintf(cmdBuf1, "GET %s", "gbdlock");
        reply = ExecuteCmd(cmdBuf1);
        if (reply && reply->type != REDIS_REPLY_NIL)
        {
            if (reply->str && !strcmp("gbdlock", reply->str))
            {
                ret = 0;
                std::cout << "Get lock succ: integer = " << reply->integer << " str = " << reply->str;
            }
        }
        std::cout << "redis set lock fail!!!"
                  << "gbdlock" << std::endl;
    }

    if (reply != NULL)
    {
        freeReplyObject(reply);
    }

EXIT:
    return ret;
}

redisReply *Redis::ExecuteCmd(const char *cmd, char *data, int dataSize)
{
    if (m_redisContent == NULL)
    {
        std::cout << "Redis: m_redisContent == NULL error";
        return NULL;
    }

    redisReply *reply = NULL;
    if (data == NULL)
    {
        reply = (redisReply *)redisCommand(m_redisContent, cmd);
    }
    else
    {
        if (dataSize > 2 * 1024 * 1024 /*一个二进制redis数据的最大长度*/)
        {
            std::cout << "redis dataSize more then MAX_REDIS_BIN_LEN ";
            return NULL;
        }
        reply = (redisReply *)redisCommand(m_redisContent, cmd, data, dataSize);
        //redisReply *reply = (redisReply *)redisCommand(m_redisContent, "RPUSH %s %b", listName, data, dataSize);
    }
    return reply;
}

makefile

redis: main.cpp redis.cpp
	g++ main.cpp redis.cpp -o test -L /usr/local/lib/ -lhiredis
clean:
	rm redis

标签:int,redis,Redis,ret,tv,c++,include,分布式
来源: https://blog.csdn.net/jenie/article/details/122393282

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有