Browse Source

refactor(internal): 将 GlobalLimit 结构体中的 State 字段类型从 int 改为 bool

- 修改了 internal/model/globallimit.go 中的 GlobalLimit 结构体
- 更新了 internal/repository/globallimit.go 中的相关方法
-
fusu 1 tháng trước cách đây
mục cha
commit
8c7ebdb7c5

+ 1 - 1
internal/model/globallimit.go

@@ -12,7 +12,7 @@ type GlobalLimit struct {
 	CdnUid          int
 	GatewayGroupId  int
 	Comment         string
-	State         int
+	State         bool
 	ExpiredAt       int64
 	createdAt       time.Time
 	updatedAt       time.Time

+ 3 - 3
internal/repository/globallimit.go

@@ -25,7 +25,7 @@ type GlobalLimitRepository interface {
 	// 获取套餐Id
 	GetNodeArea(ctx context.Context, nodeAreaName string) (int64, error)
 	// 修改套餐状态
-	EditHostState(ctx context.Context, hostId int64, state int) error
+	EditHostState(ctx context.Context, hostId int64, state bool) error
 }
 
 func NewGlobalLimitRepository(
@@ -62,7 +62,7 @@ func (r *globalLimitRepository) UpdateGlobalLimitByHostId(ctx context.Context, r
 
 func (r *globalLimitRepository) IsGlobalLimitExistByHostId(ctx context.Context, hostId int64) (bool, error) {
 	var count int64
-	err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ? AND state = 1", hostId).Count(&count).Error
+	err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ? AND state = 0", hostId).Count(&count).Error
 
 	if err != nil {
 		return false, err
@@ -163,7 +163,7 @@ func (r *globalLimitRepository) GetNodeArea(ctx context.Context, nodeAreaName st
 
 }
 
-func (r *globalLimitRepository) EditHostState(ctx context.Context, hostId int64, state int) error {
+func (r *globalLimitRepository) EditHostState(ctx context.Context, hostId int64, state bool) error {
 	if err := r.DB(ctx).Model(&model.GlobalLimit{}).Where("host_id = ?", hostId).Update("state", state).Error; err != nil {
 		return err
 	}

+ 3 - 3
internal/service/globallimit.go

@@ -356,7 +356,7 @@ func (s *globalLimitService) AddGlobalLimit(ctx context.Context, req v1.GlobalLi
 			CdnUid:         int(userId),
 			Comment:        req.Comment,
 			ExpiredAt:      expiredAt,
-			State:       1,
+			State:       true,
 		})
 		if err != nil {
 			return err
@@ -376,7 +376,7 @@ func (s *globalLimitService) AddGlobalLimit(ctx context.Context, req v1.GlobalLi
 		GatewayGroupId: gatewayGroupId,
 		CdnUid:         int(userId),
 		Comment:        req.Comment,
-		State:          1,
+		State:          true,
 		ExpiredAt:      expiredAt,
 	})
 	if err != nil {
@@ -513,7 +513,7 @@ func (s *globalLimitService) DeleteGlobalLimit(ctx context.Context, req v1.Globa
 	}
 
 
-	if err := s.globalLimitRepository.EditHostState(ctx, int64(req.HostId), 0); err != nil {
+	if err := s.globalLimitRepository.EditHostState(ctx, int64(req.HostId), false); err != nil {
 		return err
 	}
 	if err := s.gateWayGroupRep.EditGatewayGroup(ctx,&model.GatewayGroup{