[英]Thread-specific data
我有一个客户端程序,如下所示,我需要使其成为多线程,即每个连接一个线程。 但是变量sockfd将被全局保留在一个线程中。 我知道要执行此操作,我需要使用pthread_key_t,pthread_key_create ...等。 但是,我很困惑如何使用它。 我将不胜感激。
int sockfd;
pthread_key_t key_to_sockfd;
void error(const char *msg)
{
perror(msg);
exit(0);
}
void set_connection(char *argv[])
{
int portno;
struct sockaddr_in serv_addr;
struct hostent *server;
char buffer[256];
portno = atoi(argv[2]);
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0)
error("ERROR opening socket");
server = gethostbyname(argv[1]);
if (server == NULL) {
fprintf(stderr,"ERROR, no such host\n");
exit(0);
}
bzero((char *) &serv_addr, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
bcopy((char *)server->h_addr,
(char *)&serv_addr.sin_addr.s_addr,server->h_length);
serv_addr.sin_port = htons(portno);
if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0)
error("ERROR connecting");
return;
}
void send_message()
{
char buffer[256];
int i=0,n;
do{
printf("Please enter the message: ");
bzero(buffer,256);
fgets(buffer,255,stdin);
n = write(sockfd,buffer,strlen(buffer));
if (n < 0)
error("ERROR writing to socket");
i++;
}while(i<3);
bzero(buffer,256);
n = read(sockfd,buffer,255);
if (n < 0)
error("ERROR reading from socket");
printf("%s\n",buffer);
return;
}
void disconnect()
{
close(sockfd);
return;
}
void client_thrd(char *argv[])
{
set_connection(argv);
send_message();
disconnect();
}
int main(int argc, char *argv[])
{
pthread_t thid[2];
int i;
void *status;
if (argc < 3) {
fprintf(stderr,"usage %s hostname port\n", argv[0]);
exit(0);
}
for(i=0;i<1;i++)
pthread_create(&thid[i],NULL,(void*)&client_thrd,(void*)argv);
for(i=0;i<1;i++)
pthread_join(thid[i],&status);
return 0;
}
我已经为一个线程执行了该程序,并且工作正常。 但是随着线程数增加到一个以上,由于全局的sockfd,它不起作用。 这是一个更大的测试程序。
谢谢并恭祝安康
这是非常容易使用。 您在程序开始时调用pthread_key_create(&key_to_sockfd, NULL)
,然后在每个线程中将看到NULL的初始值。 在每个线程中使用函数pthread_setspecific(key_to_sockfd, <pointer to a struct which contains sockfd>)
。
从那时起,每个线程将看到一个指向包含sockfd的结构的不同指针。 完成线程处理后,删除结构并使用pthread_setspecific(key_to_sockfd, NULL)
线程完成后,请调用pthread_key_delete(key_to_sockfd)
删除存储。 您还可以通过向pthread_key_create
提供回调函数来自动清理,以在线程完成后释放内存。
从我的角度来看,您不需要全局变量即可实现您想要的工作。 创建一个“状态”数据结构,并在启动每个线程之前对其进行初始化。 为此在线程接口中设置了void*
参数。
如果我查看您的代码,则用每个线程覆盖,则启动套接字描述符sockfd。 因此,每个线程将建立一个新的连接并获得一个新的sockfd。 使用此代码,您很可能一个线程正在关闭另一个线程的连接。 如果要为每个线程使用自己的套接字描述符,为什么不使用指针在同一线程内共享同一套接字描述符?
void client_thrd(char *argv[])
{
int sockfd;
set_connection(&sockfd, argv);
send_message(&sockfd);
disconnect(&sockfd);
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.