Skip to content

Commit 91693f1

Browse files
committed
fix: 完善审计场景 Header 处理逻辑,增加 Set 方法并修复 CCHeader 丢失字段的问题
1 parent f96d96f commit 91693f1

File tree

2 files changed

+21
-3
lines changed

2 files changed

+21
-3
lines changed

src/common/http/header/accessor.go

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,3 +231,18 @@ func GetSceneDesc(header http.Header) string {
231231
func GetSceneCode(header http.Header) string {
232232
return header.Get(CCSceneCodeHeader)
233233
}
234+
235+
// SetScene set audit scene to http header
236+
func SetScene(header http.Header, value string) {
237+
header.Set(CCSceneHeader, value)
238+
}
239+
240+
// SetSceneDesc set audit scene description to http header
241+
func SetSceneDesc(header http.Header, value string) {
242+
header.Set(CCSceneDescHeader, value)
243+
}
244+
245+
// SetSceneCode set audit scene code to http header
246+
func SetSceneCode(header http.Header, value string) {
247+
header.Set(CCSceneCodeHeader, value)
248+
}

src/common/http/header/util/util.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,9 @@ func CCHeader(header http.Header) http.Header {
4141
httpheader.SetReqFromWeb(newHeader)
4242
}
4343
newHeader.Add(common.ReadReferenceKey, header.Get(common.ReadReferenceKey))
44+
httpheader.SetScene(newHeader, httpheader.GetScene(header))
45+
httpheader.SetSceneDesc(newHeader, httpheader.GetSceneDesc(header))
46+
httpheader.SetSceneCode(newHeader, httpheader.GetSceneCode(header))
4447

4548
return newHeader
4649
}
@@ -97,9 +100,9 @@ func NewHeader(header http.Header) http.Header {
97100
}
98101

99102
// Copy audit scene headers
100-
newHeader.Set(httpheader.CCSceneHeader, httpheader.GetScene(header))
101-
newHeader.Set(httpheader.CCSceneDescHeader, httpheader.GetSceneDesc(header))
102-
newHeader.Set(httpheader.CCSceneCodeHeader, httpheader.GetSceneCode(header))
103+
httpheader.SetScene(newHeader, httpheader.GetScene(header))
104+
httpheader.SetSceneDesc(newHeader, httpheader.GetSceneDesc(header))
105+
httpheader.SetSceneCode(newHeader, httpheader.GetSceneCode(header))
103106

104107
return newHeader
105108
}

0 commit comments

Comments
 (0)