apache / apache/beam

GBK Python streaming load tests are too slow

Open
#20,403 0 comments 0 reactions 0 assignees View on GitHub
bug core dataflow P3 python runners tests
Dominant language
Java
Stars
8.7k
Forks
4.7k
Avg merge
1d 20h
Merged PRs (30d)
196

Description

The following GBK streaming test cases take too long on Dataflow:

 

1) 2GB of 10B records

2) 2GB of 100B records

4) fanout 4 times with 2GB 10-byte records total

5) fanout 8 times with 2GB 10-byte records total

 

Each of them needs at least 1 hour to execute, which is way too long for one Jenkins job.

Job's definition: [https://github.com/apache/beam/blob/master/.test-infra/jenkins/job_LoadTests_GBK_Python.groovy](https://github.com/apache/beam/blob/master/.test-infra/jenkins/job_LoadTests_GBK_Python.groovy)

Test pipeline: [https://github.com/apache/beam/blob/master/sdks/python/apache_beam/testing/load_tests/group_by_key_test.py](https://github.com/apache/beam/blob/master/sdks/python/apache_beam/testing/load_tests/group_by_key_test.py)

It is probable that those cases are too extreme. The first two cases involve grouping 20M unique keys, which is a stressful operation. A solution might be to overhaul the cases so that they would be less complex.

Both the current production Dataflow runner and the new Dataflow Runner V2 were tested.

Imported from Jira [BEAM-10774](https://issues.apache.org/jira/browse/BEAM-10774). Original Jira may contain additional context.
Reported by: kamilwu.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.