Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FLINK-16949] Enhance AbstractStreamOperatorTestHarness to use customized TtlTimeProvider #11624

Merged

Conversation

Myasuka
Copy link
Member

@Myasuka Myasuka commented Apr 2, 2020

What is the purpose of the change

Enhance specific UTs to use customized TtlTimeProvider to simulate changed current time. This would introduce some changes to AbstractStreamOperatorTestHarness and add new StreamTaskStateInitializerTestImpl for different state backends.

Brief change log

Introduce some changes to AbstractStreamOperatorTestHarness and add new StreamTaskStateInitializerTestImpl for different state backends.

Verifying this change

This change added tests and can be verified as follows:

AbstractStreamOperatorTestHarnessTest#testSetTtlTimeProvider

Does this pull request potentially affect one of the following parts:

  • Dependencies (does it add or upgrade a dependency): no
  • The public API, i.e., is any changed class annotated with @Public(Evolving): no
  • The serializers: no
  • The runtime per-record code paths (performance sensitive): no
  • Anything that affects deployment or recovery: JobManager (and its components), Checkpointing, Kubernetes/Yarn/Mesos, ZooKeeper: no
  • The S3 file system connector: no

Documentation

  • Does this pull request introduce a new feature? no
  • If yes, how is the feature documented? not applicable

@flinkbot
Copy link
Collaborator

flinkbot commented Apr 2, 2020

Thanks a lot for your contribution to the Apache Flink project. I'm the @flinkbot. I help the community
to review your pull request. We will use this comment to track the progress of the review.

Automated Checks

Last check on commit 6687976 (Thu Apr 02 18:53:18 UTC 2020)

Warnings:

  • No documentation files were touched! Remember to keep the Flink docs up to date!
  • This pull request references an unassigned Jira ticket. According to the code contribution guide, tickets need to be assigned before starting with the implementation work.

Mention the bot in a comment to re-run the automated checks.

Review Progress

  • ❓ 1. The [description] looks good.
  • ❓ 2. There is [consensus] that the contribution should go into to Flink.
  • ❓ 3. Needs [attention] from.
  • ❓ 4. The change fits into the overall [architecture].
  • ❓ 5. Overall code [quality] is good.

Please see the Pull Request Review Guide for a full explanation of the review process.


The Bot is tracking the review progress through labels. Labels are applied according to the order of the review items. For consensus, approval by a Flink committer of PMC member is required Bot commands
The @flinkbot bot supports the following commands:

  • @flinkbot approve description to approve one or more aspects (aspects: description, consensus, architecture and quality)
  • @flinkbot approve all to approve all aspects
  • @flinkbot approve-until architecture to approve everything until architecture
  • @flinkbot attention @username1 [@username2 ..] to require somebody's attention
  • @flinkbot disapprove architecture to remove an approval you gave earlier

@flinkbot
Copy link
Collaborator

flinkbot commented Apr 2, 2020

CI report:

Bot commands The @flinkbot bot supports the following commands:
  • @flinkbot run travis re-run the last Travis build
  • @flinkbot run azure re-run the last Azure build

@@ -258,6 +258,10 @@ protected OperatorStateBackend operatorStateBackend(
}
}

protected TtlTimeProvider getTtlTimeProvider() {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why not to inject this in the constructor of StreamTaskStateInitializerImpl?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I just don't want to introduce too many changes here. If we plan to inject this as a filed, I am in favor of moving TtlTimeProvider at operator level (by changing StreamTaskStateInitializer#streamOperatorStateContext) but not in the constructor which was FLINK-14156 fixed.

Copy link
Contributor

@azagrebin azagrebin Apr 3, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think either approach can be refactored in future.
Using constructor is the usual way how we inject dependencies for tests. The inheritance is usually used to change more complicated internal behaviour.
You could also keep the current constructor and add another one annotated with @VisibleForTesting which could accept the custom TtlTimeProvider field.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

There existed another problem why we not change the constructor of StreamTaskStateInitializerImpl.
Current AbstractStreamOperatorTestHarness is not created from a builder, and once a new AbstractStreamOperatorTestHarness is created, the inner streamTaskStateInitializer has been created with the default TtlTimeProvider. Even we set ttl time provider to AbstractStreamOperatorTestHarness later, the inner streamTaskStateInitializer would not notice the changed ttl time provider unless we call AbstractStreamOperatorTestHarness#setup to re-create the inner streamTaskStateInitializer.
However, AbstractStreamOperatorTestHarness#setup actually call a deprecated SetupableStreamOperator#setup interface.
In a nutshell, unless we refactor how we build AbstractStreamOperatorTestHarness, to make the customized ttl time provider take effect, we must call AbstractStreamOperatorTestHarness#setup each time which might already be treated as a deprecated interface.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Then I think it makes sense to consider builders to avoid adding more harness constructors.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why not AbstractStreamOperatorTestHarness always use a mocked TtlTimeProvider? Just like always use the TestProcessingTimeService. Then we don't need to re-create the inner streamTaskStateInitializer.

I also prefer the way to set processing time on the harness object AbstractStreamOperatorTestHarness#setProcessingTime instead of on other object mockTtlTimeProvider.setCurrentTimeStamp(0L). We can provide a method setStateTtlTime() on the AbstractStreamOperatorTestHarness too.

Copy link
Contributor

@azagrebin azagrebin Apr 10, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This would indeed work. This is not flexible in general but may be good enough and we already have it for processing time.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think @wuchong 's suggestion is also a available choice. However, I think FLINK-17011 is the most clean solution to clean up these test code. Although I have to admit that PR #11676 is a bit large for reviewing.

Copy link
Member

@wuchong wuchong Apr 10, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What about going with this approach first? This can be merged faster and unblock other works. We can also continue the cleanup work using new builders after that, and expose custom time provider in builders to make it more flexible.

Copy link
Member

@wuchong wuchong left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the updating @Myasuka , LGTM. Just left a minor comment.

wuchong pushed a commit to wuchong/flink that referenced this pull request Apr 11, 2020
@wuchong
Copy link
Member

wuchong commented Apr 12, 2020

Merging...

@wuchong wuchong merged commit afb4645 into apache:master Apr 12, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
5 participants