-
Notifications
You must be signed in to change notification settings - Fork 5.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
close #49414
- Loading branch information
1 parent
26e9a3a
commit cba3c91
Showing
8 changed files
with
263 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,124 @@ | ||
| // Copyright 2023 PingCAP, Inc. Licensed under Apache-2.0. | ||
|
|
||
| package storage | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "os" | ||
| "time" | ||
|
|
||
| "github.com/pingcap/errors" | ||
| "github.com/pingcap/log" | ||
| "github.com/pingcap/tidb/br/pkg/logutil" | ||
| "go.uber.org/zap" | ||
| ) | ||
|
|
||
| // LockMeta is the meta information of a lock. | ||
| type LockMeta struct { | ||
| LockedAt time.Time `json:"locked_at"` | ||
| LockerHost string `json:"locker_host"` | ||
| LockerPID int `json:"locker_pid"` | ||
| Hint string `json:"hint"` | ||
| } | ||
|
|
||
| func (l LockMeta) String() string { | ||
| return fmt.Sprintf("Locked(at: %s, host: %s, pid: %d, hint: %s)", l.LockedAt.Format(time.DateTime), l.LockerHost, l.LockerPID, l.Hint) | ||
| } | ||
|
|
||
| // ErrLocked is the error returned when the lock is held by others. | ||
| type ErrLocked struct { | ||
| Meta LockMeta | ||
| } | ||
|
|
||
| func (e ErrLocked) Error() string { | ||
| return fmt.Sprintf("locked, meta = %s", e.Meta) | ||
| } | ||
|
|
||
| // MakeLockMeta creates a LockMeta by the current node's metadata. | ||
| // Including current time and hostname, etc.. | ||
| func MakeLockMeta(hint string) LockMeta { | ||
| hname, err := os.Hostname() | ||
| if err != nil { | ||
| hname = fmt.Sprintf("UnknownHost(err=%s)", err) | ||
| } | ||
| now := time.Now() | ||
| meta := LockMeta{ | ||
| LockedAt: now, | ||
| LockerHost: hname, | ||
| Hint: hint, | ||
| LockerPID: os.Getpid(), | ||
| } | ||
| return meta | ||
| } | ||
|
|
||
| func readLockMeta(ctx context.Context, storage ExternalStorage, path string) (LockMeta, error) { | ||
| file, err := storage.ReadFile(ctx, path) | ||
| if err != nil { | ||
| return LockMeta{}, errors.Annotatef(err, "failed to read existed lock file %s", path) | ||
| } | ||
| meta := LockMeta{} | ||
| err = json.Unmarshal(file, &meta) | ||
| if err != nil { | ||
| return meta, errors.Annotatef(err, "failed to parse lock file %s", path) | ||
| } | ||
|
|
||
| return meta, nil | ||
| } | ||
|
|
||
| func putLockMeta(ctx context.Context, storage ExternalStorage, path string, meta LockMeta) error { | ||
| file, err := json.Marshal(meta) | ||
| if err != nil { | ||
| return errors.Annotatef(err, "failed to marshal lock meta %s", path) | ||
| } | ||
| err = storage.WriteFile(ctx, path, file) | ||
| if err != nil { | ||
| return errors.Annotatef(err, "failed to write lock meta at %s", path) | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // TryLockRemote tries to create a "lock file" at the external storage. | ||
| // If success, we will create a file at the path provided. So others may not access the file then. | ||
| // Will return a `ErrLocked` if there is another process already creates the lock file. | ||
| // This isn't a strict lock like flock in linux: that means, the lock might be forced removed by | ||
| // manually deleting the "lock file" in external storage. | ||
| func TryLockRemote(ctx context.Context, storage ExternalStorage, path, hint string) (err error) { | ||
| defer func() { | ||
| log.Info("Trying lock remote file.", zap.String("path", path), zap.String("hint", hint), logutil.ShortError(err)) | ||
| }() | ||
| exists, err := storage.FileExists(ctx, path) | ||
| if err != nil { | ||
| return errors.Annotatef(err, "failed to check lock file %s exists", path) | ||
| } | ||
| if exists { | ||
| meta, err := readLockMeta(ctx, storage, path) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| return ErrLocked{Meta: meta} | ||
| } | ||
|
|
||
| meta := MakeLockMeta(hint) | ||
| return putLockMeta(ctx, storage, path, meta) | ||
| } | ||
|
|
||
| // UnlockRemote removes the lock file at the specified path. | ||
| // Removing that file will release the lock. | ||
| func UnlockRemote(ctx context.Context, storage ExternalStorage, path string) error { | ||
| meta, err := readLockMeta(ctx, storage, path) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| // NOTE: this is for debug usage. For now, there isn't an Compare-And-Swap | ||
| // operation in our ExternalStorage abstraction. | ||
| // So, once our lock has been overwritten or we are overwriting other's lock, | ||
| // this information will be useful for troubleshooting. | ||
| log.Info("Releasing lock.", zap.Stringer("meta", meta), zap.String("path", path)) | ||
| err = storage.DeleteFile(ctx, path) | ||
| if err != nil { | ||
| return errors.Annotatef(err, "failed to delete lock file %s", path) | ||
| } | ||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| // Copyright 2023 PingCAP, Inc. Licensed under Apache-2.0. | ||
|
|
||
| package storage_test | ||
|
|
||
| import ( | ||
| "context" | ||
| "os" | ||
| "path/filepath" | ||
| "testing" | ||
|
|
||
| backup "github.com/pingcap/kvproto/pkg/brpb" | ||
| "github.com/pingcap/tidb/br/pkg/storage" | ||
| "github.com/stretchr/testify/require" | ||
| ) | ||
|
|
||
| func createMockStorage(t *testing.T) (storage.ExternalStorage, string) { | ||
| tempdir := t.TempDir() | ||
| storage, err := storage.New(context.Background(), &backup.StorageBackend{ | ||
| Backend: &backup.StorageBackend_Local{ | ||
| Local: &backup.Local{ | ||
| Path: tempdir, | ||
| }, | ||
| }, | ||
| }, nil) | ||
| require.NoError(t, err) | ||
| return storage, tempdir | ||
| } | ||
|
|
||
| func requireFileExists(t *testing.T, path string) { | ||
| _, err := os.Stat(path) | ||
| require.NoError(t, err) | ||
| } | ||
|
|
||
| func requireFileNotExists(t *testing.T, path string) { | ||
| _, err := os.Stat(path) | ||
| require.True(t, os.IsNotExist(err)) | ||
| } | ||
|
|
||
| func TestTryLockRemote(t *testing.T) { | ||
| ctx := context.Background() | ||
| strg, pth := createMockStorage(t) | ||
| err := storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!") | ||
| require.NoError(t, err) | ||
| requireFileExists(t, filepath.Join(pth, "test.lock")) | ||
| err = storage.UnlockRemote(ctx, strg, "test.lock") | ||
| require.NoError(t, err) | ||
| requireFileNotExists(t, filepath.Join(pth, "test.lock")) | ||
| } | ||
|
|
||
| func TestConflictLock(t *testing.T) { | ||
| ctx := context.Background() | ||
| strg, pth := createMockStorage(t) | ||
| err := storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!") | ||
| require.NoError(t, err) | ||
| err = storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!") | ||
| require.ErrorContains(t, err, "locked, meta = Locked") | ||
| requireFileExists(t, filepath.Join(pth, "test.lock")) | ||
| err = storage.UnlockRemote(ctx, strg, "test.lock") | ||
| require.NoError(t, err) | ||
| requireFileNotExists(t, filepath.Join(pth, "test.lock")) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters