zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-07 06:59:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix compile errors (#5085)

Browse Source 
Signed-off-by: kevin <wanjunfeng@gmail.com>
This commit is contained in:
Kevin Wan
2025-08-14 23:09:35 +08:00
committed by GitHub
parent d68cf4920c
commit 94562ded74
1 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
gateway/internal/eventhandler.go
View File

@@ -28,6 +28,17 @@ func NewEventHandler(writer io.Writer, resolver jsonpb.AnyResolver) *EventHandle
}
}
func (h *EventHandler) OnReceiveHeaders(md metadata.MD) {
w, ok := h.writer.(http.ResponseWriter)
if ok {
for k, v := range md {
for _, val := range v {
w.Header().Add(k, val)
}
}
}
}
func (h *EventHandler) OnReceiveResponse(message proto.Message) {
if err := h.marshaler.Marshal(h.writer, message); err != nil {
logx.Error(err)
@@ -38,28 +49,17 @@ func (h *EventHandler) OnReceiveTrailers(status *status.Status, md metadata.MD)
w, ok := h.writer.(http.ResponseWriter)
if ok {
for k, v := range md {
for _, v2 := range v {
w.Header().Add(http.CanonicalHeaderKey(k), v2)
for _, val := range v {
w.Header().Add(k, val)
}
}
}
h.Status = status
}
func (h *EventHandler) OnResolveMethod(_ *desc.MethodDescriptor) {
}
func (h *EventHandler) OnSendHeaders(_ metadata.MD) {
}
func (h *EventHandler) OnReceiveHeaders(md metadata.MD) {
w, ok := h.writer.(http.ResponseWriter)
if ok {
for k, v := range md {
for _, v2 := range v {
canonicalKey := http.CanonicalHeaderKey(k)
for _, v2 := range v {
w.Header().Add(canonicalKey, v2)
}
}
}
}