• redis的c++ 客户端 redis-plus-plus

    hiredis 多线程不安全，要想多线程使用就得维护一个对象池，使用起来比较麻烦

    //redis.hpp
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
// typedef std::vector CharVector;


class RedisClient {
public:
    RedisClient()
    {
    }
    ~RedisClient()
    {
        for (int i = 0; i < conn_num; ++i) {
            if (conn_pool[i] != NULL) {
                redisFree(conn_pool[i]);
                conn_pool[i] = NULL;
            }
        }
        delete[] conn_pool;
		conn_flag.clear();
        // if (conn_flag != NULL) {
        //     delete[] conn_flag;
        //     conn_flag = NULL;
        // }
    }

    int init(std::string ip_, int port_, int conn_num_)
    {
        ip = ip_;
        port = port_;
        conn_num = conn_num_;

        conn_pool = new redisContext*[conn_num];
        if (conn_pool == NULL) {
            return 1;
        }

        conn_flag.reserve(conn_num) ;
        if (conn_flag.capacity() != conn_num) {
            return 2;
        }

        for (int i = 0; i < conn_num; ++i) {
            conn_pool[i] = redisConnect(ip.c_str(), port);
            if (conn_pool[i] == NULL || conn_pool[i]->err) {
                return 3;
            }

            conn_flag[i] = 0;
        }

        empty_num = conn_num;
        current_conn = 0;

        return 0;
    }
	


    bool isConnOk(redisContext* conn);
    // bool Reconnect();
    // bool Conn(const char* ip, int port = 6379);
    // void Close();
    // int checkError();

    // size_t SetHValue(redisContext* conn,std::string& table, std::string& field, const char* value, size_t len);
    // size_t SetHValue(redisContext* conn,std::string& table, std::string& field, std::string& value);
    // int64_t INCR(redisContext* conn,std::string& table, std::string& field, int count);
    // size_t QueryKeys(redisContext* conn,std::string& table, std::vector& vec);
    // size_t QueryKeys(redisContext* conn,std::string& table, std::map& map);
    // size_t Query(redisContext* conn,std::string& table, std::string& field, std::string& value);
    // size_t Query(std::string & table, std::string & field, CharVector & value);
    // size_t QueryCount(string& table);

    // size_t SetHValue(const char * table, const char *  field, const char * value, size_t len);
    size_t Query(const char* table, const char* field, char** pValue );
    // size_t QueryCount(redisContext* conn,const char* table);
    // int64_t INCR(redisContext* conn,const char* table, const char* field, int count);
    // CharVector buffer;

private:
	void set_conn(int id){
		
		if(conn_pool[id] !=NULL){
			redisFree(conn_pool[id]);
		}
		conn_pool[id] = redisConnect(ip.c_str(), port);
        if(conn_pool[id] == NULL || conn_pool[id]->err)
        {
            std::cout << "connect error!" << std::endl;
            return;  
        }
        std::cout << "reconnect" << std::endl;
	}
    redisContext* get_conn(int &out_id)
    {
        if (empty_num == 0) {
            return NULL;
        }

        mtx.lock();

        while (conn_flag[current_conn] != 0) {
            current_conn = (current_conn + 1) % conn_num;
        }

        conn_flag[current_conn] = 1;
        --empty_num;
        out_id = current_conn;
        current_conn = (current_conn + 1) % conn_num;

        mtx.unlock();

        return conn_pool[out_id];
    }

    void put_conn(int id)
    {
        if (id < conn_num && id >= 0) {
            mtx.lock();

            conn_flag[id] = 0;
            ++empty_num;

            mtx.unlock();
        }

        return;
    }

    // redisContext* conn = nullptr;
    // redisReply* getReply; // used for get

    std::string ip;
    int port;
    int conn_num;

    redisContext** conn_pool;
    std::vector<int> conn_flag;
    int empty_num;
    int current_conn;

    std::mutex mtx;
};

    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
    • 42
    • 43
    • 44
    • 45
    • 46
    • 47
    • 48
    • 49
    • 50
    • 51
    • 52
    • 53
    • 54
    • 55
    • 56
    • 57
    • 58
    • 59
    • 60
    • 61
    • 62
    • 63
    • 64
    • 65
    • 66
    • 67
    • 68
    • 69
    • 70
    • 71
    • 72
    • 73
    • 74
    • 75
    • 76
    • 77
    • 78
    • 79
    • 80
    • 81
    • 82
    • 83
    • 84
    • 85
    • 86
    • 87
    • 88
    • 89
    • 90
    • 91
    • 92
    • 93
    • 94
    • 95
    • 96
    • 97
    • 98
    • 99
    • 100
    • 101
    • 102
    • 103
    • 104
    • 105
    • 106
    • 107
    • 108
    • 109
    • 110
    • 111
    • 112
    • 113
    • 114
    • 115
    • 116
    • 117
    • 118
    • 119
    • 120
    • 121
    • 122
    • 123
    • 124
    • 125
    • 126
    • 127
    • 128
    • 129
    • 130
    • 131
    • 132
    • 133
    • 134
    • 135
    • 136
    • 137
    • 138
    • 139
    • 140
    • 141
    • 142
    • 143
    • 144
    • 145
    • 146
    • 147
    • 148
    • 149
    • 150
    • 151
    //redis.cpp
#include "redis.hpp"
#include 
using namespace robin;
#define MAX_CONN_TIMES 10
#define DEBUG
bool RedisClient::isConnOk(redisContext* conn)
{
    if (conn == nullptr)
        return false;

    // if (checkError() > 0)
    //     return false;

    if (!(conn->flags & REDIS_CONNECTED)) {
        return false;
    }

    bool ret = false;
    redisReply* reply = static_cast<redisReply*>(redisCommand(conn, "PING"));
    if (reply == nullptr)
        return false;

    if (reply->len > 0)
        ret = true;

    freeReplyObject(reply);

    return ret;
}

// bool RedisClient::Reconnect()
// {
//     this->Close();
//     int n = 0;
//     bool ret = Conn(_ip.c_str(), _port);
//     while (ret == false && n < MAX_CONN_TIMES) {
//         ret = Conn(_ip.c_str(), _port);

//         n++;
//     }

//     return ret;
// }

// bool RedisClient::Conn(const char* ip, int port)
// {
//     this->_ip = ip;
//     this->_port = port;
//     if (conn != nullptr)
//         return 0;

//     conn = redisConnect(ip, port);

//     // struct timeval tm;
//     // tm.tv_sec = 2;
//     // tm.tv_sec = 0;
//     // conn = redisConnectWithTimeout(ip, port, tm);
//     if (conn->err) {
//         printf("redis connection error:%s\n", conn->errstr);
//         return false;
//     }

//     return true;
// }

// size_t RedisClient::SetHValue(redisContext* conn,std::string& table, std::string& field, const char* value, size_t len)
// {
//     redisReply* reply = static_cast(
//         redisCommand(conn, "HSET  %b %b %b", table.c_str(), table.size(),
//             field.c_str(),
//             field.size(),
//             value, len));
//     if (reply == nullptr)
//         return 0;

//     size_t ret = 0;
//     if (reply->type == REDIS_REPLY_INTEGER) {
//         ret = reply->integer;
//     }
//     freeReplyObject(reply);
//     return ret;
// }

// size_t RedisClient::SetHValue(redisContext* conn,std::string& table, std::string& field, std::string& value)
// {
//     if (conn == nullptr)
//         return -1;

//     std::string cmd;

//     std::stringstream stream;
//     stream << "HSET " << table << " " << field << " " << value;
//     cmd = stream.str();
//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));
//     if (reply == nullptr)
//         return 0;

//     size_t ret = 0;
//     if (reply->type == REDIS_REPLY_INTEGER) {
//         ret = reply->integer;
//     }
//     freeReplyObject(reply);
//     return ret;
// }

// int64_t RedisClient::INCR(redisContext* conn,const char* table, const char* field, int count)
// {
//     if (conn == nullptr)
//         return -1;

//     std::string cmd = "";
//     std::stringstream stream;

//     stream << "hincrby " << table << " " << field << " " << count;

//     cmd = stream.str();
//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));
//     if (reply == nullptr)
//         return 0;

//     int64_t ret = reply->integer;
//     freeReplyObject(reply);

//     return ret;
// }

// int64_t RedisClient::INCR(redisContext* conn,std::string& table, std::string& field, int count)
// {
//     return INCR(conn,table.c_str(), field.c_str(), count);
// }
// size_t RedisClient::QueryKeys(redisContext* conn,std::string& table, std::map& map)
// {
//     std::string cmd = "";
//     std::stringstream stream;
//     stream << " HKEYS " << table;
//     cmd = stream.str();

//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));
//     size_t len = 0;
//     if (reply == nullptr)
//         return 0;

//     for (size_t i = 0; i < reply->elements; i++) {
//         redisReply* item = *(reply->element + i);
//         uint32_t id = atoi(item->str);
//         map.insert(std::pair(id, item->str));
//         if (id > len)
//             len = id;
//     }

//     freeReplyObject(reply);
//     return len;
// }

// size_t RedisClient::QueryKeys(redisContext* conn,std::string& table, std::vector& vec)
// {
//     std::string cmd = "";
//     std::stringstream stream;
//     stream << " HKEYS " << table;
//     cmd = stream.str();

//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));
//     if (reply == nullptr)
//         return 0;

//     size_t len = reply->elements;
//     for (size_t i = 0; i < reply->elements; i++) {
//         redisReply* item = *(reply->element + i);
//         vec.push_back(item->str);
//         // item->str;
//     }

//     freeReplyObject(reply);
//     return len;
// }

// size_t RedisClient::QueryCount(redisContext* conn,const char* table)
// {
//     std::string cmd = "";
//     std::stringstream stream;
//     stream << " HLEN " << table;
//     cmd = stream.str();

//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));

//     if (reply == nullptr)
//         return 0;

//     //_atoi64(char *)
//     // char *end;
//     // strtoll(reply->str, &end, 10);
//     size_t len = reply->integer;

//     freeReplyObject(reply);
//     return len;
// }
// size_t RedisClient::QueryCount(string& table)
// {
//     return QueryCount(table.c_str());
// }

// size_t RedisClient::Query(redisContext* conn,std::string& table, std::string& field, std::string& value)
// {
//     std::string cmd = "";
//     std::stringstream stream;
//     stream << " HGET " << table << " " << field;
//     cmd = stream.str();
//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));

//     if (reply == nullptr)
//         return 0;

//     size_t len = reply->len;
//     value = reply->str;

//     freeReplyObject(reply);
//     return len;
// }
// size_t RedisClient::Query(std::string& table, std::string& field, CharVector& value)
// {
//     std::string cmd = "";
//     std::stringstream stream;
//     stream << " HGET " << table << " " << field;
//     cmd = stream.str();
//     redisReply* reply = static_cast(redisCommand(conn, cmd.c_str()));
//     if (reply == nullptr)
//         return 0;

//     size_t len = reply->len;
//     value.reserve(len);
//     value.append(reply->str, len);

//     freeReplyObject(reply);
//     return len;
// }

size_t RedisClient::Query(const char* table, const char* field, char** pValue)
{
    int current_conn = 0;
    redisContext* conn = get_conn(current_conn);
    #ifdef DEBUG
    bool res = isConnOk(conn);
    cout<<res<<endl;
    if(!res){
        set_conn(current_conn);
    }
    #endif
    std::string cmd = "";
    std::stringstream stream;
    stream << " HGET " << table << " " << field;
    cmd = stream.str();

    // if (getReply)
    //     freeReplyObject(getReply);

    redisReply* getReply = static_cast<redisReply*>(redisCommand(conn, cmd.c_str()));
    if (getReply == nullptr)
        return 0;
    return 0;

    *pValue = getReply->str;
    size_t len = getReply->len;
    put_conn(current_conn);
    return len;
}

// void RedisClient::Close()
// {
//     if (getReply != nullptr) {
//         freeReplyObject(getReply);
//         getReply = nullptr;
//     }
//     if (conn != nullptr) {
//         redisFree(conn);
//         conn = nullptr;
//     }
// }

// int RedisClient::checkError()
// {
//     if (conn->err == REDIS_ERR_IO || conn->err == REDIS_ERR_EOF) {
//         return 1;
//     }

//     return 0;
// }



    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
    • 42
    • 43
    • 44
    • 45
    • 46
    • 47
    • 48
    • 49
    • 50
    • 51
    • 52
    • 53
    • 54
    • 55
    • 56
    • 57
    • 58
    • 59
    • 60
    • 61
    • 62
    • 63
    • 64
    • 65
    • 66
    • 67
    • 68
    • 69
    • 70
    • 71
    • 72
    • 73
    • 74
    • 75
    • 76
    • 77
    • 78
    • 79
    • 80
    • 81
    • 82
    • 83
    • 84
    • 85
    • 86
    • 87
    • 88
    • 89
    • 90
    • 91
    • 92
    • 93
    • 94
    • 95
    • 96
    • 97
    • 98
    • 99
    • 100
    • 101
    • 102
    • 103
    • 104
    • 105
    • 106
    • 107
    • 108
    • 109
    • 110
    • 111
    • 112
    • 113
    • 114
    • 115
    • 116
    • 117
    • 118
    • 119
    • 120
    • 121
    • 122
    • 123
    • 124
    • 125
    • 126
    • 127
    • 128
    • 129
    • 130
    • 131
    • 132
    • 133
    • 134
    • 135
    • 136
    • 137
    • 138
    • 139
    • 140
    • 141
    • 142
    • 143
    • 144
    • 145
    • 146
    • 147
    • 148
    • 149
    • 150
    • 151
    • 152
    • 153
    • 154
    • 155
    • 156
    • 157
    • 158
    • 159
    • 160
    • 161
    • 162
    • 163
    • 164
    • 165
    • 166
    • 167
    • 168
    • 169
    • 170
    • 171
    • 172
    • 173
    • 174
    • 175
    • 176
    • 177
    • 178
    • 179
    • 180
    • 181
    • 182
    • 183
    • 184
    • 185
    • 186
    • 187
    • 188
    • 189
    • 190
    • 191
    • 192
    • 193
    • 194
    • 195
    • 196
    • 197
    • 198
    • 199
    • 200
    • 201
    • 202
    • 203
    • 204
    • 205
    • 206
    • 207
    • 208
    • 209
    • 210
    • 211
    • 212
    • 213
    • 214
    • 215
    • 216
    • 217
    • 218
    • 219
    • 220
    • 221
    • 222
    • 223
    • 224
    • 225
    • 226
    • 227
    • 228
    • 229
    • 230
    • 231
    • 232
    • 233
    • 234
    • 235
    • 236
    • 237
    • 238
    • 239
    • 240
    • 241
    • 242
    • 243
    • 244
    • 245
    • 246
    • 247
    • 248
    • 249
    • 250
    • 251
    • 252
    • 253
    • 254
    • 255
    • 256
    • 257
    • 258
    • 259
    • 260
    • 261
    • 262
    • 263
    • 264
    • 265
    • 266
    • 267
    • 268
    • 269
    • 270
    • 271
    • 272
    • 273
    • 274
    • 275
    • 276
    • 277
    • 278
    • 279
    • 280
    • 281
    • 282
    • 283
    • 284
    • 285
    • 286
    • 287
    • 288
    • 289

    使用redis-plus-plus来操作redis更加的分布，redis类多线程安全，内部维护了一个连接池
    demo

    
#include 
#include 
#include 
#include 
#include 

extern "C" {
#include  // for getopt on non-Windows platform
}

std::string host = "127.0.0.1";
int port = 6379;
std::string auth;
std::string cluster_node;
int cluster_port = 0;
bool benchmark = false;
int resp = 3;

int main(){
    sw::redis::Optional<sw::redis::ConnectionOptions> opts;
    if (!host.empty() && port > 0) {
        sw::redis::ConnectionOptions tmp;
        tmp.host = host;
        tmp.port = port;
        tmp.password = auth;
        tmp.resp = resp;

        opts = sw::redis::Optional<sw::redis::ConnectionOptions>(tmp);
    }
    if(opts.has_value()){
        sw::redis::Redis redis(*opts);
        redis.set("hello", "world");
        std::string value;
        auto stringopts = redis.get("hello2");
        if(stringopts.has_value()){
            value = *stringopts;
        }
        std::cout<<value<<std::endl;
    }
}

    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
  • 相关阅读:
    【计算机网络笔记】OSI参考模型中端-端层（传输层、会话层、表示层、应用层）功能介绍
    算法总篇章
    Java中可以用的大数据推荐算法
    demo(一)eureka----服务注册与提供
    Qt--无边框窗口完美(FrameLess)实现,包含缩放和移动功能重写。
    Java面试题：线程的run()和start()有什么区别？
    《爵士乐史》乔德.泰亚 笔记
    解决Netty那些事儿之Reactor在Netty中的实现(创建篇)-上
    视频封面：从视频中提取封面，轻松制作吸引人的视频
    软考高级软件架构师论文——论软件架构评估
  • 原文地址：https://blog.csdn.net/weixin_43862733/article/details/133880233