33 lines
874 B
Go
33 lines
874 B
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
|
|
)
|
|
|
|
var SharedHTTPFirewallRuleSetDAO = new(HTTPFirewallRuleSetDAO)
|
|
|
|
type HTTPFirewallRuleSetDAO struct {
|
|
BaseDAO
|
|
}
|
|
|
|
// 查找规则集配置
|
|
func (this *HTTPFirewallRuleSetDAO) FindRuleSetConfig(ctx context.Context, setId int64) (*firewallconfigs.HTTPFirewallRuleSet, error) {
|
|
resp, err := this.RPC().HTTPFirewallRuleSetRPC().FindEnabledHTTPFirewallRuleSetConfig(ctx, &pb.FindEnabledHTTPFirewallRuleSetConfigRequest{FirewallRuleSetId: setId})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(resp.FirewallRuleSetJSON) == 0 {
|
|
return nil, err
|
|
}
|
|
config := &firewallconfigs.HTTPFirewallRuleSet{}
|
|
err = json.Unmarshal(resp.FirewallRuleSetJSON, config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|