jprdonnelly/kestra
1
0
Fork 0
mirror of https://github.com/kestra-io/kestra.git synced 2025-12-23 21:04:39 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: disable seqscan on Postgres

Browse Source
This commit is contained in:
Loïc Mathieu
2023-06-19 09:35:45 +02:00
parent 467003be0f
commit 87b7912607
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
jdbc-postgres/src/main/java/io/kestra/runner/postgres/PostgresQueue.java
View File

@@ -12,9 +12,18 @@ import org.jooq.impl.DSL;
import java.util.List;
import java.util.Map;
import static org.jooq.impl.DSL.*;
public class PostgresQueue<T> extends JdbcQueue<T> {
private boolean disableSeqScan = false;
public PostgresQueue(Class<T> cls, ApplicationContext applicationContext) {
super(cls, applicationContext);
var maybeDisableSeScan = applicationContext.getProperty("kestra.queue.postgres.disable-seq-scan", Boolean.class);
if (maybeDisableSeScan.isPresent() && maybeDisableSeScan.get()) {
disableSeqScan = true;
}
}
@Override
@@ -65,6 +74,10 @@ public class PostgresQueue<T> extends JdbcQueue<T> {
@Override
protected Result<Record> receiveFetch(DSLContext ctx, String consumerGroup, String queueType) {
if (disableSeqScan) {
ctx.setLocal(name("enable_seqscan"), val("off")).execute();
}
var select = ctx.select(
AbstractJdbcRepository.field("value"),
AbstractJdbcRepository.field("offset")