TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-12-06 01:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加字符编码/HTTP Header管理

Browse Source
This commit is contained in:
刘祥超
2020-09-16 20:29:18 +08:00
parent eae2e63db3
commit 58bce36fb3
18 changed files with 857 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
internal/rpc/services/service_http_web.go
View File

@@ -48,6 +48,9 @@ func (this *HTTPWebService) FindEnabledHTTPWeb(ctx context.Context, req *pb.Find
result.IsOn = web.IsOn == 1
result.Root = web.Root
result.GzipId = int64(web.GzipId)
result.Charset = web.Charset
result.RequestHeaderPolicyId = int64(web.RequestHeaderPolicyId)
result.ResponseHeaderPolicyId = int64(web.ResponseHeaderPolicyId)
return &pb.FindEnabledHTTPWebResponse{Web: result}, nil
}
@@ -82,3 +85,51 @@ func (this *HTTPWebService) UpdateHTTPWebGzip(ctx context.Context, req *pb.Updat
return &pb.UpdateHTTPWebGzipResponse{}, nil
}
// 修改字符集配置
func (this *HTTPWebService) UpdateHTTPWebCharset(ctx context.Context, req *pb.UpdateHTTPWebCharsetRequest) (*pb.UpdateHTTPWebCharsetResponse, error) {
// 校验请求
_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
if err != nil {
return nil, err
}
err = models.SharedHTTPWebDAO.UpdateWebCharset(req.WebId, req.Charset)
if err != nil {
return nil, err
}
return &pb.UpdateHTTPWebCharsetResponse{}, nil
}
// 更改请求Header策略
func (this *HTTPWebService) UpdateHTTPWebRequestHeaderPolicy(ctx context.Context, req *pb.UpdateHTTPWebRequestHeaderPolicyRequest) (*pb.UpdateHTTPWebRequestHeaderPolicyResponse, error) {
// 校验请求
_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
if err != nil {
return nil, err
}
err = models.SharedHTTPWebDAO.UpdateHTTPWebRequestHeaderPolicy(req.WebId, req.HeaderPolicyId)
if err != nil {
return nil, err
}
return &pb.UpdateHTTPWebRequestHeaderPolicyResponse{}, nil
}
// 更改响应Header策略
func (this *HTTPWebService) UpdateHTTPWebResponseHeaderPolicy(ctx context.Context, req *pb.UpdateHTTPWebResponseHeaderPolicyRequest) (*pb.UpdateHTTPWebResponseHeaderPolicyResponse, error) {
// 校验请求
_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
if err != nil {
return nil, err
}
err = models.SharedHTTPWebDAO.UpdateHTTPWebResponseHeaderPolicy(req.WebId, req.HeaderPolicyId)
if err != nil {
return nil, err
}
return &pb.UpdateHTTPWebResponseHeaderPolicyResponse{}, nil
}