From fba248889c9ed4fc47f82afca5d213096c229ce4 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Tue, 29 Nov 2016 14:56:19 +0000
Subject: [PATCH] [Fix] Miltiple fixes to fann_redis module

---
 src/plugins/lua/fann_redis.lua | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/src/plugins/lua/fann_redis.lua b/src/plugins/lua/fann_redis.lua
index df995ee51..d75d1ec1c 100644
--- a/src/plugins/lua/fann_redis.lua
+++ b/src/plugins/lua/fann_redis.lua
@@ -350,7 +350,7 @@ local function load_or_invalidate_fann(data, id, ev_base)
       end
     end
     -- Invalidate ANN
-    rspamd_logger.infox('invalidate ANN %s')
+    rspamd_logger.infox(rspamd_config, 'invalidate ANN %s', id)
     redis_make_request(ev_base,
       rspamd_config,
       nil,
@@ -459,7 +459,7 @@ local function train_fann(_, ev_base, elt)
       redis_make_request(ev_base,
         rspamd_config,
         nil,
-        false, -- is write
+        true, -- is write
         redis_unlock_cb, --callback
         'DEL', -- command
         {fann_prefix .. elt .. '_lock'}
@@ -489,17 +489,17 @@ local function train_fann(_, ev_base, elt)
       redis_make_request(ev_base,
         rspamd_config,
         nil,
-        false, -- is write
+        true, -- is write
         redis_unlock_cb, --callback
         'DEL', -- command
         {fann_prefix .. elt .. '_lock'}
       )
     else
       -- Decompress and convert to numbers each training vector
-      ham_elts = fun.map(function(tok)
+      ham_elts = fun.totable(fun.map(function(tok)
         local _,str = rspamd_util.zstd_decompress(tok)
-        return fun.map(tonumber, rspamd_str_split(tostring(str), ';'))
-      end, data)
+        return fun.totable(fun.map(tonumber, rspamd_str_split(tostring(str), ';')))
+      end, data))
 
       -- Now we need to join inputs and create the appropriate test vectors
       local inputs = {}
@@ -511,9 +511,9 @@ local function train_fann(_, ev_base, elt)
       end
 
       fun.each(function(spam_sample, ham_sample)
-        table.insert(inputs, fun.totable(spam_sample))
+        table.insert(inputs, spam_sample)
         table.insert(outputs, {1.0})
-        table.insert(inputs, fun.totable(ham_sample))
+        table.insert(inputs, ham_sample)
         table.insert(outputs, {-1.0})
       end, fun.zip(fun.filter(filt, spam_elts), fun.filter(filt, ham_elts)))
 
@@ -560,17 +560,17 @@ local function train_fann(_, ev_base, elt)
       redis_make_request(ev_base,
         rspamd_config,
         nil,
-        false, -- is write
+        true, -- is write
         redis_unlock_cb, --callback
         'DEL', -- command
         {fann_prefix .. elt .. '_lock'}
       )
     else
       -- Decompress and convert to numbers each training vector
-      spam_elts = fun.map(function(tok)
+      spam_elts = fun.totable(fun.map(function(tok)
         local _,str = rspamd_util.zstd_decompress(tok)
-        return fun.map(tonumber, rspamd_str_split(tostring(str), ';'))
-      end, data)
+        return fun.totable(fun.map(tonumber, rspamd_str_split(tostring(str), ';')))
+      end, data))
       redis_make_request(ev_base,
         rspamd_config,
         nil,
-- 
2.39.5