PTAL, Thanks~
/cc @breeswish @zhongzc @crazycs520

Upstream PR: pingcap/tidb#29044.

How about something like this (feel free to discuss):

type ResourceGroupTagger interface {
    tagCleanup(c *twoPhaseCommitter, bo *retry.Backoffer, batch batchMutations) []byte
    tagCommit(...)
    tagPrewrite(...)
    tagScan(...)
    tagGet(...)
    ....
}

or this:

type ResourceGroupTagger func(r *Request)

or even this:

type RequestPostBuildFn func(r *Request)

or:

type RequestPreSendFn func(r *Request)

@disksing What do you think?

ResourceGroupTagger voted.

Generally LGTM. However I believe there should be at least some unit tests to ensure that:

1. When ResourceGroup is set, it appears in the final request.
2. When ResourceGroupTagger is set, its modification take effects in the final request.
3. When ResourceGroup and ResourceGroupTagger are both set, ResourceGroupTagger does not take effect (according to your logic).

@mornyx I opened an empty PR and seems that integration tests work fine #372

@mornyx I opened an empty PR and seems that integration tests work fine #372

Got, seems that problem occurred in the use of mock store in the new integration tests, I will fix it.

@mornyx I opened an empty PR and seems that integration tests work fine #372

The problem did appear in my own integration test. Because of the incorrect use of other component, the error message did not occur in my code. Now I have fixed it (at least it works on my local machine, Github actions need to wait too long...), and next I need to find a way to fix Compatibility Test.

I've checked with a last merged PR 8a4ce38 the compatibility test also failed and it is merged anyway. I think we can ignore it and focus on other tests.

@disksing Please take a look~ Thanks! All tests (except for the compatibility test) are now passing.