38 lines
876 B
Go
38 lines
876 B
Go
package proxy
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/cloudwego/hertz/pkg/app"
|
|
)
|
|
|
|
/*
|
|
// 设置请求头
|
|
func setRequestHeaders(c *app.RequestContext, req *http.Request) {
|
|
c.Request.Header.VisitAll(func(key, value []byte) {
|
|
req.Header.Set(string(key), string(value))
|
|
})
|
|
}
|
|
*/
|
|
|
|
func setRequestHeaders(c *app.RequestContext, req *http.Request) {
|
|
c.Request.Header.VisitAll(func(key, value []byte) {
|
|
headerKey := string(key)
|
|
headerValue := string(value)
|
|
if _, shouldRemove := reqHeadersToRemove[headerKey]; !shouldRemove {
|
|
req.Header.Set(headerKey, headerValue)
|
|
}
|
|
|
|
})
|
|
}
|
|
|
|
/*
|
|
// removeWSHeader removes the "Upgrade" and "Connection" headers from the given
|
|
// Request, which are added by the client when it wants to upgrade the
|
|
// connection to a WebSocket connection.
|
|
func removeWSHeader(req *http.Request) {
|
|
req.Header.Del("Upgrade")
|
|
req.Header.Del("Connection")
|
|
}
|
|
*/
|