From 612d56c5e953193ff13fb6e88d679a30f6e92bb5 Mon Sep 17 00:00:00 2001
From: henrylee2cn <henrylee_cn@foxmail.com>
Date: Mon, 12 Mar 2018 19:25:34 +0800
Subject: [PATCH] Plugin: Update proxy

---
 plugin/proxy.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/plugin/proxy.go b/plugin/proxy.go
index 2a19ec24..eaf7376e 100644
--- a/plugin/proxy.go
+++ b/plugin/proxy.go
@@ -87,7 +87,12 @@ func (p *proxy) pull(ctx tp.UnknownPullCtx) (interface{}, *tp.Rerror) {
 	pullcmd.InputMeta().VisitAll(func(key, value []byte) {
 		ctx.SetMeta(goutil.BytesToString(key), goutil.BytesToString(value))
 	})
-	return reply, pullcmd.Rerror()
+	rerr := pullcmd.Rerror()
+	if rerr != nil && rerr.Code < 200 && rerr.Code > 99 {
+		rerr.Code = tp.CodeBadGateway
+		rerr.Message = tp.CodeText(tp.CodeBadGateway)
+	}
+	return reply, rerr
 }
 
 func (p *proxy) push(ctx tp.UnknownPushCtx) *tp.Rerror {
@@ -98,5 +103,10 @@ func (p *proxy) push(ctx tp.UnknownPushCtx) *tp.Rerror {
 	if len(ctx.PeekMeta(tp.MetaRealIp)) == 0 {
 		settings = append(settings, tp.WithAddMeta(tp.MetaRealIp, ctx.Ip()))
 	}
-	return p.pushFunc(ctx.Uri(), ctx.InputBodyBytes(), settings...)
+	rerr := p.pushFunc(ctx.Uri(), ctx.InputBodyBytes(), settings...)
+	if rerr != nil && rerr.Code < 200 && rerr.Code > 99 {
+		rerr.Code = tp.CodeBadGateway
+		rerr.Message = tp.CodeText(tp.CodeBadGateway)
+	}
+	return rerr
 }