Update mobile gaming groovy scripts

apache · Jan 29, 2025 · 288ab30 · 288ab30
1 parent 3d651eb
commit 288ab30
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 14 deletions.
diff --git a/release/src/main/groovy/mobilegaming-java-dataflow.groovy b/release/src/main/groovy/mobilegaming-java-dataflow.groovy
@@ -98,15 +98,20 @@ class LeaderBoardRunner {
     def isSuccess = false
     String query_result = ""
     while ((System.currentTimeMillis() - startTime) / 60000 < mobileGamingCommands.EXECUTION_TIMEOUT_IN_MINUTES) {
-      tables = t.run "bq query SELECT table_id FROM ${t.bqDataset()}.__TABLES_SUMMARY__"
-      if (tables.contains("leaderboard_${runner}_user") && tables.contains("leaderboard_${runner}_team")) {
-        query_result = t.run """bq query --batch "SELECT user FROM [${t.gcpProject()}:${
-          t.bqDataset()
-        }.leaderboard_${runner}_user] LIMIT 10\""""
-        if (t.seeAnyOf(mobileGamingCommands.COLORS, query_result)) {
-          isSuccess = true
-          break
+      try {
+        tables = t.run "bq query --use_legacy_sql=false SELECT table_name FROM ${t.bqDataset()}.INFORMATION_SCHEMA.TABLES"
+        if (tables.contains("leaderboard_${runner}_user") && tables.contains("leaderboard_${runner}_team")) {
+          query_result = t.run """bq query --batch "SELECT user FROM [${t.gcpProject()}:${
+            t.bqDataset()
+          }.leaderboard_${runner}_user] LIMIT 10\""""
+          if (t.seeAnyOf(mobileGamingCommands.COLORS, query_result)) {
+            isSuccess = true
+            break
+          }
         }
+      } catch (Exception e) {
+        println "Warning: Exception while checking tables: ${e.message}"
+        println "Retrying..."
       }
       println "Waiting for pipeline to produce more results..."
       sleep(60000) // wait for 1 min

diff --git a/release/src/main/groovy/mobilegaming-java-direct.groovy b/release/src/main/groovy/mobilegaming-java-direct.groovy
@@ -87,13 +87,18 @@ def startTime = System.currentTimeMillis()
 def isSuccess = false
 String query_result = ""
 while((System.currentTimeMillis() - startTime)/60000 < mobileGamingCommands.EXECUTION_TIMEOUT_IN_MINUTES) {
-  tables = t.run "bq query SELECT table_id FROM ${t.bqDataset()}.__TABLES_SUMMARY__"
-  if(tables.contains("leaderboard_${runner}_user") && tables.contains("leaderboard_${runner}_team")){
-    query_result = t.run """bq query --batch "SELECT user FROM [${t.gcpProject()}:${t.bqDataset()}.leaderboard_${runner}_user] LIMIT 10\""""
-    if(t.seeAnyOf(mobileGamingCommands.COLORS, query_result)){
-      isSuccess = true
-      break
+  try {
+    tables = t.run "bq query --use_legacy_sql=false SELECT table_name FROM ${t.bqDataset()}.INFORMATION_SCHEMA.TABLES"
+    if(tables.contains("leaderboard_${runner}_user") && tables.contains("leaderboard_${runner}_team")) {
+      query_result = t.run """bq query --batch "SELECT user FROM [${t.gcpProject()}.${t.bqDataset()}.leaderboard_${runner}_user] LIMIT 10\""""
+      if(t.seeAnyOf(mobileGamingCommands.COLORS, query_result)){
+        isSuccess = true
+        break
+      }
     }
+  } catch (Exception e) {
+    println "Warning: Exception while checking tables: ${e.message}"
+    println "Retrying..."
   }
   println "Waiting for pipeline to produce more results..."
   sleep(60000) // wait for 1 min