#include <stdlib.h>
|
#include <stdio.h>
|
#include <string.h>
|
#include <time.h>
|
#include <sys/time.h>
|
#include <pthread.h>
|
|
#include <nng/nng.h>
|
#include <nng/protocol/reqrep0/rep.h>
|
#include <nng/protocol/reqrep0/req.h>
|
|
#define NODE0 "node0"
|
#define NODE1 "node1"
|
#define DATE "DATE"
|
|
|
typedef struct Targ {
|
const char* url;
|
int id;
|
|
}Targ;
|
|
void
|
fatal(const char *func, int rv)
|
{
|
fprintf(stderr, "%s: %s\n", func, nng_strerror(rv));
|
exit(1);
|
}
|
|
char *
|
date(void)
|
{
|
time_t now = time(&now);
|
struct tm *info = localtime(&now);
|
char *text = asctime(info);
|
text[strlen(text) - 1] = '\0'; // remove '\n'
|
return (text);
|
}
|
|
int
|
server(const char *url)
|
{
|
nng_socket sock;
|
int rv;
|
char sendbuf[512];
|
|
if ((rv = nng_rep0_open(&sock)) != 0)
|
{
|
fatal("nng_rep0_open", rv);
|
}
|
if ((rv = nng_listen(sock, url, NULL, 0)) != 0)
|
{
|
fatal("nng_listen", rv);
|
}
|
|
for (;;)
|
{
|
char *buf = NULL;
|
size_t sz;
|
if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0)
|
{
|
fatal("nng_recv", rv);
|
}
|
|
|
// printf("RECEIVED:%s\n", buf);
|
char *d = date();
|
sprintf(sendbuf, "%s: %s", d, buf);
|
// printf("SENDING DATE %s\n", d);
|
if ((rv = nng_send(sock, sendbuf, strlen(sendbuf) + 1, 0)) != 0)
|
{
|
fatal("nng_send", rv);
|
}
|
nng_free(buf, sz);
|
}
|
}
|
|
|
|
int
|
client(const char *url)
|
{
|
nng_socket sock;
|
int rv;
|
size_t sz;
|
char * buf;
|
char sendbuf[512];
|
|
if ((rv = nng_req0_open(&sock)) != 0)
|
{
|
fatal("nng_socket", rv);
|
}
|
if ((rv = nng_dial(sock, url, NULL, 0)) != 0)
|
{
|
fatal("nng_dial", rv);
|
}
|
//printf("NODE1: SENDING DATE REQUEST %s\n", DATE);
|
while (true)
|
{
|
printf("say something:\n");
|
scanf("%s", sendbuf);
|
if ((rv = nng_send(sock, sendbuf, strlen(sendbuf) + 1, 0)) != 0)
|
{
|
fatal("nng_send", rv);
|
}
|
if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0)
|
{
|
fatal("nng_recv", rv);
|
}
|
printf("RECEIVED DATE %s\n", buf);
|
nng_free(buf, sz);
|
}
|
|
nng_close(sock);
|
return (0);
|
}
|
|
|
|
|
#define SCALE 100000
|
|
void *runclient(void *arg) {
|
Targ *targ = (Targ *)arg;
|
const char* url = targ->url;
|
char sendbuf[512];
|
size_t sz;
|
void *buf;
|
|
int i,j, n;
|
nng_socket sock;
|
int rv;
|
char filename[512];
|
|
|
if ((rv = nng_req0_open(&sock)) != 0)
|
{
|
fatal("nng_socket", rv);
|
}
|
if ((rv = nng_dial(sock, url, NULL, 0)) != 0)
|
{
|
fatal("nng_dial", rv);
|
}
|
|
|
sprintf(filename, "test%d.tmp", targ->id);
|
FILE *fp = NULL;
|
fp = fopen(filename, "w+");
|
// fp = stdout;
|
|
|
for (i = 0; i < SCALE; i++) {
|
sprintf(sendbuf, "thread(%d) %d", targ->id, i);
|
fprintf(fp, "requst:%s\n", sendbuf);
|
|
if ((rv = nng_send(sock, sendbuf, strlen(sendbuf) + 1, 0)) != 0)
|
{
|
fatal("nng_send", rv);
|
}
|
if ((rv = nng_recv(sock, &buf, &sz, NNG_FLAG_ALLOC)) != 0)
|
{
|
fatal("nng_recv", rv);
|
}
|
fprintf(fp, "NODE1: RECEIVED DATE %s\n", buf);
|
nng_free(buf, sz);
|
|
}
|
fclose(fp);
|
nng_close(sock);
|
return (void *)i;
|
}
|
|
|
void mclient(const char *url) {
|
|
int status, i = 0, processors = 1;
|
void *res[processors];
|
// Targ *targs = (Targ *)calloc(processors, sizeof(Targ));
|
Targ targs[processors];
|
pthread_t tids[processors];
|
char sendbuf[512];
|
struct timeval start, end;
|
long total = 0;
|
|
printf("开始测试...\n");
|
gettimeofday(&start, NULL);
|
for (i = 0; i < processors; i++) {
|
targs[i].url = url;
|
targs[i].id = i;
|
pthread_create(&tids[i], NULL, runclient, (void *)&targs[i]);
|
}
|
|
for (i = 0; i < processors; i++) {
|
if (pthread_join(tids[i], &res[i]) != 0) {
|
perror("multyThreadClient pthread_join");
|
} else {
|
total += (long)res[i];
|
//fprintf(stderr, "client(%d) 写入 %ld 条数据\n", i, (long)res[i]);
|
}
|
}
|
|
gettimeofday(&end, NULL);
|
|
double difftime = end.tv_sec * 1000000 + end.tv_usec - (start.tv_sec * 1000000 + start.tv_usec);
|
long diffsec = (long) (difftime/1000000);
|
long diffusec = difftime - diffsec * 1000000;
|
fprintf(stderr,"发送数目: %ld, 用时: (%ld sec %ld usec), 平均: %f\n", total, diffsec, diffusec, difftime/total );
|
// fflush(stdout);
|
}
|
|
|
int
|
main(const int argc, const char **argv)
|
{
|
const char *url = "tcp://192.168.20.10:5001";
|
//const char *url = "ipc:///tmp/reqrep.ipc";
|
|
if (argc < 2) {
|
fprintf(stderr, "Usage: %s %s|%s \n", argv[0], "server", "client");
|
return 1;
|
}
|
|
// url = atoi(argv[2]);
|
|
|
if (strcmp("server", argv[1]) == 0 ) {
|
server(url);
|
}
|
|
if (strcmp("client", argv[1]) == 0)
|
client(url);
|
|
if (strcmp("mclient", argv[1]) == 0)
|
mclient(url);
|
|
}
|
