[Project] Http_stat: Notice statfiles when creating runtime

diff --git a/src/libstat/backends/http_backend.cxx b/src/libstat/backends/http_backend.cxx
index 51ac6e0d83466a8988b6521f9453f94c2d4edb06..eaf2b18045fe0a8fda711611b40d56cfdf3c9c14 100644 (file)
--- a/src/libstat/backends/http_backend.cxx
+++ b/src/libstat/backends/http_backend.cxx
@@ -80,9 +80,13 @@ private:
 class http_backend_runtime final {
 public:
        static auto create(struct rspamd_task *task, bool is_learn) -> http_backend_runtime *;
+       /* Add a new statfile with a specific id to the list of statfiles */
+       auto notice_statfile(int id, const struct rspamd_statfile_config *st) -> void {
+               seen_statfiles[id] = st;
+       }
 private:
        http_backends_collection *all_backends;
-       robin_hood::unordered_flat_map<int, struct rspamd_statfile *> seen_statfiles;
+       robin_hood::unordered_flat_map<int, const struct rspamd_statfile_config *> seen_statfiles;
        struct upstream *selected;
 private:
        http_backend_runtime(struct rspamd_task *task, bool is_learn) :
@@ -249,17 +253,21 @@ rspamd_http_runtime(struct rspamd_task* task,
                                        struct rspamd_statfile_config* stcf,
                                        gboolean learn,
                                        gpointer ctx,
-                                       gint _id)
+                                       gint id)
 {
        auto maybe_existing = rspamd_mempool_get_variable(task->task_pool, RSPAMD_MEMPOOL_HTTP_STAT_BACKEND_RUNTIME);
 
        if (maybe_existing != nullptr) {
+               auto real_runtime = (rspamd::stat::http::http_backend_runtime *)maybe_existing;
+               real_runtime->notice_statfile(id, stcf);
+
                return maybe_existing;
        }
 
        auto runtime = rspamd::stat::http::http_backend_runtime::create(task, learn);
 
        if (runtime) {
+               runtime->notice_statfile(id, stcf);
                rspamd_mempool_set_variable(task->task_pool, RSPAMD_MEMPOOL_HTTP_STAT_BACKEND_RUNTIME,
                                (void *)runtime, nullptr);
        }