-
Notifications
You must be signed in to change notification settings - Fork 283
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
checker(dm): port some precheck from lightning (#7617)
ref #3510
- Loading branch information
Showing
5 changed files
with
351 additions
and
14 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
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,137 @@ | ||
| // Copyright 2022 PingCAP, Inc. | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| package checker | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "github.com/pingcap/tidb/br/pkg/lightning/restore" | ||
| ) | ||
|
|
||
| func convertLightningPrecheck( | ||
| ctx context.Context, | ||
| dmResult *Result, | ||
| lightningPrechecker restore.PrecheckItem, | ||
| failLevel State, | ||
| instruction string, | ||
| ) { | ||
| lightningResult, err := lightningPrechecker.Check(ctx) | ||
| if err != nil { | ||
| markCheckError(dmResult, err) | ||
| return | ||
| } | ||
| if !lightningResult.Passed { | ||
| dmResult.State = failLevel | ||
| dmResult.Instruction = instruction | ||
| dmResult.Errors = append(dmResult.Errors, &Error{Severity: failLevel, ShortErr: lightningResult.Message}) | ||
| return | ||
| } | ||
| dmResult.State = StateSuccess | ||
| } | ||
|
|
||
| // LightningEmptyRegionChecker checks whether there are too many empty regions in the cluster. | ||
| type LightningEmptyRegionChecker struct { | ||
| inner restore.PrecheckItem | ||
| } | ||
|
|
||
| // NewLightningEmptyRegionChecker creates a new LightningEmptyRegionChecker. | ||
| func NewLightningEmptyRegionChecker(lightningChecker restore.PrecheckItem) RealChecker { | ||
| return &LightningEmptyRegionChecker{inner: lightningChecker} | ||
| } | ||
|
|
||
| // Name implements the RealChecker interface. | ||
| func (c *LightningEmptyRegionChecker) Name() string { | ||
| return "lightning_empty_region" | ||
| } | ||
|
|
||
| // Check implements the RealChecker interface. | ||
| func (c *LightningEmptyRegionChecker) Check(ctx context.Context) *Result { | ||
| result := &Result{ | ||
| Name: c.Name(), | ||
| Desc: "check whether there are too many empty Regions in the TiKV under Physical import mode", | ||
| State: StateFailure, | ||
| } | ||
| convertLightningPrecheck( | ||
| ctx, | ||
| result, | ||
| c.inner, | ||
| StateWarning, | ||
| `you can change "region merge" related configuration in PD to speed up eliminating empty regions`, | ||
| ) | ||
| return result | ||
| } | ||
|
|
||
| // LightningRegionDistributionChecker checks whether the region distribution is balanced. | ||
| type LightningRegionDistributionChecker struct { | ||
| inner restore.PrecheckItem | ||
| } | ||
|
|
||
| // NewLightningRegionDistributionChecker creates a new LightningRegionDistributionChecker. | ||
| func NewLightningRegionDistributionChecker(lightningChecker restore.PrecheckItem) RealChecker { | ||
| return &LightningRegionDistributionChecker{inner: lightningChecker} | ||
| } | ||
|
|
||
| // Name implements the RealChecker interface. | ||
| func (c *LightningRegionDistributionChecker) Name() string { | ||
| return "lightning_region_distribution" | ||
| } | ||
|
|
||
| // Check implements the RealChecker interface. | ||
| func (c *LightningRegionDistributionChecker) Check(ctx context.Context) *Result { | ||
| result := &Result{ | ||
| Name: c.Name(), | ||
| Desc: "check whether the Regions in the TiKV cluster are distributed evenly under Physical import mode", | ||
| State: StateFailure, | ||
| } | ||
| convertLightningPrecheck( | ||
| ctx, | ||
| result, | ||
| c.inner, | ||
| StateWarning, | ||
| `you can change "region schedule" related configuration in PD to speed up balancing regions`, | ||
| ) | ||
| return result | ||
| } | ||
|
|
||
| // LightningClusterVersionChecker checks whether the cluster version is compatible with Lightning. | ||
| type LightningClusterVersionChecker struct { | ||
| inner restore.PrecheckItem | ||
| } | ||
|
|
||
| // NewLightningClusterVersionChecker creates a new LightningClusterVersionChecker. | ||
| func NewLightningClusterVersionChecker(lightningChecker restore.PrecheckItem) RealChecker { | ||
| return &LightningClusterVersionChecker{inner: lightningChecker} | ||
| } | ||
|
|
||
| // Name implements the RealChecker interface. | ||
| func (c *LightningClusterVersionChecker) Name() string { | ||
| return "lightning_cluster_version" | ||
| } | ||
|
|
||
| // Check implements the RealChecker interface. | ||
| func (c *LightningClusterVersionChecker) Check(ctx context.Context) *Result { | ||
| result := &Result{ | ||
| Name: c.Name(), | ||
| Desc: "check whether the downstream TiDB/PD/TiKV version meets the requirements of Physical import mode", | ||
| State: StateFailure, | ||
| } | ||
| convertLightningPrecheck( | ||
| ctx, | ||
| result, | ||
| c.inner, | ||
| StateFailure, | ||
| `you can switch to logical import mode which has no requirements on downstream cluster version`, | ||
| ) | ||
| return result | ||
| } |
Oops, something went wrong.