greg/stasis-bLSM
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

check_tcpclient now works (sort of...)

Browse source 
git-svn-id: svn+ssh://svn.corp.yahoo.com/yahoo/yrl/labs/pnuts/code/logstore@541 8dad8b1f-cf64-0410-95b6-bcf113ffbcfe
This commit is contained in:
sears 2010-02-02 22:50:09 +00:00
parent 094a0d41ee
commit 3cbcb1e274
3 changed files with 32 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
logserver.cpp
View file

@ -440,6 +440,11 @@ void * thread_work_fn( void * args)
//step 1: read the opcode
uint8_t opcode;
ssize_t n = read(*(item->data->workitem), &opcode, sizeof(uint8_t));
if(n == 0) {
opcode = logserver::OP_DONE;
n = sizeof(uint8_t);
printf("Obsolescent client closed connection uncleanly\n");
}
assert( n == sizeof(uint8_t));
assert( opcode < logserver::OP_INVALID );

19
test/check_tcpclient.cpp
View file

@ -16,11 +16,15 @@
#undef begin
#undef end
static char * svrname = "localhost";
static int svrport = 32432;
void insertProbeIter(size_t NUM_ENTRIES)
{
srand(1000);
std::string servername = "sherpa4";
int serverport = 32432;
std::string servername = svrname; //"sherpa4";
int serverport = svrport; //32432;
double delete_freq = .05;
double update_freq = .15;
@ -140,7 +144,8 @@ void insertProbeIter(size_t NUM_ENTRIES)
gettimeofday(&ti_st,0);
//send the data
sendTuple(servername, serverport, logserver::OP_INSERT, newtuple);
datatuple * ret = sendTuple(servername, serverport, logserver::OP_INSERT, newtuple);
assert(ret);
gettimeofday(&ti_end,0);
insert_time += tv_to_double(ti_end) - tv_to_double(ti_st);
@ -217,8 +222,14 @@ void insertProbeIter(size_t NUM_ENTRIES)
/** @test
*/
int main()
int main(int argc, char* argv[])
{
if(argc > 1) {
svrname = argv[1];
}
if(argc > 2) {
svrport = atoi(argv[2]);
}
//insertProbeIter(25000);
insertProbeIter(100000);
/*

19
test/check_util.h
View file

@ -162,13 +162,14 @@ datatuple * sendTuple(std::string & servername, int serverport, uint8_t opcode,
if (sockfd < 0)
{
printf("ERROR opening socket.\n");
perror("ERROR opening socket.\n");
return 0;
}
server = gethostbyname(servername.c_str());
if (server == NULL) {
fprintf(stderr,"ERROR, no such host as %s\n", servername.c_str());
close(sockfd);
exit(0);
}
@ -182,7 +183,8 @@ datatuple * sendTuple(std::string & servername, int serverport, uint8_t opcode,
/* connect: create a connection with the server */
if (connect(sockfd, (sockaddr*) &serveraddr, sizeof(serveraddr)) < 0)
{
printf("ERROR connecting\n");
perror("ERROR connecting\n");
close(sockfd);
return 0;
}
@ -206,6 +208,8 @@ datatuple * sendTuple(std::string & servername, int serverport, uint8_t opcode,
uint8_t rcode;
n = read(sockfd, (byte*) &rcode, sizeof(uint8_t));
datatuple * ret;
if(rcode == logserver::OP_SENDING_TUPLE)
{
datatuple *rcvdtuple = (datatuple*)malloc(sizeof(datatuple));
@ -227,14 +231,15 @@ datatuple * sendTuple(std::string & servername, int serverport, uint8_t opcode,
logserver::readfromsocket(sockfd, (byte*) rcvdtuple->data, *rcvdtuple->datalen);
}
close(sockfd);
return rcvdtuple;
ret = rcvdtuple;
} else if(rcode == logserver::OP_SUCCESS) {
ret = &tuple;
} else {
ret = 0;
}
else
assert(rcode == logserver::OP_SUCCESS);
close(sockfd);
return 0;
return ret;
}