4106fc0436
* fix(tikv): improve context propagation and refactor batch delete logic Address review comments from PR #7557: 1. Replace context.TODO() with ctx in txn.Get calls - Fixes timeout/cancellation propagation in FindEntry - Fixes timeout/cancellation propagation in KvGet 2. Refactor DeleteFolderChildren to use flush helper - Eliminates code duplication - Cleaner and more maintainable These changes ensure proper context propagation throughout all TiKV operations and improve code maintainability. * error formatting
51 lines
1011 B
Go
51 lines
1011 B
Go
//go:build tikv
|
|
// +build tikv
|
|
|
|
package tikv
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/filer"
|
|
"github.com/tikv/client-go/v2/txnkv"
|
|
)
|
|
|
|
func (store *TikvStore) KvPut(ctx context.Context, key []byte, value []byte) error {
|
|
tw, err := store.getTxn(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return tw.RunInTxn(ctx, func(txn *txnkv.KVTxn) error {
|
|
return txn.Set(key, value)
|
|
})
|
|
}
|
|
|
|
func (store *TikvStore) KvGet(ctx context.Context, key []byte) ([]byte, error) {
|
|
tw, err := store.getTxn(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var data []byte = nil
|
|
err = tw.RunInTxn(ctx, func(txn *txnkv.KVTxn) error {
|
|
val, err := txn.Get(ctx, key)
|
|
if err == nil {
|
|
data = val
|
|
}
|
|
return err
|
|
})
|
|
if isNotExists(err) {
|
|
return data, filer.ErrKvNotFound
|
|
}
|
|
return data, err
|
|
}
|
|
|
|
func (store *TikvStore) KvDelete(ctx context.Context, key []byte) error {
|
|
tw, err := store.getTxn(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return tw.RunInTxn(ctx, func(txn *txnkv.KVTxn) error {
|
|
return txn.Delete(key)
|
|
})
|
|
}
|