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;
}