优化代码

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,17 +488,21 @@ 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{
Id: int64(policy.Id), HttpFirewallPolicy: &pb.HTTPFirewallPolicy{
ServerId: int64(policy.ServerId), Id: int64(policy.Id),
Name: policy.Name, ServerId: int64(policy.ServerId),
Description: policy.Description, Name: policy.Name,
IsOn: policy.IsOn, Description: policy.Description,
InboundJSON: policy.Inbound, IsOn: policy.IsOn,
OutboundJSON: policy.Outbound, InboundJSON: policy.Inbound,
Mode: policy.Mode, OutboundJSON: policy.Outbound,
SynFloodJSON: policy.SynFlood, Mode: policy.Mode,
}}, nil SynFloodJSON: policy.SynFlood,
BlockOptionsJSON: policy.BlockOptions,
CaptchaOptionsJSON: policy.CaptchaOptions,
},
}, nil
} }
// ImportHTTPFirewallPolicy 导入策略数据 // ImportHTTPFirewallPolicy 导入策略数据