From 5722cfb5741b4644fe4324164f00f5495beacaa8 Mon Sep 17 00:00:00 2001
From: Thibault Duplessis <t@lichess.org>
Date: Wed, 27 Jan 2021 22:38:55 +0100
Subject: [PATCH] remember which puzzles were selected for storm runs

---
 modules/storm/src/main/StormSelector.scala | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/modules/storm/src/main/StormSelector.scala b/modules/storm/src/main/StormSelector.scala
index d86d13bc66..c85a324949 100644
--- a/modules/storm/src/main/StormSelector.scala
+++ b/modules/storm/src/main/StormSelector.scala
@@ -136,6 +136,13 @@ final class StormSelector(colls: PuzzleColls, cacheApi: CacheApi)(implicit ec: E
           lila.mon.storm.selector.ratingSlice(i).record(r.toInt)
         }
       }
+      colls.puzzle {
+        _.update.one(
+          $inIds(puzzles.map(_.id.value)),
+          $inc("storm" -> 1),
+          multi = true
+        )
+      }.unit
     }
   }
 }