TCP长连接、短链接测试代码

背景

TCP长连接和短连接主要指的是客户端与服务器之间建立的TCP连接在通信结束后是保持还是立即关闭的状态
TCP长连接与短连接的测试是网络通信性能测试中的重要环节，它们各自具有不同的特点和适用场景。

测试

TCP长连接

TCP长连接指的是在一个TCP连接上可以发送多个数据包，
即使在没有数据包发送时，也需要双方发送检测包以维持这个连接。
长连接的优势在于可以减少建立连接和断开连接的开销（三次握手和四次挥手），节省资源和时间。
长连接也需要发送更多的心跳包来维持连接，增加了服务器的负担，一般会在心跳包中增加一下业务状态值

测试要点

• 连接稳定性：测试长连接在长时间运行下的稳定性，包括连接的保持时间、连接中断的频率等。
• 数据传输效率：在长连接上发送多个数据包，测试数据传输的速率和效率。
• 并发处理能力：测试服务器在处理多个长连接并发请求时的性能，包括响应时间、吞吐量等。
• 资源使用情况：监控服务器在长时间运行长连接时的CPU、内存、网络等资源使用情况。

TCP短连接

TCP短连接指的是当双方需要数据交互时，就建立一个TCP连接，本次交互完成后就断开这个连接。短连接的优势在于管理简单，存在的连接都是有用的连接，不需要额外的控制手段。如果客户端请求频繁，就会在TCP的建立连接和断开连接上浪费较大的资源和时间

测试要点

• 连接建立与断开速度：测试短连接建立和断开的速度，包括三次握手和四次握手的耗时。
• 数据传输效率：在短连接上发送数据包，测试数据传输的速率和效率。
• 并发处理能力：测试服务器在处理多个短连接并发请求时的性能，包括响应时间、吞吐量等。
• 资源回收情况：监控服务器在短连接断开后资源回收的情况，确保没有资源泄露。

代码

短连接

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
typedef unsigned long long  uint64;
uint64 totalnum;
uint64 errornum;
int main(int argc, char *const arvg[])
{int i;totalnum=0;errornum=0;char buff[1024];const char *hello = "Test-Short_conn\n"; // 根据具体业务调整int helloLen = strlen(hello);for (i = 0; i < 1000000; ++i) {int sock = socket(AF_INET, SOCK_STREAM, 0);    struct sockaddr_in srvAddr;srvAddr.sin_family = AF_INET;srvAddr.sin_addr.s_addr = inet_addr("127.0.0.1");srvAddr.sin_port = htons(7474);        connect(sock, (struct sockaddr *)&srvAddr, sizeof(srvAddr));        int size = 0;while (size < helloLen) {int ret = write(sock, hello + size, helloLen - size);if (ret > 0) size += ret;}totalnum++;recv(sock, buff, helloLen, MSG_WAITALL);printf("recv %s\n");// 接受服务器返回数据 根据业务逻辑判断是否正常close(sock);}return 0;}

长连接

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>int test_long_conn(void *arg) {int thread_id = (int)arg;printf("Test Client thread %d\n", thread_id);int sock = socket(AF_INET, SOCK_STREAM, 0);    struct sockaddr_in srvAddr;srvAddr.sin_family = AF_INET;srvAddr.sin_addr.s_addr = inet_addr("127.0.0.1");srvAddr.sin_port = htons(7474);connect(sock, (struct sockaddr *)&srvAddr, sizeof(srvAddr));int i;char buff[1024];const char *hello = "hello world\n";int helloLen = strlen(hello);for (i = 0; i < 1000; ++i) {int size = 0;while (size < helloLen) {int ret = write(sock, hello + size, helloLen - size);if (ret > 0) size += ret;}recv(sock, buff, helloLen, MSG_WAITALL);}close(sock);pthread_exit(NULL); // 线程结束
}//根据需求自定义线程数量 
int main() {pthread_t threads[100]; // 用于存储线程句柄的数组int rc;                // 存储返回代码的变量// 创建100个线程for (int i = 0; i < 100; i++) {rc = pthread_create(&threads[i], NULL, test_long_conn, (void *)i);if (rc) {printf("Error: unable to create thread, error code=%d\n", rc);return -1;}}// 等待所有线程完成for (int i = 0; i < 100; i++) {rc = pthread_join(threads[i], NULL);if (rc) {printf("Error: unable to join thread, error code=%d\n", rc);return -1;}}return 0;
}