[FLINK-38804][runtime] Ensure channelStateWriter is closed after the inputGates

[Efrat19]

What is the purpose of the change

Ensure the RemoteInputChannel (or remote components) is closed before ChannelStatePersister and ChannelStateWriteRequestExecutorImpl.
Dueing TaskManager shutdown sequence, ChannelStateWriteRequestExecutorImpl is being closed prematurely while RemoteInputChannel is still active. If a new buffer arrives, the active RemoteInputChannel attempts to use the already-closed executor, resulting in the failure.

Brief change log

Bubble ChannelStateWriter up from the StreamTask to Task, and close it after the InputGates.

Verifying this change

This change added tests and can be verified as follows:

• The ChannelStateWriter is closed at the Task level, in both completion and failure.

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, 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]

CI report:

• bf9abb7 Azure: SUCCESS

Bot commands

The @flinkbot bot supports the following commands:

• @flinkbot run azure re-run the last Azure build

[Efrat19]

@flinkbot run azure

[1996fanrui]

Thanks @Efrat19 for the fix!

Overall LGTM, I only left some minor comments, please take a look when you are available, thanks

[Efrat19]

@flinkbot run azure

[1996fanrui]

LGTM assuming CI is green