int main(int argc, char* argv[]) {
pthread_mutex_t mutex_lock = PTHREAD_MUTEX_INITIALIZER;
socklen_t clilen;
struct sockaddr_in cli_addr;
int newsockfd = 0;
int senttuples = 0;
// Check parameter
if (argc < 2) {
cout << "You have to provider input data file name as a parameter." << endl;
return 0;
}
int second_increment = 0;
int nseconds = 10; // 1 minute
int port = atoi(argv[3]);
int tupleRate = atoi(argv[2]);
char* dataFile = argv[1];
//CLRDataProvider
CLRDataProvider* provider = new CLRDataProvider();
//Initialize the provider
cout << "Initializing..." << endl;
int ret = provider->Initialize(dataFile, 40000, &mutex_lock);
//Allocate caller's buffer
if (ret != SUCCESS) {
ErrorHandler(ret);
return 0;
}
int sockfd = init_connection_server(port);
if (sockfd < 0) {
provider->Uninitialize();
delete provider;
return -1;
}
newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
if (newsockfd < 0) {
perror("ERROR on accept");
provider->Uninitialize();
delete provider;
return -1;
}
//Using the provider
if (provider->PrepareData(provider) != SUCCESS) {
close(newsockfd);
close(sockfd);
provider->Uninitialize();
delete provider;
return -1;
}
int totalDuration = 6 * 60 * 1000;
int currentDuration = 0;
int nTuplesRead = 0;
char* str = new char[1024];
int mseconds = 1000;
int nTuplesToSend = 1;
float sleep_time = 1000 * 1/(double) tupleRate;
cout << "Sleep_time " << sleep_time << " tupleRate " << tupleRate << endl;
if (sleep_time < 1) {
nTuplesToSend = (int) ceil(1/sleep_time);
sleep_time = 1;
}
cout << "Sleeping " << sleep_time << " sending tuples " << nTuplesToSend << " at a time" << endl;
for (currentDuration =0; currentDuration <= totalDuration; currentDuration += sleep_time) {
// Sleep s seconds
usleep(1000*sleep_time);
int nMaxTuples = nTuplesToSend;
LPTuple lpTuples = new Tuple[nMaxTuples];
int ret;
int nTuplesSent = 0;
//Gets available data
ret = provider->GetData(lpTuples, nMaxTuples, nTuplesRead);
if (ret < 0) {
//Handle errors including eof
ErrorHandler(ret);
break;
}
if (nTuplesRead == 0) {
//No tuple available
//cout << "No buffer is available!" << endl;
continue;
}
nTuplesSent+= nTuplesRead;
//Using the return data
struct timeval start;
gettimeofday(&start, NULL);
memset(str, '\0', 1024);
for (int i = 0; i < nTuplesRead; i++) {
senttuples++;
memset(str, '\0', 1024);
sprintf(str, "%d,%ld,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d,%d\n",
lpTuples[i].m_iType, (start.tv_sec) * 1000 + (start.tv_usec) / 1000, lpTuples[i].m_iVid,
lpTuples[i].m_iSpeed, lpTuples[i].m_iXway, lpTuples[i].m_iLane,
lpTuples[i].m_iDir, lpTuples[i].m_iSeg, lpTuples[i].m_iPos, lpTuples[i].m_iTime,
lpTuples[i].m_iQid, lpTuples[i].m_iSinit, lpTuples[i].m_iSend,
lpTuples[i].m_iDow, lpTuples[i].m_iTod, lpTuples[i].m_iDay);
// send on the network
int n = write(newsockfd, str, strlen(str));
if (n < 0) {
cout <<"ERROR writing to socket" << endl;
}
}
if (nTuplesRead < nMaxTuples) {
//Last tuple has been read
cout << "Read of last tuple!" << endl;
continue;
}
delete lpTuples;
}
//Uninitialize the provider
cout << "Uninitialize..." << endl;
provider->Uninitialize();
delete provider;
close(newsockfd);
close(sockfd);
return 0;
}
int main(int argc, char* argv[])
{
pthread_mutex_t mutex_lock = PTHREAD_MUTEX_INITIALIZER;
// Check parameter
if (argc < 2)
{
cout << "You have to provide the input data file name as a commad line argument." << endl;
return 0;
}
char* dataFile = argv[1];
int64_t eventCnt = 0;
//CLRDataProvider
CLRDataProvider* provider = new CLRDataProvider();
//Initialize the provider
cout << "Initializing..." << endl;
int ret = provider->Initialize(dataFile, 10000, &mutex_lock);
//Allocate caller's buffer
if ( ret != SUCCESS )
{
ErrorHandler(ret);
return 0;
}
else
{
//Using the provider
if( provider->PrepareData(provider) == SUCCESS )
{
int nTuplesRead = 0;
int nMaxTuples = 100;
LPTuple lpTuples = new LRTuple[nMaxTuples ];
int seconds = 0;
long emittedCarEventsCnt = 0;
for(;;) {
// Get a random number between 5 and 15
srand(time(NULL));
int s = (int) ((((double) rand())/ RAND_MAX) * 10) + 5;
// Sleep s seconds
sleep(s);
int ret;
for(;;) {
//Gets available data
ret = provider->GetData(lpTuples, nMaxTuples, nTuplesRead);
if ( ret < 0 ) {
//Handle erros including eof
ErrorHandler(ret);
break;
}
if ( nTuplesRead == 0 ) {
//No tuple available
break;
}
//Using the return data
for(int i = 0; i < nTuplesRead; i++ ) {
eventCnt++;
// Always print the very first event.
// After that, print only once for every 500K events.
if ((eventCnt == 1ul) || ((eventCnt % 500000ul) == 0ul)) {
time_t now = time(0);
struct tm tstruct;
char buf[80];
tstruct = *localtime(&now);
strftime(buf, sizeof(buf), "%Y-%m-%d.%X", &tstruct);
char* str = lpTuples[i].ToString();
cout << string(buf) << " " << eventCnt << "-->" << str << endl;
delete str;
}
emittedCarEventsCnt++;
}
if ( nTuplesRead < nMaxTuples )
{
//Last tuple has been read
break;
}
}
if ( ret < SUCCESS )
{
break;
}
seconds += s;
cout << seconds << " seconds have passed. Number of car events emitted=" << emittedCarEventsCnt << endl;
}
}
//Uninitialize the provider
cout << "Uninitialize..." << endl;
provider->Uninitialize();
}
delete provider;
return 0;
}