dpdk应用场景系列03--rte_ring 使用(多个线程之间传递数据）

dpdk应用案例03
rte_ring 在多个线程之间传递数据

• 多线程数据传递：在多核环境下，线程之间需要高效地传递数据，而不需要复杂的锁机制。rte_ring 提供了无锁的多生产者、多消费者队列，极大地提高了性能。
• 任务队列：可以用于实现任务调度系统，多个生产者线程将任务放入队列，多个消费者线程从队列中取出任务处理。
• 数据包缓存：在网络应用中，可以用于缓存数据包，多个线程可以并发地将数据包放入或取出环形缓冲区。

---

rte_ring 在多个线程之间传递数据

#include <rte_eal.h>
#include <rte_ring.h>
#include <rte_mempool.h>
#include <rte_mbuf.h>
#include <stdio.h>
#include <pthread.h>

#define RING_SIZE 1024
#define NUM_MBUFS 8192
#define MBUF_CACHE_SIZE 250
#define BURST_SIZE 32

struct rte_ring *ring;
struct rte_mempool *mbuf_pool;

void *producer(void *arg) {
    while (1) {
        struct rte_mbuf *mbufs[BURST_SIZE];
        for (int i = 0; i < BURST_SIZE; i++) {
            mbufs[i] = rte_pktmbuf_alloc(mbuf_pool);
            if (mbufs[i] == NULL) {
                printf("Failed to allocate mbuf\n");
                continue;
            }
            char *data = rte_pktmbuf_mtod(mbufs[i], char *);
            snprintf(data, rte_pktmbuf_tailroom(mbufs[i]), "Hello from producer %ld!", (long)arg);
            rte_pktmbuf_pkt_len(mbufs[i]) = strlen(data);
            rte_pktmbuf_data_len(mbufs[i]) = strlen(data);
        }
        if (rte_ring_enqueue_bulk(ring, (void **)mbufs, BURST_SIZE, NULL) == 0) {
            for (int i = 0; i < BURST_SIZE; i++) {
                rte_pktmbuf_free(mbufs[i]);
            }
        }
    }
    return NULL;
}

void *consumer(void *arg) {
    while (1) {
        struct rte_mbuf *mbufs[BURST_SIZE];
        unsigned int nb_rx = rte_ring_dequeue_bulk(ring, (void **)mbufs, BURST_SIZE, NULL);
        for (unsigned int i = 0; i < nb_rx; i++) {
            char *data = rte_pktmbuf_mtod(mbufs[i], char *);
            printf("Consumer %ld received: %s\n", (long)arg, data);
            rte_pktmbuf_free(mbufs[i]);
        }
    }
    return NULL;
}

int main(int argc, char **argv) {
    if (rte_eal_init(argc, argv) < 0)
        rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");

    mbuf_pool = rte_pktmbuf_pool_create("MBUF_POOL", NUM_MBUFS,
                                        MBUF_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE,
                                        rte_socket_id());
    if (mbuf_pool == NULL)
        rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");

    ring = rte_ring_create("RING", RING_SIZE, rte_socket_id(), RING_F_SP_ENQ | RING_F_SC_DEQ);
    if (ring == NULL)
        rte_exit(EXIT_FAILURE, "Cannot create ring\n");

    pthread_t prod_threads[2], cons_threads[2];
    for (long i = 0; i < 2; i++) {
        pthread_create(&prod_threads[i], NULL, producer, (void *)i);
        pthread_create(&cons_threads[i], NULL, consumer, (void *)i);
    }

    for (int i = 0; i < 2; i++) {
        pthread_join(prod_threads[i], NULL);
        pthread_join(cons_threads[i], NULL);
    }

    return 0;
}