commit

internal/service/service.go

@@ -4,8 +4,8 @@ import "context"
 
 type Service interface {
 	// QueryIP(ip string) (*model.IPInfo, error)
-	GetWaf(ctx context.Context, gatewayName string) (*Waf, error)
-	CreateWaf(ctx context.Context, req *CreateWafReq) (*Waf, error)
-	UpdateMode(ctx context.Context, req *UpdateModeReq) (*Waf, error)
+	GetWaf(ctx context.Context, gatewayName string) (*WafService, error)
+	CreateWaf(ctx context.Context, req *CreateWafReq) (*WafService, error)
+	UpdateMode(ctx context.Context, req *UpdateModeReq) (*WafService, error)
 	SaveRuleCategoryToDB(ctx context.Context) error
 }

internal/service/waf.go

@@ -48,7 +48,7 @@ func (s *wafService) GetWaf(ctx context.Context, gatewayName string) (*WafServic
 	}, nil
 }
 
-func (s *wafService) CreateWaf(ctx context.Context, req *CreateWafReq) (*Waf, error) {
+func (s *wafService) CreateWaf(ctx context.Context, req *CreateWafReq) (*WafService, error) {
 	// Create the WAF service resource
 	service := &v1alpha1.Service{
 		ObjectMeta: metav1.ObjectMeta{
@@ -137,7 +137,7 @@ func (s *wafService) CreateWaf(ctx context.Context, req *CreateWafReq) (*Waf, er
 	return nil, nil
 }
 
-func (s *wafService) UpdateMode(ctx context.Context, req *UpdateModeReq) (*Waf, error) {
+func (s *wafService) UpdateMode(ctx context.Context, req *UpdateModeReq) (*WafService, error) {
 	// Check if WAF service exists
 	wafService := &model.WafService{}
 	err := s.db.Model(&model.WafService{}).Where("gateway_name = ?", req.GatewayName).First(wafService).Error