gfwleak/stellar-stellar
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feature: session pool support forearch session by index

Browse Source
This commit is contained in:
luwenpeng
2024-07-25 18:32:14 +08:00
parent f43da9002b
commit 5fdf0e2aa4
3 changed files with 54 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/session/session_pool.cpp
View File

@@ -5,25 +5,28 @@
struct session_pool
{
uint64_t count;
uint64_t capacity;
uint64_t available;
struct list_head free_queue;
};
struct session_pool *session_pool_new(uint64_t count)
struct session_pool *session_pool_new(uint64_t capacity)
{
struct session_pool *pool = (struct session_pool *)calloc(1, sizeof(struct session_pool) + count * sizeof(struct session));
struct session_pool *pool = (struct session_pool *)calloc(1, sizeof(struct session_pool) + capacity * sizeof(struct session));
if (pool == NULL)
{
return NULL;
}
pool->available = 0;
pool->capacity = capacity;
INIT_LIST_HEAD(&pool->free_queue);
struct session *array = (struct session *)(pool + 1);
for (uint64_t i = 0; i < count; i++)
for (uint64_t i = 0; i < capacity; i++)
{
struct session *sess = &array[i];
list_add_tail(&sess->free, &pool->free_queue);
pool->count++;
pool->available++;
}
return pool;
@@ -37,7 +40,7 @@ void session_pool_free(struct session_pool *pool)
{
struct session *sess = list_first_entry(&pool->free_queue, struct session, free);
list_del(&sess->free);
pool->count--;
pool->available--;
}
free(pool);
@@ -59,7 +62,7 @@ struct session *session_pool_pop(struct session_pool *pool)
struct session *sess = list_first_entry(&pool->free_queue, struct session, free);
list_del(&sess->free);
pool->count--;
pool->available--;
return sess;
}
@@ -72,15 +75,36 @@ void session_pool_push(struct session_pool *pool, struct session *sess)
}
list_add_tail(&sess->free, &pool->free_queue);
pool->count++;
pool->available++;
}
uint64_t session_pool_get_count(struct session_pool *pool)
const struct session *session_pool_get0(const struct session_pool *pool, uint64_t idx)
{
if (pool == NULL || idx >= pool->capacity)
{
return NULL;
}
struct session *array = (struct session *)(pool + 1);
return &array[idx];
}
uint64_t session_pool_available_num(const struct session_pool *pool)
{
if (pool == NULL)
{
return 0;
}
return pool->count;
}
return pool->available;
}
uint64_t session_pool_capacity_size(const struct session_pool *pool)
{
if (pool == NULL)
{
return 0;
}
return pool->capacity;
}

7
src/session/session_pool.h
View File

@@ -8,12 +8,15 @@ extern "C"
#include <stdint.h>
struct session_pool;
struct session_pool *session_pool_new(uint64_t count);
struct session_pool *session_pool_new(uint64_t capacity);
void session_pool_free(struct session_pool *pool);
struct session *session_pool_pop(struct session_pool *pool);
void session_pool_push(struct session_pool *pool, struct session *sess);
uint64_t session_pool_get_count(struct session_pool *pool);
const struct session *session_pool_get0(const struct session_pool *pool, uint64_t idx);
uint64_t session_pool_available_num(const struct session_pool *pool);
uint64_t session_pool_capacity_size(const struct session_pool *pool);
#ifdef __cplusplus
}

27
src/session/test/gtest_session_pool.cpp
View File

@@ -12,45 +12,46 @@ TEST(SESSION_POOL, POP_PUSH)
sess_pool = session_pool_new(3);
EXPECT_TRUE(sess_pool != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 3);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 3);
EXPECT_TRUE(session_pool_capacity_size(sess_pool) == 3);
sess1 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess1 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 2);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 2);
sess2 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess2 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 1);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 1);
sess3 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess3 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 0);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 0);
sess4 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess4 == NULL);
session_pool_push(sess_pool, sess1);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 1);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 1);
session_pool_push(sess_pool, sess2);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 2);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 2);
session_pool_push(sess_pool, sess3);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 3);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 3);
sess1 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess1 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 2);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 2);
sess2 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess2 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 1);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 1);
sess3 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess3 != NULL);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 0);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 0);
sess4 = session_pool_pop(sess_pool);
EXPECT_TRUE(sess4 == NULL);
session_pool_push(sess_pool, sess1);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 1);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 1);
session_pool_push(sess_pool, sess2);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 2);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 2);
session_pool_push(sess_pool, sess3);
EXPECT_TRUE(session_pool_get_count(sess_pool) == 3);
EXPECT_TRUE(session_pool_available_num(sess_pool) == 3);
session_pool_free(sess_pool);
}