C++訪問Redis的mset 二進制數(shù)據(jù)接口封裝方案

來源：本站原創(chuàng)|時間：2020-01-10|欄目：C語言|點擊：次

需求

C++中使用hiredis客戶端接口訪問redis；
需要使用mset一次設(shè)置多個二進制數(shù)據(jù)

以下給出三種封裝實現(xiàn)方案；

簡單拼接方案

在redis-cli中，mset的語法是這樣的：

復(fù)制代碼代碼如下:

/opt/colin$./redis-cli mset a 11 b 22 c 333

按照這樣的語法拼接后，直接使用hiredis字符串接口redisCommand傳遞：

void msetNotBinary(redisContext *c, const vector<string> &vtKey, const vector<string> & vtVal )
{
  if(vtKey.size() != vtVal.size())
  {
    throw runtime_error( "Redis error" );
  }

  string strCmd = "MSET";
  for(int i = 0; i < vtKey.size(); i++)
  {
    strCmd += " "+vtKey[i]+" "+vtVal[i];
  }
  cout << "strCmd:" << strCmd << endl;

  void * r = redisCommand(c, strCmd.c_str() );
  if ( !r )
    throw runtime_error( "Redis error" );
  freeReplyObject( r );
}

void do_test( redisContext *c )
{  
  vector<string> vtKey;
  vector<string> vtVal;

  vtKey.push_back("A");
  vtVal.push_back("AAAA");
  vtKey.push_back("B");
  vtVal.push_back("BBBB");
  vtKey.push_back("C");
  vtVal.push_back("CCCC");
  //add a binary data
  vtKey.push_back("D");
  vtVal.push_back("");
  char a[] = "ABCDE";
  a[2] = 0;
  vtVal[3].assign(a,5);

  try
  {
    msetNotBinary(c, vtKey, vtVal );
    //mset1( c, vtKey, vtVal );
    //mset2( c, vtKey, vtVal );
  }
  catch ( runtime_error & )
  {
    cout << "Error" << endl;
  }
}

int main(int argc, char *argv[])
{
  redisContext *c;

  c = redisConnect("127.0.0.1",6379);
  if (c->err)
   {
    cout << "Connection error: " << c->errstr << endl;
    return -1;
  }

  do_test(c);

  redisFree(c);

  return 0;
}

這種方式可以處理mset多個字符串?dāng)?shù)據(jù)，但對于數(shù)據(jù)內(nèi)容為二進制數(shù)據(jù)的無能為力；

redisCommandArgv接口傳遞方案

對于多個參數(shù)傳遞，hiredis提供了以下接口，這個接口中最后一個參數(shù)是所有的傳入數(shù)據(jù)的內(nèi)容長度，
就是說這個接口是二進制安全的：

void *redisCommandArgv(redisContext *c, int argc, const char **argv, const size_t *argvlen);
主要工作就是構(gòu)造一個動態(tài)的二維數(shù)組char ** argv,其中涉及到char **到const char **的轉(zhuǎn)換，有一定的風(fēng)險，
關(guān)于這一點前一篇文章已經(jīng)談到；

void mset1( redisContext *c, const vector<string> &vtKey, const vector<string> & vtVal )
{
  if(vtKey.size() != vtVal.size())
  {
    throw runtime_error( "Redis error" );
  }

  char ** argv = new char*[vtKey.size() + vtVal.size() + 1 ];
  size_t * argvlen = new size_t[vtKey.size() + vtVal.size() + 1 ];

  int j = 0;
  argv[j] = new char[5];
  memcpy(argv[j],"MSET",4);
  argvlen[j] = 4;
  ++j;


  for(int i = 0 ; i < vtKey.size();i++)
  {  
    argvlen[j] = vtKey[i].length();
    argv[j] = new char[argvlen[j]];
     memset((void*)argv[j],0,argvlen[j] );
    memcpy((void*)argv[j],vtKey[i].data(),vtKey[i].length());
    j++;

    argvlen[j] = vtVal[i].length();
    argv[j] = new char[argvlen[j]];
    memset((void*)argv[j],0,argvlen[j]);
    memcpy((void*)argv[j],vtVal[i].data(),vtVal[i].length());
    j++;
  }

  //if not use const_cast<const char**> ,compile error
  //for why assign from char** to const char** error, see my blog ...
   void *r = redisCommandArgv(c, vtKey.size() + vtVal.size() + 1, const_cast<const char**>(argv), argvlen );
  if ( !r )
    throw runtime_error( "Redis error" );
  freeReplyObject( r );

  for(int i = 0;i < vtKey.size();i++)
  {
    delete [] argv[i];
    argv[i] = NULL;
  }

  delete []argv;
  delete []argvlen;
  argv = NULL;
}

redisCommandArgv接口傳遞的Vector方案

還是使用redisCommandArgv接口，使用vector來構(gòu)造這個const char **，這個方法是從參考資料1中學(xué)到的：

void mset2( redisContext *c, const vector<string> &vtKey, const vector<string> & vtVal)
{
  if(vtKey.size() != vtVal.size())
  {
    throw runtime_error( "Redis error" );
  }

  vector<const char *> argv( vtKey.size() + vtVal.size() + 1 );
  vector<size_t> argvlen( vtKey.size() + vtVal.size() + 1 );
  int j = 0;

  static char msetcmd[] = "MSET";
  argv[j] = msetcmd;
  argvlen[j] = sizeof(msetcmd)-1;
  ++j;

  for(int i = 0;i< vtKey.size();++i)
  {
    argvlen[j] = vtKey[i].length();
    argv[j] = new char[argvlen[j]];
     memset((void*)argv[j],0,argvlen[j] );
    memcpy((void*)argv[j],vtKey[i].data(),vtKey[i].length());
    j++;

    argvlen[j] = vtVal[i].length();
    argv[j] = new char[argvlen[j]];
    memset((void*)argv[j],0,argvlen[j]);
    memcpy((void*)argv[j],vtVal[i].data(),vtVal[i].length());
    j++;
  }

  void *r = redisCommandArgv(c, argv.size(), &(argv[0]), &(argvlen[0]) );
  if ( !r )
    throw runtime_error( "Redis error" );
  freeReplyObject( r );
}

這樣，就實現(xiàn)二進制數(shù)據(jù)的傳遞；

二進制校驗

程序執(zhí)行后，可以用redis-cli來驗證：

對于非二進制安全的實現(xiàn)，二進制內(nèi)容是截斷的：

復(fù)制代碼代碼如下:

/opt/app/colin$./redis-cli get D

"AB"

而二進制安全的實現(xiàn)接口，二進制數(shù)據(jù)的0通過轉(zhuǎn)義方式顯示：

復(fù)制代碼代碼如下:

/opt/app/colin$./redis-cli get D

"AB\x00DE"

完整可執(zhí)行的代碼詳見github：https://github.com/me115/cppset/tree/master/2DimArray

以上所述就是本文的全部內(nèi)容了，希望大家能夠喜歡。

上一篇：C++ Custom Control控件向父窗體發(fā)送對應(yīng)的消息

欄目：C語言

下一篇：C++ 關(guān)于MFC多線程編程的注意事項

本文標(biāo)題：C++訪問Redis的mset 二進制數(shù)據(jù)接口封裝方案

本文地址：http://www.jygsgssxh.com/a1/Cyuyan/3014.html

更多C語言

雷火电竞-中国电竞赛事及体育赛事平台

C語言

C++訪問Redis的mset 二進制數(shù)據(jù)接口封裝方案

您可能感興趣的文章

閱讀排行

本欄相關(guān)

隨機閱讀