denozord/strongswan-ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pool: Change transaction handling

Browse Source
This commit is contained in:
Tobias Brunner
2013-10-11 15:16:05 +02:00
parent ec6ad6b086
commit 03c801cb2b
1 changed files with 8 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pool/pool.c
+8 -46
View File
@@ -51,41 +51,6 @@ bool replace_pool = FALSE;
static void del(char *name);
static void do_args(int argc, char *argv[]);
/**
* nesting counter for database transaction functions
*/
int nested_transaction = 0;
/**
* start a database transaction
*/
static void begin_transaction()
{
if (db->get_driver(db) == DB_SQLITE)
{
if (!nested_transaction)
{
db->execute(db, NULL, "BEGIN EXCLUSIVE TRANSACTION");
}
++nested_transaction;
}
}
/**
* commit a database transaction
*/
static void commit_transaction()
{
if (db->get_driver(db) == DB_SQLITE)
{
--nested_transaction;
if (!nested_transaction)
{
db->execute(db, NULL, "END TRANSACTION");
}
}
}
/**
* Create or replace a pool by name
*/
@@ -370,8 +335,7 @@ static void add(char *name, host_t *start, host_t *end, int timeout)
id = create_pool(name, start_addr, end_addr, timeout);
printf("allocating %d addresses... ", count);
fflush(stdout);
/* run population in a transaction for sqlite */
begin_transaction();
db->transaction(db);
while (TRUE)
{
db->execute(db, NULL,
@@ -384,7 +348,7 @@ static void add(char *name, host_t *start, host_t *end, int timeout)
}
chunk_increment(cur_addr);
}
commit_transaction();
db->commit(db);
printf("done.\n");
}
@@ -449,8 +413,7 @@ static void add_addresses(char *pool, char *path, int timeout)
host_t *addr;
FILE *file;
/* run population in a transaction for sqlite */
begin_transaction();
db->transaction(db);
addr = host_create_from_string("%any", 0);
pool_id = create_pool(pool, addr->get_address(addr),
@@ -510,7 +473,7 @@ static void add_addresses(char *pool, char *path, int timeout)
addr->destroy(addr);
}
commit_transaction();
db->commit(db);
printf("%d addresses done.\n", count);
}
@@ -596,6 +559,7 @@ static void resize(char *name, host_t *end)
}
DESTROY_IF(old_end);
db->transaction(db);
if (db->execute(db, NULL,
"UPDATE pools SET end = ? WHERE name = ?",
DB_BLOB, new_addr, DB_TEXT, name) <= 0)
@@ -606,8 +570,6 @@ static void resize(char *name, host_t *end)
printf("allocating %d new addresses... ", count);
fflush(stdout);
/* run population in a transaction for sqlite */
begin_transaction();
while (count-- > 0)
{
chunk_increment(cur_addr);
@@ -616,7 +578,7 @@ static void resize(char *name, host_t *end)
"VALUES (?, ?, ?, ?, ?)",
DB_UINT, id, DB_BLOB, cur_addr, DB_UINT, 0, DB_UINT, 0, DB_UINT, 1);
}
commit_transaction();
db->commit(db);
printf("done.\n");
}
@@ -900,7 +862,7 @@ static void batch(char *argv0, char *name)
exit(EXIT_FAILURE);
}
begin_transaction();
db->transaction(db);
while (fgets(command, sizeof(command), file))
{
char *argv[ARGV_SIZE], *start;
@@ -939,7 +901,7 @@ static void batch(char *argv0, char *name)
do_args(argc, argv);
}
commit_transaction();
db->commit(db);
if (file != stdin)
{