#include "dgram_mod_socket.h"
|
#include "shm_mm_wraper.h"
|
#include "usg_common.h"
|
#include "mm.h"
|
|
void * server_socket;
|
void sigint_handler(int sig) {
|
dgram_mod_close_socket(server_socket);
|
exit(0);
|
}
|
|
void server(int key, bool restart) {
|
server_socket = dgram_mod_open_socket();
|
|
|
if(restart) {
|
dgram_mod_force_bind(server_socket, key);
|
} else {
|
dgram_mod_bind(server_socket, key);
|
}
|
|
|
dgram_mod_start_bus(server_socket);
|
}
|
|
|
void *run_recv(void *socket) {
|
pthread_detach(pthread_self());
|
void *recvbuf;
|
int size;
|
int key;
|
while (dgram_mod_recvfrom( socket, &recvbuf, &size, &key) == 0) {
|
printf("收到订阅消息:%s\n", recvbuf);
|
free(recvbuf);
|
}
|
|
}
|
|
void client(int key) {
|
void *socket = dgram_mod_open_socket();
|
|
pthread_t tid;
|
pthread_create(&tid, NULL, run_recv, socket);
|
int size;
|
|
char action[512];
|
char topic[512];
|
char content[512];
|
long i = 0;
|
while (true) {
|
//printf("Usage: pub <topic> [content] or sub <topic>\n");
|
printf("Can I help you? sub, pub, desub or quit\n");
|
scanf("%s",action);
|
|
if(strcmp(action, "sub") == 0) {
|
printf("Please input topic!\n");
|
scanf("%s", topic);
|
if (dgram_mod_sub(socket, topic, strlen(topic), key) == 0) {
|
printf("%d Sub success!\n", dgram_mod_get_port(socket));
|
} else {
|
printf("Sub failture!\n");
|
exit(0);
|
}
|
|
} else if(strcmp(action, "desub") == 0) {
|
printf("Please input topic!\n");
|
scanf("%s", topic);
|
if (dgram_mod_desub(socket, topic, strlen(topic), key) == 0) {
|
printf("%d Desub success!\n", dgram_mod_get_port(socket));
|
} 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(dgram_mod_pub(socket, topic, strlen(topic)+1, content, strlen(content)+1, key) == 0){
|
printf("%d Pub success!\n", dgram_mod_get_port(socket));
|
} else {
|
printf("Pub failture!\n");
|
}
|
|
} else if(strcmp(action, "quit") == 0) {
|
printf("(%d) quit\n", dgram_mod_get_port(socket));
|
dgram_mod_close_socket(socket);
|
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|rmkey <key> ...\n", argv[0], "server", "client");
|
return 1;
|
}
|
|
key = atoi(argv[2]);
|
|
if (strcmp("server", argv[1]) == 0) {
|
if(argc >= 4 && strcmp("restart", argv[3]) == 0) {
|
server(key, true);
|
}
|
else{
|
server(key, false);
|
}
|
|
} else if (strcmp("client", argv[1]) == 0) {
|
client(key);
|
} else if(strcmp("rmkey", argv[1]) == 0) {
|
for(int i = 2; i < argc; i++) {
|
key = atoi(argv[i]);
|
dgram_mod_remove_key(key);
|
// printf("%d\n", key);
|
}
|
}
|
|
|
|
return 0;
|
}
|
