集中分词服务器-原型

1.基本思路：建立两个监听，一个接收用户的查询，一个接收配置服务器的自动注册。采用线程+网络编程，c语言 under ubuntu

/* * ===================================================================================== * * Filename: core_server.c * * Description: 中心服务器，负责进行分词服务器的分发以及信息收集 * * Version: 1.0 * Created: 2011年01月05日 16时57分54秒 * Revision: none * Compiler: gcc * * Author: 齐保元 (qby), qibaoyuan@126.com * Company: ict,gucas * * ===================================================================================== */ #include <signal.h> #include <stdlib.h> #include <stdio.h> #include <sys/types.h> #include <unistd.h> #include <pthread.h> #include <fcntl.h> #include <errno.h> #include <sys/socket.h> #include <arpa/inet.h> #include <netinet/in.h> #include <netdb.h> #include <sys/shm.h> #include <sys/stat.h> #include <string.h> #define DISPATCH_PORT 6784 #define CONFIG_PORT 4327 #define MAX_MSG_SIZE 1024 #define MAX_SERVER_SIZE 300 typedef struct _seg_info{ char ip[200]; int port; }seg_info; seg_info servers[MAX_SERVER_SIZE]; int server_index=0; int init_daemon();//daemon进程 int UDP_listen(int); void dispatch_response(int); void config_response(int); void get_config(char*); void put_config(seg_info); int main(int argc,char *argv[]){ printf("分配端口:%d,配置端口:%d/n",DISPATCH_PORT,CONFIG_PORT); //隐藏进程 //init_daemon(); //分发标识 int dispatch_fd; //接收分词服务器的配置信息 int config_fd; dispatch_fd=UDP_listen(DISPATCH_PORT); config_fd=UDP_listen(CONFIG_PORT); pthread_t id; int ret; ret=pthread_create(&id,NULL,(void*)dispatch_response,(void*)dispatch_fd); if(0!=ret){ printf("创建线程出错!/n"); exit(1); } config_response(config_fd); pthread_join(id,NULL); return 1; } int init_daemon(){ struct sigaction act; int i,max_fd; if(fork()!=0)exit(0);//fork if(setsid()<0) return (-1); act.sa_handler=SIG_IGN; act.sa_flags=0; sigaction(SIGHUP,&act,0); if(fork()!=0)exit(0); chdir("/");//root umask(0);//mask max_fd=sysconf(_SC_OPEN_MAX); for(i=0;i<max_fd;++i){ close(i); } open("/dev/null",O_RDWR); dup(0); dup(1); dup(2); return(0); } int UDP_listen(int port){ struct sockaddr_in addr; int sockfd; sockfd=socket(AF_INET,SOCK_DGRAM,0); if(sockfd<0){ fprintf(stderr,"socker error:%s/n",strerror(errno)); return -1; } bzero(&addr,sizeof(struct sockaddr_in)); addr.sin_family=AF_INET; addr.sin_addr.s_addr=htonl(INADDR_ANY); addr.sin_port=htons(port); if(bind(sockfd,(struct sockaddr*)&addr,sizeof(struct sockaddr_in))<0){ fprintf(stderr,"bind error:%s/n",strerror(errno)); return -1; } return sockfd; } void dispatch_response(int fd){ struct sockaddr_in addr; socklen_t addr_len; addr_len=sizeof(struct sockaddr); char msg[MAX_MSG_SIZE]; char res[MAX_MSG_SIZE]; srand((unsigned)time(NULL)); while(1){ bzero(msg,sizeof(msg)); bzero(res,sizeof(res)); recvfrom(fd,msg,MAX_MSG_SIZE,0,(struct sockaddr*)&addr,&addr_len); fprintf(stdout,"you are asking for a seg server:%s/n",msg); get_config(res); sendto(fd,res,sizeof(res),0,(struct sockaddr*)&addr,addr_len); } } void config_response(int fd){ struct sockaddr_in addr; socklen_t addr_len; addr_len=sizeof(struct sockaddr); char msg[MAX_MSG_SIZE]; char res[MAX_MSG_SIZE]; while(1){ bzero(msg,sizeof(msg)); bzero(res,sizeof(res)); recvfrom(fd,msg,MAX_MSG_SIZE,0,(struct sockaddr*)&addr,&addr_len); fprintf(stdout,"got seg server info:%s/n",msg); seg_info seg; bzero(seg.ip,sizeof(seg.ip)); strcpy(seg.ip,inet_ntoa(addr.sin_addr)); seg.port=atoi(msg); printf("get seg_server:%s,port:%d/n",inet_ntoa(addr.sin_addr),ntohs(addr.sin_port)); //放入config_table中 put_config(seg); strcpy(res,"saving your info to tables."); sendto(fd,res,sizeof(res),0,(struct sockaddr*)&addr,addr_len); } } void get_config(char* dest){ int i=0; if(server_index>0) i=rand()%server_index; else i=server_index; printf("server_index:%d,i:%d/n",server_index,i); printf("i.ip:%s,port:%d/n",servers[i].ip,servers[i].port); strcpy(dest,servers[i].ip); sprintf(dest,"%s%s%d",dest,":",servers[i].port); printf("返回的seg_server_ip:%s/n",dest); } void put_config(seg_info seg){ if(server_index>=MAX_SERVER_SIZE-1){ fprintf(stderr,"too many servers to allocate,rounding...!/n"); server_index=0; } servers[server_index]=seg; printf("入库后：%s,%d/n",servers[server_index].ip,servers[server_index].port); server_index++; printf("server_index:%d,ip:%s,port:%d/n",server_index,seg.ip,seg.port); }