|
@@ -17,7 +17,7 @@ type UdpForWardingService interface {
|
|
|
GetUdpForWarding(ctx context.Context,req v1.GetForwardingRequest) (v1.UdpForwardingDataRequest, error)
|
|
|
AddUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) error
|
|
|
EditUdpForwarding(ctx context.Context, req *v1.UdpForwardingRequest) error
|
|
|
- DeleteUdpForwarding(ctx context.Context, Ids []int) error
|
|
|
+ DeleteUdpForwarding(ctx context.Context, req v1.DeleteUdpForwardingRequest) error
|
|
|
GetUdpForwardingWafUdpAllIps(ctx context.Context, req v1.GetForwardingRequest) ([]v1.UdpForwardingDataRequest, error)
|
|
|
}
|
|
|
|
|
@@ -379,13 +379,17 @@ func (s *udpForWardingService) EditUdpForwarding(ctx context.Context, req *v1.Ud
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
-func (s *udpForWardingService) DeleteUdpForwarding(ctx context.Context, Ids []int) error {
|
|
|
- for _, id := range Ids {
|
|
|
+func (s *udpForWardingService) DeleteUdpForwarding(ctx context.Context, req v1.DeleteUdpForwardingRequest) error {
|
|
|
+ for _, id := range req.Ids {
|
|
|
oldData, err := s.udpForWardingRepository.GetUdpForWarding(ctx, int64(id))
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
|
|
|
+ if oldData.HostId != req.HostId {
|
|
|
+ return fmt.Errorf("用户权限不足")
|
|
|
+ }
|
|
|
+
|
|
|
err = s.cdn.DelServer(ctx, int64(oldData.CdnWebId))
|
|
|
if err != nil {
|
|
|
return err
|