ShadowIRCd 6.2.0-beta1

[irc/rqf/shadowircd.git] / bandb / bandb.c

diff --git a/bandb/bandb.c b/bandb/bandb.c
index cce310c4bf9d1350a8b8814950ac71cff49a58c4..33166b1f377b799fbea1d6fa0be29e379a91e60f 100644 (file)
--- a/bandb/bandb.c
+++ b/bandb/bandb.c
@@ -38,6 +38,8 @@
 
 #define MAXPARA 10
 
+#define COMMIT_INTERVAL 3 /* seconds */
+
 typedef enum
 {
        BANDB_KLINE,
@@ -57,9 +59,17 @@ static const char *bandb_table[LAST_BANDB_TYPE] = {
 
 
 static rb_helper *bandb_helper;
+static int in_transaction;
 
 static void check_schema(void);
 
+static void
+bandb_commit(void *unused)
+{
+       rsdb_transaction(RSDB_TRANS_END);
+       in_transaction = 0;
+}
+
 static void
 parse_ban(bandb_type type, char *parv[], int parc)
 {
@@ -89,6 +99,14 @@ parse_ban(bandb_type type, char *parv[], int parc)
        perm = parv[para++];
        reason = parv[para++];
 
+       if(!in_transaction)
+       {
+               rsdb_transaction(RSDB_TRANS_START);
+               in_transaction = 1;
+               rb_event_addonce("bandb_commit", bandb_commit, NULL,
+                               COMMIT_INTERVAL);
+       }
+
        rsdb_exec(NULL,
                  "INSERT INTO %s (mask1, mask2, oper, time, perm, reason) VALUES('%Q', '%Q', '%Q', %s, %s, '%Q')",
                  bandb_table[type], mask1, mask2 ? mask2 : "", oper, curtime, perm, reason);
@@ -113,6 +131,14 @@ parse_unban(bandb_type type, char *parv[], int parc)
        if(type == BANDB_KLINE)
                mask2 = parv[2];
 
+       if(!in_transaction)
+       {
+               rsdb_transaction(RSDB_TRANS_START);
+               in_transaction = 1;
+               rb_event_addonce("bandb_commit", bandb_commit, NULL,
+                               COMMIT_INTERVAL);
+       }
+
        rsdb_exec(NULL, "DELETE FROM %s WHERE mask1='%Q' AND mask2='%Q'",
                  bandb_table[type], mask1, mask2 ? mask2 : "");
 }
@@ -215,6 +241,8 @@ parse_request(rb_helper *helper)
 static void
 error_cb(rb_helper *helper)
 {
+       if(in_transaction)
+               rsdb_transaction(RSDB_TRANS_END);
        exit(1);
 }