SecurePlay/SecurePlayAuxServer/communication/FromRedis.cpp

#include "FromRedis.h"
#include "spdlog/spdlog.h"

FromRedis::FromRedis()
{

}


FromRedis::~FromRedis()
{

}

/* 设置IP和端口 */
void FromRedis::setRedisIPAndPort(const std::string& ip, int port)
{
    m_redisIP = ip;
    m_redisPort = port;
}

/* 设置密码 */
void FromRedis::setRedisPassword(const std::string& password)
{
    m_redisPassword = password;
}

/* 连接Redis */
bool FromRedis::connectRedis()
{
    SPDLOG_DEBUG("Redis IP: {}, Port: {}", m_redisIP, m_redisPort);
    SPDLOG_DEBUG("Redis Password: {}", m_redisPassword);
    if(m_redisContext != nullptr)
    {
        redisFree(m_redisContext);
        m_redisContext = nullptr;
    }
    if(m_redisIP.empty())
    {
        SPDLOG_ERROR( "Redis IP is empty");
        return false;
    }

    /* 连接Redis */
    m_redisContext = redisConnect(m_redisIP.c_str(), m_redisPort);
    if( (m_redisContext == nullptr) || ( m_redisContext->err != 0) )
    {
        SPDLOG_ERROR( "Connect redis failed:{}", m_redisContext->errstr);
        return false;
    }
    SPDLOG_INFO("Connect redis success");
    /* 登录认证 */
    if(!m_redisPassword.empty())
    {
        if(!authRedis(m_redisPassword))
        {
            SPDLOG_ERROR( "Auth redis failed");
            return false;
        }
        // SPDLOG_INFO( "Auth redis success");
    }
    return true;
}

/* 断开Redis连接 */
void FromRedis::disConnectRedis()
{
    if(m_redisContext != nullptr)
    {
        redisFree(m_redisContext);
        m_redisContext = nullptr;
    }
}



/* 获取redis中的String数据 */
bool FromRedis::getRedisString(const std::string& key, std::string& value)
{
    if(m_redisContext == nullptr)
    {
        SPDLOG_ERROR( "Redis no connect");
        return false;
    }
    value.clear();
    /* 获取数据 */
    auto reply = (redisReply*)redisCommand(m_redisContext, "GET %s", key.c_str());
    // auto reply = (redisReply*)redisCommand(m_redisContext, "GET 113:P100104000");
    if(reply == nullptr)
    {
        SPDLOG_ERROR( "Get redis value data failed");
        return false;
    }
    if(reply->type != REDIS_REPLY_STRING)
    {
        if(reply->type == REDIS_REPLY_NIL)
        {
            SPDLOG_WARN( "获取Redis数据失败，Key对应的值不存在，Key: {}", key);
        }else {
            /* 其他类型的返回值 */
            SPDLOG_ERROR( "Get redis data failed, type is not string, type: {}", reply->type);
        }
        freeReplyObject(reply);
        return false;
    }
    value = reply->str;
    freeReplyObject(reply);
    return true;
}


/* 登录认证Redis，这个需要在连接成功后调用 */
bool FromRedis::authRedis(const std::string& password)
{
    if(m_redisContext == nullptr)
    {
        SPDLOG_ERROR( "Redis no connect");
        return false;
    }
    auto reply = (redisReply*)redisCommand(m_redisContext, "AUTH %s", password.c_str());
    if(reply == nullptr)
    {
        SPDLOG_ERROR( "Redis Auth failed");
        return false;
    }
    /* 判断返回值 */
    if( reply->type == REDIS_REPLY_STATUS && strcasecmp(reply->str, "OK") == 0)
    {
        SPDLOG_INFO( "Redis Auth success");
        freeReplyObject(reply);
        return true;
    }
    else
    {
        SPDLOG_ERROR( "Redis Auth failed:{}", reply->str);
        freeReplyObject(reply);
        return false;
    }
}