|
@@ -341,6 +341,18 @@ func (s *webForwardingService) AddWebForwarding(ctx context.Context, req *v1.Web
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
+
|
|
|
+ // 验证域名限制
|
|
|
+ err = s.wafformatter.validateWafDomainCount(ctx, v1.GlobalRequire{
|
|
|
+ HostId: req.HostId,
|
|
|
+ Domain: req.WebForwardingData.Domain,
|
|
|
+ Comment: req.WebForwardingData.Comment,
|
|
|
+ Uid: req.Uid,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
err = s.wafformatter.validateWafPortCount(ctx, require.HostId)
|
|
|
if err != nil {
|
|
|
return err
|
|
@@ -524,6 +536,17 @@ func (s *webForwardingService) AddWebForwarding(ctx context.Context, req *v1.Web
|
|
|
|
|
|
func (s *webForwardingService) EditWebForwarding(ctx context.Context, req *v1.WebForwardingRequest) error {
|
|
|
|
|
|
+
|
|
|
+ err := s.wafformatter.validateWafDomainCount(ctx, v1.GlobalRequire{
|
|
|
+ HostId: req.HostId,
|
|
|
+ Domain: req.WebForwardingData.Domain,
|
|
|
+ Comment: req.WebForwardingData.Comment,
|
|
|
+ Uid: req.Uid,
|
|
|
+ })
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
oldData, err := s.webForwardingRepository.GetWebForwarding(ctx, int64(req.WebForwardingData.Id))
|
|
|
if err != nil {
|
|
|
return err
|