X-Git-Url: https://jfr.im/git/solanum.git/blobdiff_plain/832ed81ad97ebb44edbd9d61bfb5e28e9f073c32..da1b891f9fae341aa97fb22d9af8c2c4c3d6210d:/bandb/bandb.c

diff --git a/bandb/bandb.c b/bandb/bandb.c
index cce310c4..33166b1f 100644
--- 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);
 }