add trace region

harmony-one · Jun 15, 2022 · 6c28424 · 6c28424
1 parent 71ee0be
commit 6c28424
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/tikv/remote/remote_database.go b/internal/tikv/remote/remote_database.go
@@ -7,6 +7,7 @@ import (
 	"github.com/harmony-one/harmony/internal/tikv/common"
 	"github.com/tikv/client-go/v2/config"
 	"github.com/tikv/client-go/v2/rawkv"
+	"runtime/trace"
 	"sync/atomic"
 )
 
@@ -53,6 +54,9 @@ func (d *RemoteDatabase) Get(key []byte) ([]byte, error) {
 		return nil, common.ErrEmptyKey
 	}
 
+	region := trace.StartRegion(context.Background(), "tikv Get")
+	defer region.End()
+
 	get, err := d.client.Get(context.Background(), key)
 	if err != nil {
 		return nil, err