#include "bus_server_socket.h"
|
#include "shm_mod_socket.h"
|
#include "shm_mm_wrapper.h"
|
#include "usg_common.h"
|
#include "mm.h"
|
#include "logger_factory.h"
|
|
#include "bus_error.h"
|
|
static Logger *logger = LoggerFactory::getLogger();
|
|
BusServerSocket * server_socket;
|
void sigint_handler(int sig) {
|
|
exit(0);
|
}
|
|
void server(int key) {
|
server_socket = new BusServerSocket();
|
|
server_socket->bind( key);
|
|
server_socket->start();
|
}
|
|
|
void *run_recv(void *skptr) {
|
pthread_detach(pthread_self());
|
void *recvbuf;
|
int size;
|
int key;
|
|
int rv;
|
ShmModSocket *sk = (ShmModSocket *)skptr;
|
struct timespec timeout = {2, 0};
|
while (true) {
|
printf("run_recv before\n");
|
|
// rv = sk->recvfrom_timeout( &recvbuf, &size, &key, &timeout);
|
rv = sk->recvfrom( &recvbuf, &size, &key);
|
if(rv == 0) {
|
printf("收到订阅消息:%s\n", recvbuf);
|
free(recvbuf);
|
} else {
|
printf("recvfrom error %d\n", rv);
|
}
|
|
}
|
return NULL;
|
|
}
|
|
void client(int key) {
|
ShmModSocket *sk = new ShmModSocket();
|
|
pthread_t tid;
|
|
|
int size;
|
|
char action[512];
|
char topic[512];
|
char content[512];
|
long i = 0;
|
|
pthread_create(&tid, NULL, run_recv, (void *)sk);
|
|
while (true) {
|
//printf("Usage: pub <topic> [content] or sub <topic>\n");
|
printf("Can I help you? sub, pub, desub or quit %d\n", i++);
|
// scanf("%s", action);
|
std::cin >> action;
|
if(strcmp(action, "sub") == 0) {
|
printf("Please input topic!\n");
|
scanf("%s", topic);
|
if (sk->sub(topic, strlen(topic), key) == 0) {
|
printf("%d Sub success!\n", sk->get_key());
|
} else {
|
printf("Sub failture!\n");
|
exit(0);
|
}
|
|
} else if(strcmp(action, "desub") == 0) {
|
printf("Please input topic!\n");
|
scanf("%s", topic);
|
if (sk->desub(topic, strlen(topic), key) == 0) {
|
printf("%d Desub success!\n", sk->get_key());
|
} else {
|
printf("Desub failture!\n");
|
exit(0);
|
}
|
|
} else if(strcmp(action, "pub") == 0) {
|
// printf("%s %s %s\n", action, topic, content);
|
printf("Please input topic and content\n");
|
scanf("%s %s", topic, content);
|
if(sk->pub(topic, strlen(topic)+1, content, strlen(content)+1, key) == 0){
|
printf("%d Pub success!\n", sk->get_key());
|
} else {
|
printf("Pub failture!\n");
|
}
|
|
} else if(strcmp(action, "quit") == 0) {
|
printf("(%d) quit\n", sk->get_key());
|
delete sk;
|
break;
|
} else {
|
printf("error input argument\n");
|
continue;
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
shm_mm_wrapper_init(512);
|
int key;
|
if (argc < 3) {
|
fprintf(stderr, "Usage: %s %s|%s <key> ...\n", argv[0], "server", "client");
|
return 1;
|
}
|
|
key = atoi(argv[2]);
|
|
if (strcmp("server", argv[1]) == 0) {
|
server(key);
|
|
} else if (strcmp("client", argv[1]) == 0) {
|
client(key);
|
}
|
|
|
|
return 0;
|
}
|
