feat(jdbc): force using the index on mysql poll queries (#1016)

kestra-io · Feb 24, 2023 · 092230e · 092230e
1 parent 0d02517
commit 092230e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jdbc-mysql/src/main/java/io/kestra/runner/mysql/MysqlQueue.java b/jdbc-mysql/src/main/java/io/kestra/runner/mysql/MysqlQueue.java
@@ -43,7 +43,8 @@ protected Result<Record> receiveFetch(DSLContext ctx, String consumerGroup) {
                 AbstractJdbcRepository.field("value"),
                 AbstractJdbcRepository.field("offset")
             )
-            .from(this.table)
+            // force using the dedicated index, or it made a scan of the PK index
+            .from(this.table.useIndex("ix_type__consumers"))
             .where(AbstractJdbcRepository.field("type").eq(this.cls.getName()))
             .and(DSL.or(List.of(
                 AbstractJdbcRepository.field("consumers").isNull(),