From f4a915c98e057d75aa6542661a25a3b0cf40bbf3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= Date: Sun, 7 Feb 2021 09:08:46 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=BB=BA=E5=8F=8D=E5=90=91=E4=BB=A3?= =?UTF-8?q?=E7=90=86=E6=97=B6=E9=BB=98=E8=AE=A4=E9=80=89=E4=B8=AD=E6=89=80?= =?UTF-8?q?=E6=9C=89Header?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/db/models/reverse_proxy_dao.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/internal/db/models/reverse_proxy_dao.go b/internal/db/models/reverse_proxy_dao.go index 6e5a3200..b3928728 100644 --- a/internal/db/models/reverse_proxy_dao.go +++ b/internal/db/models/reverse_proxy_dao.go @@ -159,7 +159,13 @@ func (this *ReverseProxyDAO) CreateReverseProxy(tx *dbs.Tx, adminId int64, userI op.State = ReverseProxyStateEnabled op.AdminId = adminId op.UserId = userId - op.AddHeaders = "[\"X-Real-IP\"]" + + defaultHeaders := []string{"X-Real-IP", "X-Forwarded-For", "X-Forwarded-By", "X-Forwarded-Host", "X-Forwarded-Proto"} + defaultHeadersJSON, err := json.Marshal(defaultHeaders) + if err != nil { + return 0, err + } + op.AddHeaders = defaultHeadersJSON if len(schedulingJSON) > 0 { op.Scheduling = string(schedulingJSON) @@ -170,7 +176,7 @@ func (this *ReverseProxyDAO) CreateReverseProxy(tx *dbs.Tx, adminId int64, userI if len(backupOriginsJSON) > 0 { op.BackupOrigins = string(backupOriginsJSON) } - err := this.Save(tx, op) + err = this.Save(tx, op) if err != nil { return 0, err }