leetcode_cpp/redis/redis.h

/*
 * @Description:
 * @Version: 1.0
 * @Autor: zhuyijun
 * @Date: 2021-11-17 13:05:50
 * @LastEditTime: 2021-11-17 13:05:50
 */
#ifndef _REDIS_H_
#define _REDIS_H_

#include <hiredis/hiredis.h>
#include <stdio.h>
#include <string.h>

#include <iostream>
#include <string>

class Redis {
 public:
  Redis() {}

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

  bool connect(std::string host, int port) {
    this->_connect = redisConnect(host.c_str(), port);
    if (this->_connect != NULL && this->_connect->err) {
      printf("connect error: %s\n", this->_connect->errstr);
      return 0;
    }
    return 1;
  }

  std::string get(std::string key) {
    this->_reply =
        (redisReply*)redisCommand(this->_connect, "GET %s", key.c_str());
    std::string str = this->_reply->str;
    freeReplyObject(this->_reply);
    return str;
  }

  void set(std::string key, std::string value) {
    redisCommand(this->_connect, "SET %s %s", key.c_str(), value.c_str());
  }

 private:
  redisContext* _connect;
  redisReply* _reply;
};

#endif  //_REDIS_H_