优化代码

This commit is contained in:
刘祥超
2023-11-29 16:58:11 +08:00
parent da391f565b
commit 707a9f8caf

View File

@@ -33,7 +33,7 @@ func (this *HTTPFirewallPolicyService) FindAllEnabledHTTPFirewallPolicies(ctx co
return nil, err return nil, err
} }
result := []*pb.HTTPFirewallPolicy{} var result = []*pb.HTTPFirewallPolicy{}
for _, p := range policies { for _, p := range policies {
result = append(result, &pb.HTTPFirewallPolicy{ result = append(result, &pb.HTTPFirewallPolicy{
Id: int64(p.Id), Id: int64(p.Id),
@@ -395,7 +395,7 @@ func (this *HTTPFirewallPolicyService) ListEnabledHTTPFirewallPolicies(ctx conte
return nil, err return nil, err
} }
result := []*pb.HTTPFirewallPolicy{} var result = []*pb.HTTPFirewallPolicy{}
for _, p := range policies { for _, p := range policies {
result = append(result, &pb.HTTPFirewallPolicy{ result = append(result, &pb.HTTPFirewallPolicy{
Id: int64(p.Id), Id: int64(p.Id),
@@ -488,7 +488,8 @@ func (this *HTTPFirewallPolicyService) FindEnabledHTTPFirewallPolicy(ctx context
if policy == nil { if policy == nil {
return &pb.FindEnabledHTTPFirewallPolicyResponse{HttpFirewallPolicy: nil}, nil return &pb.FindEnabledHTTPFirewallPolicyResponse{HttpFirewallPolicy: nil}, nil
} }
return &pb.FindEnabledHTTPFirewallPolicyResponse{HttpFirewallPolicy: &pb.HTTPFirewallPolicy{ return &pb.FindEnabledHTTPFirewallPolicyResponse{
HttpFirewallPolicy: &pb.HTTPFirewallPolicy{
Id: int64(policy.Id), Id: int64(policy.Id),
ServerId: int64(policy.ServerId), ServerId: int64(policy.ServerId),
Name: policy.Name, Name: policy.Name,
@@ -498,7 +499,10 @@ func (this *HTTPFirewallPolicyService) FindEnabledHTTPFirewallPolicy(ctx context
OutboundJSON: policy.Outbound, OutboundJSON: policy.Outbound,
Mode: policy.Mode, Mode: policy.Mode,
SynFloodJSON: policy.SynFlood, SynFloodJSON: policy.SynFlood,
}}, nil BlockOptionsJSON: policy.BlockOptions,
CaptchaOptionsJSON: policy.CaptchaOptions,
},
}, nil
} }
// ImportHTTPFirewallPolicy 导入策略数据 // ImportHTTPFirewallPolicy 导入策略数据