replica selector refactor (#1142)

* init

Signed-off-by: crazycs520 <[email protected]>

* add ReplicaSelector interface

Signed-off-by: crazycs520 <[email protected]>

* add replica_selector_v2, todo: fix test

Signed-off-by: crazycs520 <[email protected]>

* fix all test

Signed-off-by: crazycs520 <[email protected]>

* fix test in another way to compatible with old version about stale-read request

Signed-off-by: crazycs520 <[email protected]>

* tiny refactor

Signed-off-by: crazycs520 <[email protected]>

* refactor to remove duplicate code

Signed-off-by: crazycs520 <[email protected]>

* add more test

Signed-off-by: crazycs520 <[email protected]>

* refine onServerIsBusy region error handing logic

Signed-off-by: crazycs520 <[email protected]>

* support forwarding by proxy

Signed-off-by: crazycs520 <[email protected]>

* support busyThreshold and tiny reractor code

Signed-off-by: crazycs520 <[email protected]>

* add config

Signed-off-by: crazycs520 <[email protected]>

* fix test

Signed-off-by: crazycs520 <[email protected]>

* fix proxy bug

Signed-off-by: crazycs520 <[email protected]>

* fix test and tiny refactor

Signed-off-by: crazycs520 <[email protected]>

* add some test

Signed-off-by: crazycs520 <[email protected]>

* add more test

Signed-off-by: crazycs520 <[email protected]>

* tidy refine

Signed-off-by: crazycs520 <[email protected]>

* address comment

Signed-off-by: crazycs520 <[email protected]>

* fix test in v1

Signed-off-by: crazycs520 <[email protected]>

* add backoff-cnt check and timeout test

Signed-off-by: crazycs520 <[email protected]>

* refine test

Signed-off-by: crazycs520 <[email protected]>

* refine code

Signed-off-by: crazycs520 <[email protected]>

* refine code

Signed-off-by: crazycs520 <[email protected]>

* add benchmark test

Signed-off-by: crazycs520 <[email protected]>

* make test stale

Signed-off-by: crazycs520 <[email protected]>

* fix golangci

Signed-off-by: crazycs520 <[email protected]>

* add more test and refine code

Signed-off-by: crazycs520 <[email protected]>

* fix race test

Signed-off-by: crazycs520 <[email protected]>

* add more comprehensive enumeration testing

Signed-off-by: crazycs520 <[email protected]>

* refine test

Signed-off-by: crazycs520 <[email protected]>

* fix test

Signed-off-by: crazycs520 <[email protected]>

* fix accessFollower with label retry bug

Signed-off-by: crazycs520 <[email protected]>

* refine test and fix some bug

Signed-off-by: crazycs520 <[email protected]>

* fix test and add more test

Signed-off-by: crazycs520 <[email protected]>

* add more test

Signed-off-by: crazycs520 <[email protected]>

* refine test

Signed-off-by: crazycs520 <[email protected]>

* refine test

Signed-off-by: crazycs520 <[email protected]>

* refine onNotLeader logic when new leader is not available

Signed-off-by: crazycs520 <[email protected]>

* refine calculateScore logic, if the replica already tried, decrease the score

Signed-off-by: crazycs520 <[email protected]>

* refine logic: replica-read request with mixed strategy and with label, should be able to retry all remain replicas

Signed-off-by: crazycs520 <[email protected]>

* prefer-leader for mixed read with label for non stale-read req

Signed-off-by: crazycs520 <[email protected]>

* add more test case

Signed-off-by: crazycs520 <[email protected]>

* refine code

Signed-off-by: crazycs520 <[email protected]>

* resolve conflict

Signed-off-by: crazycs520 <[email protected]>

* remove some duplicate test

Signed-off-by: crazycs520 <[email protected]>

* refactor test

Signed-off-by: crazycs520 <[email protected]>

* refine code

Signed-off-by: crazycs520 <[email protected]>

* refine test

Signed-off-by: crazycs520 <[email protected]>

* add busy_threshold test and fix bug

Signed-off-by: crazycs520 <[email protected]>

* fix proxy bug and add more test

Signed-off-by: crazycs520 <[email protected]>

* refine code and add test

Signed-off-by: crazycs520 <[email protected]>

* add learner test

Signed-off-by: crazycs520 <[email protected]>

* add more test

Signed-off-by: crazycs520 <[email protected]>

* remove old test and refine test

Signed-off-by: crazycs520 <[email protected]>

* move test

Signed-off-by: crazycs520 <[email protected]>

* address comment

Signed-off-by: crazycs520 <[email protected]>

* use new score calculation by flag bit

Signed-off-by: crazycs520 <[email protected]>

* make test stable

Signed-off-by: crazycs520 <[email protected]>

* refine comment,license and fix lint

Signed-off-by: crazycs520 <[email protected]>

* refine comment

Signed-off-by: crazycs520 <[email protected]>

* refine comment

Signed-off-by: crazycs520 <[email protected]>

* fix race test timeout

Signed-off-by: crazycs520 <[email protected]>

* add flagNotAttemp in score

Signed-off-by: crazycs520 <[email protected]>

* address comment

Signed-off-by: crazycs520 <[email protected]>

---------

Signed-off-by: crazycs520 <[email protected]>

config/config.go

@@ -82,23 +82,26 @@ type Config struct {
 	RegionsRefreshInterval uint64
 	// EnablePreload indicates whether to preload region info when initializing the client.
 	EnablePreload bool
+	// EnableReplicaSelectorV2 indicates whether to enable ReplicaSelectorV2.
+	EnableReplicaSelectorV2 bool
 }
 
 // DefaultConfig returns the default configuration.
 func DefaultConfig() Config {
 	return Config{
-		CommitterConcurrency:  128,
-		MaxTxnTTL:             60 * 60 * 1000, // 1hour
-		TiKVClient:            DefaultTiKVClient(),
-		PDClient:              DefaultPDClient(),
-		TxnLocalLatches:       DefaultTxnLocalLatches(),
-		StoresRefreshInterval: DefStoresRefreshInterval,
-		OpenTracingEnable:     false,
-		Path:                  "",
-		EnableForwarding:      false,
-		TxnScope:              "",
-		EnableAsyncCommit:     false,
-		Enable1PC:             false,
+		CommitterConcurrency:    128,
+		MaxTxnTTL:               60 * 60 * 1000, // 1hour
+		TiKVClient:              DefaultTiKVClient(),
+		PDClient:                DefaultPDClient(),
+		TxnLocalLatches:         DefaultTxnLocalLatches(),
+		StoresRefreshInterval:   DefStoresRefreshInterval,
+		OpenTracingEnable:       false,
+		Path:                    "",
+		EnableForwarding:        false,
+		TxnScope:                "",
+		EnableAsyncCommit:       false,
+		Enable1PC:               false,
+		EnableReplicaSelectorV2: true,
 	}
 }