linux多线程实现数据接收和存储转发

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <string.h>

#define NUM_THREADS 3

typedef struct {
    int thread_id;
    char *data;
} ThreadData;

void *receive_data(void *arg) {
    ThreadData *td = (ThreadData *)arg;
    printf("Thread %d: Receiving data...\n", td->thread_id);
    // 模拟数据接收
    td->data = strdup("Received data");
    return NULL;
}

void *store_data(void *arg) {
    ThreadData *td = (ThreadData *)arg;
    printf("Thread %d: Storing data...\n", td->thread_id);
    // 模拟数据存储
    free(td->data); // 释放之前接收的数据
    td->data = strdup("Stored data");
    return NULL;
}

void *forward_data(void *arg) {
    ThreadData *td = (ThreadData *)arg;
    printf("Thread %d: Forwarding data...\n", td->thread_id);
    // 模拟数据转发
    free(td->data); // 释放之前存储的数据
    td->data = strdup("Forwarded data");
    return NULL;
}

int main() {
    pthread_t threads[NUM_THREADS];
    ThreadData thread_data[NUM_THREADS];

    for (int i = 0; i < NUM_THREADS; i++) {
        thread_data[i].thread_id = i;

        if (i % 3 == 0) {
            pthread_create(&threads[i], NULL, receive_data, (void *)&thread_data[i]);
        } else if (i % 3 == 1) {
            pthread_create(&threads[i], NULL, store_data, (void *)&thread_data[i]);
        } else {
            pthread_create(&threads[i], NULL, forward_data, (void *)&thread_data[i]);
        }
    }

    for (int i = 0; i < NUM_THREADS; i++) {
        pthread_join(threads[i], NULL);
    }

    return 0;
}