fix lint

Signed-off-by: l1b0k <[email protected]>
l1b0k · Nov 25, 2024 · bbb8608 · bbb8608
1 parent edd1d1b
commit bbb8608
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/plugin/datapath/ipvlan_linux.go b/plugin/datapath/ipvlan_linux.go
@@ -301,7 +301,7 @@ func (d *IPvlanDriver) Setup(ctx context.Context, cfg *types.SetupConfig, netNS
 		if cfg.Egress == 0 && cfg.Ingress == 0 {
 			return nil
 		}
-		if cfg.BandwidthMode == "edt" {
+		if cfg.BandwidthMode == types.BandwidthModeEDT {
 			return ensureFQ(ctx, contLink)
 		}
 		return utils.SetupTC(contLink, cfg.Egress)

diff --git a/plugin/datapath/policy_router_linux.go b/plugin/datapath/policy_router_linux.go
@@ -311,7 +311,7 @@ func (d *PolicyRoute) Setup(ctx context.Context, cfg *types.SetupConfig, netNS n
 		return err
 	}
 
-	if cfg.BandwidthMode != "edt" && cfg.Egress > 0 {
+	if cfg.BandwidthMode != types.BandwidthModeEDT && cfg.Egress > 0 {
 		err = ensureMQ(ctx, eni)
 		if err != nil {
 			return err
@@ -348,7 +348,7 @@ func (d *PolicyRoute) Setup(ctx context.Context, cfg *types.SetupConfig, netNS n
 		if err != nil {
 			return err
 		}
-		if cfg.BandwidthMode != "edt" && cfg.Egress > 0 {
+		if cfg.BandwidthMode != types.BandwidthModeEDT && cfg.Egress > 0 {
 			return utils.SetupTC(contLink, cfg.Egress)
 		}
 		return nil
@@ -396,7 +396,7 @@ func (d *PolicyRoute) Setup(ctx context.Context, cfg *types.SetupConfig, netNS n
 		return fmt.Errorf("setup host veth config, %w", err)
 	}
 
-	if cfg.BandwidthMode != "edt" && cfg.Ingress > 0 {
+	if cfg.BandwidthMode != types.BandwidthModeEDT && cfg.Ingress > 0 {
 		return utils.SetupTC(hostVETH, cfg.Ingress)
 	}
 	return nil

diff --git a/plugin/driver/types/types.go b/plugin/driver/types/types.go
@@ -136,3 +136,8 @@ type TeardownCfg struct {
 
 	EnableNetworkPriority bool
 }
+
+const (
+	BandwidthModeEDT = "edt"
+	BandwidthModeTC  = "tc"
+)