mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-11 18:30:25 +08:00
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package headers
|
||
|
||
import (
|
||
"encoding/json"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/oplogs"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
|
||
)
|
||
|
||
type DeleteDeletingHeaderAction struct {
|
||
actionutils.ParentAction
|
||
}
|
||
|
||
func (this *DeleteDeletingHeaderAction) RunPost(params struct {
|
||
HeaderPolicyId int64
|
||
HeaderName string
|
||
}) {
|
||
// 日志
|
||
defer this.CreateLog(oplogs.LevelInfo, "删除需要删除的请求Header，HeaderPolicyId:%d, HeaderName:%s", params.HeaderPolicyId, params.HeaderName)
|
||
|
||
policyConfigResp, err := this.RPC().HTTPHeaderPolicyRPC().FindEnabledHTTPHeaderPolicyConfig(this.AdminContext(), &pb.FindEnabledHTTPHeaderPolicyConfigRequest{HttpHeaderPolicyId: params.HeaderPolicyId})
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
policyConfigJSON := policyConfigResp.HttpHeaderPolicyJSON
|
||
policyConfig := &shared.HTTPHeaderPolicy{}
|
||
err = json.Unmarshal(policyConfigJSON, policyConfig)
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
headerNames := []string{}
|
||
for _, h := range policyConfig.DeleteHeaders {
|
||
if h == params.HeaderName {
|
||
continue
|
||
}
|
||
headerNames = append(headerNames, h)
|
||
}
|
||
_, err = this.RPC().HTTPHeaderPolicyRPC().UpdateHTTPHeaderPolicyDeletingHeaders(this.AdminContext(), &pb.UpdateHTTPHeaderPolicyDeletingHeadersRequest{
|
||
HttpHeaderPolicyId: params.HeaderPolicyId,
|
||
HeaderNames: headerNames,
|
||
})
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
this.Success()
|
||
}
|