*: fix ctrl+c can not kill tidb during statistics init

[tiancaiamao]

What problem does this PR solve?

Issue Number: close #54589

Problem Summary:

When there a a lot of tables, statistics initialize takes a long time.
For example, 1M tables takes 13-14min with infoschema v2 currently.

When TiDB start up and initialize statistics, it does not handle any interrupt signal.
So if I start tidb and then Ctrl+C, domain.Close will wait the background worker to exit.
But the statistics load worker are handling the statistics initialization, it never response to the exit signal.
So tidb can't exit.

What changed and how does it work?

Add a context.Context to the related statistics code and handle context.Cancel signal.

Check List

Tests

• Unit test
• Integration test
• Manual test (add detailed scripts or steps below)

Before this commit, the log is this:

[2024/07/11 16:25:27.006 +08:00] [INFO] [domain.go:1164] ["domain closed"] ["take time"=12m54.727120057s]

And after the fix, tidb can exit quickly:

[2024/07/12 10:42:00.562 +08:00] [INFO] [wait_group_wrapper.go:140] ["background process exited"] [source=domain] [process=loadStatsWorker]
[2024/07/12 10:42:02.552 +08:00] [INFO] [manager.go:326] ["revoke session"] ["owner info"="[bindinfo] /tidb/bindinfo/owner ownerManager 906be1c7-194e-4bb4-a742-09a9a4db15de"] [error="rpc error: code = Canceled desc = grpc: the client connection is closing"]
[2024/07/12 10:42:02.552 +08:00] [INFO] [domain.go:1984] ["globalBindHandleWorkerLoop exited."]
[2024/07/12 10:42:02.552 +08:00] [INFO] [wait_group_wrapper.go:140] ["background process exited"] [source=domain] [process=globalBindHandleWorkerLoop]
[2024/07/12 10:42:02.574 +08:00] [INFO] [manager.go:326] ["revoke session"] ["owner info"="[stats] /tidb/stats/owner ownerManager 906be1c7-194e-4bb4-a742-09a9a4db15de"] [error="rpc error: code = Canceled desc = grpc: the client connection is closing"]
[2024/07/12 10:42:02.574 +08:00] [INFO] [domain.go:2509] ["updateStatsWorker exit preprocessing finished"]
[2024/07/12 10:42:02.574 +08:00] [INFO] [domain.go:2552] ["updateStatsWorker exited."]
[2024/07/12 10:42:02.574 +08:00] [INFO] [wait_group_wrapper.go:140] ["background process exited"] [source=domain] [process=updateStatsWorker]

[2024/07/12 10:42:02.575 +08:00] [INFO] [domain.go:1168] ["domain closed"] ["take time"=2.55484294s]

[2024/07/12 10:42:02.575 +08:00] [INFO] [gc_worker.go:223] [quit] [category="gc worker"] [uuid=6429345da3c0005]
[2024/07/12 10:42:02.575 +08:00] [INFO] [pd_service_discovery.go:550] ["[pd] exit member loop due to context canceled"]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_client.go:147] ["[tso] exit tso dispatcher check loop"]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_dispatcher.go:137] ["[tso] exit tso deadline watcher"] [dc-location=global]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_dispatcher.go:210] ["[tso] stop fetching the pending tso requests due to context canceled"] [dc-location=global]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_dispatcher.go:350] ["[tso] exit tso connection contexts updater"] [dc-location=global]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_dispatcher.go:171] ["[tso] exit tso dispatcher"] [dc-location=global]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_batch_controller.go:160] ["[pd] clear the tso batch controller"] [max-batch-size=10000] [best-batch-size=1] [collected-request-count=0] [pending-request-count=0]
[2024/07/12 10:42:02.575 +08:00] [INFO] [resource_manager_client.go:290] ["[resource manager] exit resource token dispatcher"]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_client.go:157] ["[tso] closing tso client"]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_client.go:162] ["[tso] close tso client"]
[2024/07/12 10:42:02.575 +08:00] [INFO] [tso_batch_controller.go:160] ["[pd] clear the tso batch controller"] [max-batch-size=10000] [best-batch-size=1] [collected-request-count=0] [pending-request-count=0]

• No need to test

  • I checked and no code files have been changed.

Side effects

• Performance regression: Consumes more CPU
• Performance regression: Consumes more Memory
• Breaking backward compatibility

Documentation

• Affects user behaviors
• Contains syntax changes
• Contains variable changes
• Contains experimental features
• Changes MySQL compatibility

Release note

Please refer to Release Notes Language Style Guide to write a quality release note.

None

[tiprow]

Hi @tiancaiamao. Thanks for your PR.

PRs from untrusted users cannot be marked as trusted with /ok-to-test in this repo meaning untrusted PR authors can never trigger tests themselves. Collaborators can still trigger tests on the PR using /test all.

I understand the commands that are listed here.

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes-sigs/prow repository.

[lance6716]

rest lgtm

[lance6716]

if ctx.Err() != nil { return }

[tiancaiamao]

done

[lance6716]

No I mean replace the whole select {case <- ctx.Done(): ..., default: ...}.

[zimulala]

I think @lance6716 means we can use if ctx.Err() != nil { return } instead of line63-69.

[ti-chi-bot]

[LGTM Timeline notifier]

Timeline:

• 2024-07-12 06:23:02.378094948 +0000 UTC m=+597879.613329046: ☑️ agreed by hawkingrei.
• 2024-07-12 07:42:14.41667926 +0000 UTC m=+602631.651913371: ☑️ agreed by lance6716.

[codecov]

Codecov Report

Attention: Patch coverage is 49.01961% with 26 lines in your changes missing coverage. Please review.

Project coverage is 73.2545%. Comparing base (05c8119) to head (f8c0095).
Report is 11 commits behind head on master.

Additional details and impacted files

@@               Coverage Diff                @@
##             master     #54594        +/-   ##
================================================
+ Coverage   72.7895%   73.2545%   +0.4649%     
================================================
  Files          1549       1549                
  Lines        436285     437999      +1714     
================================================
+ Hits         317570     320854      +3284     
+ Misses        99067      97478      -1589     
- Partials      19648      19667        +19     

Flag	Coverage Δ
integration	43.5413% <49.0196%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

Components	Coverage Δ
dumpling	52.9656% <ø> (ø)
parser	∅ <ø> (∅)
br	45.8706% <ø> (+0.0100%)	⬆️

[ti-chi-bot]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: hawkingrei, lance6716, zimulala

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [hawkingrei,lance6716,zimulala]
• pkg/ddl/OWNERS [zimulala]
• pkg/domain/OWNERS [lance6716,zimulala]
• pkg/planner/OWNERS [hawkingrei]
• pkg/statistics/OWNERS [hawkingrei]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

[hawkingrei]

/ok-to-test

[tiancaiamao]

/retest