分享一种很简单的不阻塞TCP发送方法

发布于 2023-03-30 11:25:51 浏览：1336 订阅该版

[tocm]

## 背景介绍

最近在做一个功能的时候，要求将采集到的数据发送到TCP Server上，TCP Server有时候可能会出问题连接不上，这时候如果使用普通的TCP编程，就会造成数据采集的代码被阻塞了。
为了解决这个问题，我将TCP发送的代码使用一个单独的线程实现，数据采集的代码在另外一个线程实现，两个线程之间使用消息队列进行通信，这样既不会破坏TCP编程的基础模型，又可以实现应用代码非阻塞发送TCP数据。

## 代码实现

代码实现如下，当采集到数据的时候，只需要调用`tcp_client_no_block_send`函数发送数据即可。

```c
#include <rtthread.h>
#include <sys/socket.h>
#include <netdb.h>
#include <sys/select.h>
#include <string.h>
#include <finsh.h>
#include <fcntl.h>

#define DBG_TAG "tcp_client"
#define DBG_LVL DBG_INFO
#include <rtdbg.h>

typedef struct
{
	uint8_t data[512];
	uint32_t len;
}S_MSG, *P_MSG;

static rt_mq_t mq = RT_NULL;

static void tcp_client_thread_entry(void* parameter)
{
	int sockfd;
	struct sockaddr_in server_addr;
	S_MSG msg;
	
	SOCKET_INIT:
	
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if( sockfd < 0 )
	{
		LOG_E("create socket failed.");
		return;
	}
	
	rt_memset(&server_addr, 0, sizeof(server_addr));
	server_addr.sin_family = AF_INET;
	server_addr.sin_addr.s_addr = inet_addr("192.168.1.100");
	server_addr.sin_port = htons(10086);
	
	if( connect(sockfd, (struct sockaddr*)(&server_addr), sizeof(server_addr)) < 0 )
	{
		LOG_E("connect failed");
		closesocket(sockfd);
		goto SOCKET_INIT;
	}
	
	while(1)
	{
		if( rt_mq_recv(mq, &msg, sizeof(msg), RT_WAITING_FOREVER) != RT_EOK )
		{
			continue;
		}
				
		if( send(sockfd, msg.data, msg.len, 0) <= 0 )
		{
			closesocket(sockfd);
			goto SOCKET_INIT;
		}
	}	
}

int tcp_client_no_block_send(const uint8_t *data, uint32_t len)
{
	RT_ASSERT(mq!=RT_NULL);
	RT_ASSERT(data!=RT_NULL);
	
	S_MSG msg;	
	
	if( len > sizeof(msg.data) )
	{
		LOG_E("send len overflow");
		return -1;
	}
	
	msg.len = len;
	rt_memcpy(msg.data, data, len);
	if( rt_mq_send(mq, &msg, sizeof(msg)) == -RT_EFULL )
	{
		//LOG_E("tcp client msg overflow!");
	}
	
	return 0;
}

static int tcp_client_init(void)
{
    mq = rt_mq_create("tcp_mq", sizeof(S_MSG), 1, RT_IPC_FLAG_FIFO);
    if (mq == RT_NULL) 
	{
        LOG_E("create message queue failed");
        return -1;
    }
	
    rt_thread_t tcp_client_thread = rt_thread_create("tcp_client", tcp_client_thread_entry, RT_NULL, 2048, 10, 10);
	if (tcp_client_thread == RT_NULL) 
	{
        LOG_E("create thread failed");
        return -1;
    }
	
	rt_thread_startup(tcp_client_thread);
	return 0;
}
INIT_APP_EXPORT(tcp_client_init);
```