IMPALA-14523: JDBC Scan Parallelism with Shared Cursor Fetching

Impala’s current JDBC DataSourceScan creates a separate JDBC
connection per scanner thread, but since JDBC cursors are
inherently sequential, this limits effective parallelism and
introduces unnecessary connection and JNI overhead. As a result,
even with MT_DOP enabled, scans do not scale efficiently and can
suffer from poor throughput for large datasets.

This patch introduces a shared JDBC connection model per fragment,
allowing multiple scanner threads to cooperatively fetch from a
single cursor. A new SharedJdbcConnection abstraction is added and
owned by DataSourceScanPlanNode, ensuring the connection is opened
and closed exactly once. Multiple DataSourceScanNode instances
invoke FetchBatch() concurrently, where cursor movement is
serialized while row materialization happens in parallel, and EOS
is tracked via an atomic flag to avoid redundant fetches.

Planner and scheduler changes ensure all JDBC scan ranges are
pinned to a single executor using a new is_data_source_scan flag,
and one virtual scan range is generated per MT_DOP instance. On the
Java side, batch fetching with minimal locking is introduced, and
JdbcDataSource is updated for safe concurrent access and improved
materialization, reducing overhead and improving scan throughput.

For instance, for query:
WITH recent_orders AS (
  SELECT o.o_orderkey, o.o_custkey, o.o_totalprice, o.o_orderdate
  FROM tpch_jdbc.orders o WHERE o.o_orderdate >= '1995-01-01'
)
SELECT c.c_custkey, c.c_name, c.c_mktsegment,
       COUNT(ro.o_orderkey) AS num_orders,
       SUM(ro.o_totalprice) AS total_spent,
       AVG(ro.o_totalprice) AS avg_order_value,
       RANK() OVER (ORDER BY SUM(ro.o_totalprice) DESC) AS spend_rank,
       CASE
         WHEN SUM(ro.o_totalprice) > 500000 THEN 'VIP'
         WHEN SUM(ro.o_totalprice) > 100000 THEN 'Premium'
         ELSE 'Regular'
       END AS customer_tier,
       s.s_name AS supplier_name,
       s.s_acctbal AS supplier_balance
FROM tpch_jdbc.customer c
JOIN recent_orders ro
     ON c.c_custkey = ro.o_custkey
JOIN tpch_jdbc.supplier s
     ON c.c_nationkey = s.s_nationkey
GROUP BY c.c_custkey, c.c_name, c.c_mktsegment, s.s_name, s.s_acctbal
ORDER BY total_spent DESC
LIMIT 5;

It takes around 260 sec with mt_dop =1 while with mt_dop =7,
it takes close to 77 sec.

Also added JdbcJniWaitTime, JdbcCursorFetchTime,
JdbcLockWaitTime and JdbcJniCallCount for better analysis
of jdbc queries, in query profile.

Testing:
Some planner tests are included in jdbc-parallel.test and benchmark
tests are added in test_ext_data_sources.py.

Generated by:
Some codes and comments are generated with
the help of claude-4.6-sonnet-medium-thinking.

Change-Id: I3b25b99f5cb77d32c111ba37c0a01378ffdc1107
Reviewed-on: http://gerrit.cloudera.org:8080/24133
Tested-by: Impala Public Jenkins <impala-public-jenkins@cloudera.com>
Reviewed-by: Zoltan Borok-Nagy <boroknagyz@cloudera.com>
13 files changed
tree: 567ad5fea7633c2146d647016f8b9c15d4803ddd
  1. .devcontainer/
  2. be/
  3. bin/
  4. cmake_modules/
  5. common/
  6. docker/
  7. docs/
  8. fe/
  9. infra/
  10. java/
  11. lib/
  12. package/
  13. security/
  14. shell/
  15. ssh_keys/
  16. testdata/
  17. tests/
  18. www/
  19. .asf.yaml
  20. .clang-format
  21. .clang-tidy
  22. .gitattributes
  23. .gitignore
  24. .isort.cfg
  25. buildall.sh
  26. CMakeLists.txt
  27. EXPORT_CONTROL.md
  28. LICENSE.txt
  29. LOGS.md
  30. NOTICE.txt
  31. README-build.md
  32. README.md
  33. setup.cfg
README.md

Welcome to Impala

Lightning-fast, distributed SQL queries for petabytes of data stored in open data and table formats.

Impala is a modern, massively-distributed, massively-parallel, C++ query engine that lets you analyze, transform and combine data from a variety of data sources:

More about Impala

The fastest way to try out Impala is a quickstart Docker container. You can try out running queries and processing data sets in Impala on a single machine without installing dependencies. It can automatically load test data sets into Apache Kudu and Apache Parquet formats and you can start playing around with Apache Impala SQL within minutes.

To learn more about Impala as a user or administrator, or to try Impala, please visit the Impala homepage. Detailed documentation for administrators and users is available at Apache Impala documentation.

If you are interested in contributing to Impala as a developer, or learning more about Impala's internals and architecture, visit the Impala wiki.

Supported Platforms

Impala only supports Linux at the moment. Impala supports x86_64 and has experimental support for arm64 (as of Impala 4.0). Impala Requirements contains more detailed information on the minimum CPU requirements.

Supported OS Distributions

Impala runs on Linux systems only. The supported distros are

  • Ubuntu 16.04/18.04
  • CentOS/RHEL 7/8

Other systems, e.g. SLES12, may also be supported but are not tested by the community.

Export Control Notice

This distribution uses cryptographic software and may be subject to export controls. Please refer to EXPORT_CONTROL.md for more information.

Build Instructions

See Impala's developer documentation to get started.

Detailed build notes has some detailed information on the project layout and build.