Browse Source

fix/cc: 修复 CC 保护查询条件错误

- 将 GetCcList 函数中的 serverId改为 sourceServerId
-将 EditCcState 函数中的 serverId 改为 sourceServerId
fusu 3 weeks ago
parent
commit
f823e8fb8e
1 changed files with 2 additions and 2 deletions
  1. 2 2
      internal/repository/cc.go

+ 2 - 2
internal/repository/cc.go

@@ -24,14 +24,14 @@ type ccRepository struct {
 
 func (r *ccRepository) GetCcList(ctx context.Context, serviceId int64) ([]v1.CCList, error) {
 	var req []v1.CCList
-	if err := r.DBWithName(ctx,"cdn").Table("cloud_ip_items").Where("serverId = ? AND state = 1", serviceId).Select("value,type,reason,sourceURL,sourceUserAgent,createdAt,expiredAt").Scan(&req).Error; err != nil {
+	if err := r.DBWithName(ctx,"cdn").Table("cloud_ip_items").Where("sourceServerId = ? AND state = 1", serviceId).Select("value,type,reason,sourceURL,sourceUserAgent,createdAt,expiredAt").Scan(&req).Error; err != nil {
 		return nil, err
 	}
 	return req, nil
 }
 
 func (r *ccRepository) EditCcState(ctx context.Context, serviceId int64, ip string) error {
-	if err := r.DBWithName(ctx,"cdn").Table("cloud_ip_items").Where("serverId = ? AND value = ?", serviceId, ip).Update("state", 0).Error; err != nil {
+	if err := r.DBWithName(ctx,"cdn").Table("cloud_ip_items").Where("sourceServerId = ? AND value = ?", serviceId, ip).Update("state", 0).Error; err != nil {
 		return err
 	}
 	return nil