thrift编写服务器/客户端实现图片传输(c++)

本文主要实现 在linux环境下thrift编写服务器/客户端实现图片传输(c++)

代码块

thrift/send_photo.thrift

service Photo {
bool SendPhoto(1: binary write_buffer, 2: i32 buffer_size)
}

命令行 通过thrift生成 cpp文件

# thrift -r --gen cpp send_photo.thrift

生成 gen-cpp文件下包括：

send_photo_constants.cpp
send_photo_constants.h
send_photo_types.cpp
send_photo_types.h
Photo.cpp
Photo.h
Photo_server.skeleton.cpp

将thrift生成的服务器修改成自己的服务器

# mv Photo_server.skeleton.cpp server.cpp

修改server.cpp，只要修改SendPhoto()函数

bool SendPhoto(const std::string& write_buffer, const int32_t buffer_size) {
// Your implementation goes here
printf("SendPhoto\n");

FILE * f = fopen("server.jpg","wb+");
if(f){
fwrite(write_buffer.c_str(),1,buffer_size,f);
fclose(f);
for(unsigned int i = 0; i < 30;i++)
{
printf("%02X ",(unsigned char)write_buffer[i]);
if (i%16 == 15)
{
printf("\n");
}
}
fseek(f,0,SEEK_END);
int size1 = ftell(f);
cout << endl << size1 <<endl;
return true;
}

编写client.cpp

#include "Photo.h"
#include <transport/TSocket.h>
#include <transport/TBufferTransports.h>
#include <protocol/TBinaryProtocol.h>
#include <iostream>
#include <fstream>
#include <string.h>

using namespace std;
using namespace apache::thrift;
using namespace apache::thrift::protocol;
using namespace apache::thrift::transport;

using boost::shared_ptr;

int main(int argc, char **argv){
boost::shared_ptr<TSocket> socket(new TSocket("localhost",9090));
boost::shared_ptr<TTransport> transport(new TBufferedTransport(socket));
boost::shared_ptr<TProtocol> protocol(new TBinaryProtocol(transport));

transport->open();
PhotoClient client(protocol);

char* photo = new char[20];
cout << "input the photo:" << endl;
photo = (char*)malloc(sizeof(char)*20);
scanf("%s",photo);

FILE * f = fopen(photo,"rb");
fseek(f,0,SEEK_END);
int size = ftell(f);
char * buf = new char[size];
memset(buf,0,size);
fseek(f,0,SEEK_SET);
int nRead=fread(buf,sizeof(char),size,f);
bool isRight = client.SendPhoto(string(buf,size),size);
cout << isRight << endl;

transport->close();
return 0;
}

编写Makefile

BOOST_DIR = /usr/local/boost/include/boost/
THRIFT_DIR = /usr/local/include/thrift
LIB_DIR = /usr/local/lib

GEN_SRC = captcha_constants.cpp captcha_types.cpp Photo.cpp

default: server client

server: server.cpp
g++ -DHAVE_NETINET_IN_H -o server -I${THRIFT_DIR} -I${BOOST_DIR}  -I../gen-cpp -L${LIB_DIR} -lthrift server.cpp ${GEN_SRC}

client: client.cpp
g++ -DHAVE_NETINET_IN_H -o client -I${THRIFT_DIR} -I${BOOST_DIR}  -I../gen-cpp -L${LIB_DIR} -lthrift client.cpp ${GEN_SRC}

clean:
$(RM) -r client server