create equal number of captchas for each parameter combination in background task

Signed-off-by: hrj <harshad.rj@gmail.com>

src/main/scala/lc/background/taskThread.scala

@@ -18,22 +18,33 @@ class BackgroundTask(config: Config, captchaManager: CaptchaManager) {
         val challengeGCPstmt = Statements.tlStmts.get.challengeGCPstmt
         challengeGCPstmt.executeUpdate()
 
-        val imageNumResult = Statements.tlStmts.get.getCountChallengeTable.executeQuery()
-        val imageNum = if (imageNumResult.next()) {
-          imageNumResult.getInt("total")
-        } else {
-          0
+        for (param <- allParameterCombinations()) {
+          val imageNum = captchaManager.getCount(param).getOrElse(0)
+          val countCreate = (config.throttle * 1.1).toInt - imageNum
+          if (countCreate > 0) {
+            println(s"Creating $countCreate captchas for $param")
+
+            for (i <- 0 until countCreate) {
+              captchaManager.generateChallenge(param)
+            }
           }
-
-        val throttle = (config.throttle * 1.1).toInt - imageNum
-
-        for (i <- 0 until throttle) {
-          captchaManager.generateChallenge(getRandomParam())
         }
       } catch { case exception: Exception => println(exception) }
     }
   }
 
+  private def allParameterCombinations(): List[Parameters] = {
+    (config.captchaConfig).flatMap {captcha =>
+      (captcha.allowedLevels).flatMap {level =>
+        (captcha.allowedMedia).flatMap {media =>
+          (captcha.allowedInputType).map {inputType =>
+            Parameters(level, media, inputType, Some(Size(0, 0)))
+          }
+        }
+      }
+    }
+  }
+
   private def getRandomParam(): Parameters = {
     val captcha = pickRandom(config.captchaConfig)
     val level = pickRandom(captcha.allowedLevels)

src/main/scala/lc/core/captchaManager.scala

@@ -105,8 +105,8 @@ class CaptchaManager(config: Config, captchaProviders: CaptchaProviders) {
     }
   }
 
-  private def getCount(param: Parameters): Option[Int] = {
-    val countPstmt = Statements.tlStmts.get.countPstmt
+  def getCount(param: Parameters): Option[Int] = {
+    val countPstmt = Statements.tlStmts.get.countForParameterPstmt
     countPstmt.setString(1, param.level)
     countPstmt.setString(2, param.media)
     countPstmt.setString(3, param.input_type)

src/main/scala/lc/database/statements.scala

@@ -70,7 +70,7 @@ class Statements(dbConn: DBConn, maxAttempts: Int) {
       "WHERE token = ?;"
   )
 
-  val countPstmt: PreparedStatement = dbConn.con.prepareStatement(
+  val countForParameterPstmt: PreparedStatement = dbConn.con.prepareStatement(
     s"""
       SELECT count(*) as count
         FROM challenge