tikv
/
client-go
mirror of https://github.com/tikv/client-go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

suuport api bench for params 

Signed-off-by: husharp <jinhao.hu@pingcap.com>
This commit is contained in:
husharp 2023-08-14 13:11:16 +08:00
parent 3f36b93a7d
commit 1a932e727e
2 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tikv/kv.go
View File

@ -586,12 +586,12 @@ func (s *KVStore) updateSafeTS(ctx context.Context) {
wg.Add(len(stores))
// Try to get the minimum resolved timestamp of the store from PD.
var err error
storeIDs := make([]string, len(stores))
for i, store := range stores {
storeIDs[i] = strconv.FormatUint(store.StoreID(), 10)
}
storeMinResolvedTSs := make(map[uint64]uint64)
if s.pdHttpClient != nil {
storeIDs := make([]string, len(stores))
for i, store := range stores {
storeIDs[i] = strconv.FormatUint(store.StoreID(), 10)
}
storeMinResolvedTSs, err = s.pdHttpClient.GetMinResolvedTSByStoresIDs(ctx, storeIDs)
if err != nil {
// If getting the minimum resolved timestamp from PD failed, log the error and need to get it from TiKV.

22
util/pd.go
View File

@ -35,7 +35,6 @@
package util
import (
"bytes"
"context"
"crypto/tls"
"encoding/json"
@ -91,21 +90,18 @@ func NewPDHTTPClient(
func (p *PDHTTPClient) GetMinResolvedTSByStoresIDs(ctx context.Context, storeIDs []string) (map[uint64]uint64, error) {
var err error
for _, addr := range p.addrs {
data, err := json.Marshal(storeIDs)
if err != nil {
logutil.BgLogger().Debug("failed to marshal store ids", zap.String("addr", addr), zap.Error(err))
return nil, errors.Trace(err)
}
v, err := pdRequest(ctx, addr, minResolvedTSPrefix, p.cli, http.MethodGet, bytes.NewBuffer(data))
if err != nil {
logutil.BgLogger().Debug("failed to get min resolved ts", zap.String("addr", addr), zap.Error(err))
// scope is an optional parameter, it can be `cluster` or specified store IDs.
// - When no scope is given, cluster-level's min_resolved_ts will be returned and storesMinResolvedTS will be nil.
// - When scope is `cluster`, cluster-level's min_resolved_ts will be returned and storesMinResolvedTS will be filled.
// - When scope given a list of stores, min_resolved_ts will be provided for each store
// and the scope-specific min_resolved_ts will be returned.
query := fmt.Sprintf("%s?scope=%s", minResolvedTSPrefix, strings.Join(storeIDs, ","))
v, e := pdRequest(ctx, addr, query, p.cli, http.MethodGet, nil)
if e != nil {
logutil.BgLogger().Debug("failed to get min resolved ts", zap.String("addr", addr), zap.Error(e))
continue
}
logutil.BgLogger().Debug("min resolved ts", zap.String("resp", string(v)))
// - When no store is given, cluster-level's min_resolved_ts will be returned,
// and min_resolved_ts for each store will be empty.
// - When given a list of stores, min_resolved_ts will be provided for each store
// and the scope-specific min_resolved_ts will be returned.
d := struct {
IsRealTime bool `json:"is_real_time,omitempty"`
StoresMinResolvedTS map[uint64]uint64 `json:"stores_min_resolved_ts"`