Function ExtendBlockPoolImpl

src/brpc/rdma/block_pool.cpp:133–182

}

static void* ExtendBlockPoolImpl(void* region_base, size_t region_size, int block_type) {
    auto region_base_guard = butil::MakeScopeGuard([region_base]() {
        free(region_base);
    });

    if (g_region_num == FLAGS_rdma_memory_pool_max_regions) {
        LOG_EVERY_SECOND(ERROR) << "Memory pool reaches max regions";
        errno = ENOMEM;
        return NULL;
    }

    uint32_t id = g_cb(region_base, region_size);
    if (id == 0) {
        errno = EINVAL;
        return NULL;
    }

    IdleNode* node[g_buckets];
    for (size_t i = 0; i < g_buckets; ++i) {
        node[i] = butil::get_object<IdleNode>();
        if (!node[i]) {
            PLOG_EVERY_SECOND(ERROR) << "Memory not enough";
            for (size_t j = 0; j < i; ++j) {
                butil::return_object<IdleNode>(node[j]);
            }
            errno = ENOMEM;
            return NULL;
        }
    }

    Region* region = &g_regions[g_region_num++];
    region->start = (uintptr_t)region_base;
    region->size = region_size;
    region->id = id;
    region->block_type = block_type;

    for (size_t i = 0; i < g_buckets; ++i) {
        node[i]->start = (void*)(region->start + i * (region_size / g_buckets));
        node[i]->len = region_size / g_buckets;
        node[i]->next = g_info->expansion_list[block_type][i];
        g_info->expansion_list[block_type][i] = node[i];
        g_info->expansion_size[block_type][i] += node[i]->len;
    }
    g_info->region_num[block_type]++;

    // `region_base' is inuse, cannot be freed.
    region_base_guard.dismiss();

    return region_base;
}

// Extend the block pool with a new region (with different region ID)
static void* ExtendBlockPool(size_t region_size, int block_type) {