add OnMemChange in LockCtx

RemHero · Apr 18, 2024 · d8af033 · d8af033
1 parent 4183ab1
commit d8af033
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/kv/kv.go b/kv/kv.go
@@ -79,6 +79,7 @@ type LockCtx struct {
 	// LockCtx specially.
 	ResourceGroupTagger func(*kvrpcpb.PessimisticLockRequest) []byte
 	OnDeadlock          func(*tikverr.ErrDeadlock)
+	OnMemChange         func(int)
 }
 
 // LockWaitTime returns lockWaitTimeInMs
@@ -135,3 +136,13 @@ func (ctx *LockCtx) IterateValuesNotLocked(f func([]byte, []byte)) {
 		}
 	}
 }
+
+// getValuesKSize returns Values key size
+func (ctx *LockCtx) GetValuesKSize() uint64 {
+	return uint64(unsafe.Sizeof(*new(string)))
+}
+
+// getValuesKSize returns Values value size
+func (ctx *LockCtx) GetValuesVSize() uint64 {
+	return uint64(unsafe.Sizeof(*new(ReturnedValue)))
+}