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

实现websocket基本功能

Browse Source
This commit is contained in:
刘祥超
2020-09-26 19:54:15 +08:00
parent 8b11e7172a
commit 08f0788349
14 changed files with 375 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
internal/db/models/origin_dao.go
View File

@@ -175,23 +175,42 @@ func (this *OriginDAO) ComposeOriginConfig(originId int64) (*serverconfigs.Origi
config.IdleTimeout = idleTimeout
}
if origin.RequestHeaderPolicyId > 0 {
policyConfig, err := SharedHTTPHeaderPolicyDAO.ComposeHeaderPolicyConfig(int64(origin.RequestHeaderPolicyId))
// headers
if IsNotNull(origin.HttpRequestHeader) {
ref := &shared.HTTPHeaderPolicyRef{}
err = json.Unmarshal([]byte(origin.HttpRequestHeader), ref)
if err != nil {
return nil, err
}
if policyConfig != nil {
config.RequestHeaderPolicy = policyConfig
config.RequestHeaderPolicyRef = ref
if ref.HeaderPolicyId > 0 {
headerPolicy, err := SharedHTTPHeaderPolicyDAO.ComposeHeaderPolicyConfig(ref.HeaderPolicyId)
if err != nil {
return nil, err
}
if headerPolicy != nil {
config.RequestHeaderPolicy = headerPolicy
}
}
}
if origin.ResponseHeaderPolicyId > 0 {
policyConfig, err := SharedHTTPHeaderPolicyDAO.ComposeHeaderPolicyConfig(int64(origin.ResponseHeaderPolicyId))
if IsNotNull(origin.HttpResponseHeader) {
ref := &shared.HTTPHeaderPolicyRef{}
err = json.Unmarshal([]byte(origin.HttpResponseHeader), ref)
if err != nil {
return nil, err
}
if policyConfig != nil {
config.ResponseHeaderPolicy = policyConfig
config.ResponseHeaderPolicyRef = ref
if ref.HeaderPolicyId > 0 {
headerPolicy, err := SharedHTTPHeaderPolicyDAO.ComposeHeaderPolicyConfig(ref.HeaderPolicyId)
if err != nil {
return nil, err
}
if headerPolicy != nil {
config.ResponseHeaderPolicy = headerPolicy
}
}
}