Use transactions for importing logs

1 files changed, 4 insertions, 1 deletions

diff --git a/weblog-model.lua b/weblog-model.lua
index a33d97c..70aff4e 100644
--- a/weblog-model.lua
+++ b/weblog-model.lua
@@ -179,27 +179,30 @@ local listhistorylogentries = function()
 end
 
 local importsquidlog = function(logentries, sourcename)
+	con:execute("START TRANSACTION")
 	for i,entry in pairs(logentries) do
 		local sql = string.format("INSERT INTO weblog VALUES ('%s', '%s', '%s', '%s', '%s', '%s')",
 			escape(sourcename), escape(entry.clientip), escape(entry.clientuserid:lower()),
 			escape(entry.logdatetime), escape(entry.URL), escape(entry.bytes))
 		local res = assert (con:execute(sql))
 	end
+	con:execute("COMMIT")
 end
 
 local importdglog = function(logentries, sourcename)
+	con:execute("START TRANSACTION")
 	for i,entry in pairs(logentries) do
 		local sql = string.format("INSERT INTO blocklog VALUES ('%s', '0.0.0.0', '%s', '%s', '%s', '%s', '%s', '%s', '%s')",
 			escape(sourcename), escape(entry.clientuserid:lower()), escape(entry.logdatetime), escape(entry.URL),
 			escape(entry.bytes), escape(entry.reason), escape(entry.score or "0"), escape(entry.shortreason))
 		local res = assert (con:execute(sql))
 	end
+	con:execute("COMMIT")
 end
 
 local listsourceentries = function(sourcename)
 	local sources = {}
 	-- retrieve a cursor
---	cur = assert (con:execute"SELECT sourcename, userid, passwd, source, tzislocal FROM source WHERE method='http' and enabled=true")
 	local sql = "SELECT sourcename, method, userid, passwd, source, tzislocal, enabled FROM source"
 	if sourcename then
 		sql = sql .. " WHERE sourcename='" .. escape(sourcename) .. "'"