执行resetall指令:Sapp重启后发送resetall状态通知SCE清空流表
This commit is contained in:
@@ -32,6 +32,7 @@ struct session_table;
|
||||
|
||||
struct session_table *session_table_create();
|
||||
void session_table_destory(struct session_table *table);
|
||||
void session_table_reset(struct session_table *table);
|
||||
uint64_t session_table_count(struct session_table *table);
|
||||
|
||||
// session_addr : deep copy
|
||||
|
||||
@@ -47,6 +47,30 @@ void session_table_destory(struct session_table *table)
|
||||
}
|
||||
}
|
||||
|
||||
void session_table_reset(struct session_table *table)
|
||||
{
|
||||
if (table)
|
||||
{
|
||||
struct session_node *temp = NULL;
|
||||
struct session_node *node = NULL;
|
||||
HASH_ITER(hh1, table->root_by_id, node, temp)
|
||||
{
|
||||
HASH_DELETE(hh1, table->root_by_id, node);
|
||||
HASH_DELETE(hh2, table->root_by_addr, node);
|
||||
|
||||
if (node->val_freecb && node->val_data)
|
||||
{
|
||||
node->val_freecb(node->val_data);
|
||||
}
|
||||
|
||||
free(node);
|
||||
node = NULL;
|
||||
|
||||
table->session_node_count--;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
uint64_t session_table_count(struct session_table *table)
|
||||
{
|
||||
if (table)
|
||||
|
||||
@@ -235,6 +235,38 @@ TEST(STREAM_TABLE, DELETE_BY_REVERSE_ADDR)
|
||||
session_table_destory(table);
|
||||
}
|
||||
|
||||
TEST(STREAM_TABLE, RESET)
|
||||
{
|
||||
// TEST Create
|
||||
struct session_table *table = session_table_create();
|
||||
EXPECT_TRUE(table != nullptr);
|
||||
|
||||
char *val_hello = strdup("HELLO");
|
||||
char *val_world = strdup("WORLD");
|
||||
INIT_ADDR_V4(addr1, "1.2.3.4", 1234, "4.3.2.1", 4321);
|
||||
INIT_ADDR_V6(addr2, "2:3:4::5", 2345, "5:4:3::2", 5342);
|
||||
|
||||
// TEST Insert
|
||||
EXPECT_TRUE(session_table_insert(table, 1, &addr1, val_hello, free) == 0);
|
||||
EXPECT_TRUE(session_table_insert(table, 1, &addr1, val_hello, free) == -1);
|
||||
EXPECT_TRUE(session_table_count(table) == 1);
|
||||
|
||||
// TEST Reset
|
||||
session_table_reset(table);
|
||||
EXPECT_TRUE(session_table_search_by_id(table, 1) == nullptr);
|
||||
EXPECT_TRUE(session_table_search_by_addr(table, &addr1) == nullptr);
|
||||
EXPECT_TRUE(session_table_count(table) == 0);
|
||||
|
||||
EXPECT_TRUE(session_table_insert(table, 2, &addr2, val_world, free) == 0);
|
||||
EXPECT_TRUE(session_table_insert(table, 2, &addr2, val_world, free) == -1);
|
||||
EXPECT_TRUE(session_table_search_by_id(table, 2) != nullptr);
|
||||
EXPECT_TRUE(session_table_search_by_addr(table, &addr2) != nullptr);
|
||||
EXPECT_TRUE(session_table_count(table) == 1);
|
||||
|
||||
// TEST Destory
|
||||
session_table_destory(table);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
|
||||
Reference in New Issue
Block a user