Fix the check

Signed-off-by: JmPotato <ghzpotato@gmail.com>

internal/client/client_interceptor.go

@@ -57,7 +57,7 @@ func (r interceptedClient) SendRequest(ctx context.Context, addr string, req *ti
 }
 
 var (
-	resourceControlSwitch      atomic.Bool
+	resourceControlSwitch      atomic.Value
 	resourceControlInterceptor client.ResourceGroupKVInterceptor
 )
 
@@ -88,7 +88,7 @@ func buildResourceControlInterceptor(
 	req *tikvrpc.Request,
 	resourceGroupName string,
 ) interceptor.RPCInterceptor {
-	if !resourceControlSwitch.Load() {
+	if !resourceControlSwitch.Load().(bool) {
 		return nil
 	}
 	// When the group name is empty or "default", we don't need to

internal/resource_control/resource_control.go

@@ -126,10 +126,12 @@ func getKVCPUMs(detailsV2 *kvrpcpb.ExecDetailsV2, details *kvrpcpb.ExecDetails)
 	return 0
 }
 
+// ReadBytes returns the read bytes of the response.
 func (res *ResponseInfo) ReadBytes() uint64 {
 	return res.readBytes
 }
 
+// KVCPUMs returns the KV CPU time in milliseconds of the response.
 func (res *ResponseInfo) KVCPUMs() uint64 {
 	return res.kvCPUMs
 }