forked from bfenetworks/bfe
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathreverseproxy.go
785 lines (680 loc) · 23.5 KB
/
reverseproxy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
// Copyright (c) 2019 Baidu, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Copyright 2011 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// HTTP reverse proxy handler
package bfe_server
import (
"io"
"net"
"reflect"
"sync"
"time"
)
import (
"github.com/baidu/go-lib/log"
)
import (
bfe_cluster_backend "github.com/baidu/bfe/bfe_balance/backend"
bal_gslb "github.com/baidu/bfe/bfe_balance/bal_gslb"
"github.com/baidu/bfe/bfe_basic"
"github.com/baidu/bfe/bfe_config/bfe_cluster_conf/cluster_conf"
"github.com/baidu/bfe/bfe_debug"
"github.com/baidu/bfe/bfe_http"
"github.com/baidu/bfe/bfe_http2"
"github.com/baidu/bfe/bfe_module"
"github.com/baidu/bfe/bfe_route"
"github.com/baidu/bfe/bfe_route/bfe_cluster"
"github.com/baidu/bfe/bfe_spdy"
"github.com/baidu/bfe/bfe_util"
)
// RoundTripperMap holds mappings from cluster-name to RoundTripper.
type RoundTripperMap map[string]bfe_http.RoundTripper
// ReverseProxy takes an incoming request and sends it to another server,
// proxying the response back to the client.
type ReverseProxy struct {
// The transport used to perform proxy requests.
// If no transport from clustername->transport map, create one.
tsMu sync.RWMutex
transports RoundTripperMap
server *BfeServer // link to bfe server
proxyState *ProxyState // state of proxy
}
// NewReverseProxy returns a new ReverseProxy.
func NewReverseProxy(server *BfeServer, state *ProxyState) *ReverseProxy {
rp := new(ReverseProxy)
rp.transports = make(RoundTripperMap)
rp.server = server
rp.proxyState = state
return rp
}
// Hop-by-hop headers. These are removed when sent to the backend.
// http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html
var hopHeaders = []string{
"Connection",
"Keep-Alive",
"Proxy-Authenticate",
"Proxy-Authorization",
"Te", // canonicalized version of "TE"
"Trailers",
"Transfer-Encoding",
"Upgrade",
}
// httpProtoSet set http proto for out request.
func httpProtoSet(outreq *bfe_http.Request) {
outreq.Proto = "HTTP/1.1"
outreq.ProtoMajor = 1
outreq.ProtoMinor = 1
outreq.Close = false
}
// hopByHopHeaderRemove remove hop-by-hop headers.
func hopByHopHeaderRemove(outreq, req *bfe_http.Request) {
// Remove hop-by-hop headers to the backend. Especially
// important is "Connection" because we want a persistent
// connection, regardless of what the client sent to us. This
// is modifying the same underlying map from req (shallow
// copied above) so we only copy it if necessary.
copiedHeaders := false
for _, h := range hopHeaders {
if outreq.Header.Get(h) != "" {
if !copiedHeaders {
outreq.Header = make(bfe_http.Header, len(req.Header))
bfe_http.CopyHeader(outreq.Header, req.Header)
copiedHeaders = true
}
outreq.Header.Del(h)
}
}
}
// setBackendAddr set backend addr to host of request url.
func setBackendAddr(req *bfe_http.Request, backend *bfe_cluster_backend.BfeBackend) {
req.URL.Scheme = "http"
req.URL.Host = backend.GetAddrInfo()
}
func (p *ReverseProxy) setTransports(clusterMap bfe_route.ClusterMap) {
p.tsMu.Lock()
defer p.tsMu.Unlock()
newTransports := make(RoundTripperMap)
for cluster, conf := range clusterMap {
transport, ok := p.transports[cluster]
if !ok {
transport = createTransport(conf)
newTransports[cluster] = transport
continue
}
t := transport.(*bfe_http.Transport)
// get transport, check if transport needs update
backendConf := conf.BackendConf()
if (t.MaxIdleConnsPerHost != *backendConf.MaxIdleConnsPerHost) ||
(t.ResponseHeaderTimeout != time.Millisecond*time.Duration(*backendConf.TimeoutResponseHeader)) ||
(t.ReqWriteBufferSize != conf.ReqWriteBufferSize()) ||
(t.ReqFlushInterval != conf.ReqFlushInterval()) {
// create new transport with newConf instead of update transport
// update transport needs lock
transport = createTransport(conf)
newTransports[cluster] = transport
continue
}
newTransports[cluster] = transport
}
p.transports = newTransports
}
// getTransport return transport from map, if not exist, create a transport.
func (p *ReverseProxy) getTransport(cluster *bfe_cluster.BfeCluster) bfe_http.RoundTripper {
p.tsMu.RLock()
defer p.tsMu.RUnlock()
transport, ok := p.transports[cluster.Name]
if !ok {
transport = createTransport(cluster)
p.transports[cluster.Name] = transport
}
return transport
}
func createTransport(cluster *bfe_cluster.BfeCluster) bfe_http.RoundTripper {
backendConf := cluster.BackendConf()
log.Logger.Debug("create a new transport for %s, timeout %d", cluster.Name, *backendConf.TimeoutResponseHeader)
// cluster has its own Connect Server Timeout.
// so each cluster has a different transport
// once cluster's timeout updated, dailer use new value
dailer := func(network, add string) (net.Conn, error) {
timeout := time.Duration(cluster.TimeoutConnSrv()) * time.Millisecond
return net.DialTimeout(network, add, timeout)
}
return &bfe_http.Transport{
Dial: dailer,
DisableKeepAlives: (*backendConf.MaxIdleConnsPerHost) == 0,
MaxIdleConnsPerHost: *backendConf.MaxIdleConnsPerHost,
ResponseHeaderTimeout: time.Millisecond * time.Duration(*backendConf.TimeoutResponseHeader),
ReqWriteBufferSize: cluster.ReqWriteBufferSize(),
ReqFlushInterval: cluster.ReqFlushInterval(),
DisableCompression: true,
}
}
// clusterInvoke invoke cluster to get response.
func (p *ReverseProxy) clusterInvoke(srv *BfeServer, cluster *bfe_cluster.BfeCluster,
request *bfe_basic.Request, rw bfe_http.ResponseWriter) (
res *bfe_http.Response, action int, err error) {
var clusterBackend *bfe_cluster_backend.BfeBackend
var bal *bal_gslb.BalanceGslb
var outreq *bfe_http.Request = request.OutRequest
// mark start/end of cluster invoke
request.Stat.ClusterStart = time.Now()
defer func() {
request.Stat.ClusterEnd = time.Now()
}()
clusterTransport := p.getTransport(cluster)
// look up for balance
bal, err = srv.balTable.Lookup(cluster.Name)
if err != nil {
log.Logger.Warn("no balance for %s", cluster.Name)
request.Stat.ResponseStart = time.Now()
request.ErrCode = bfe_basic.ErrBkNoCluster
request.ErrMsg = err.Error()
p.proxyState.ErrBkNoBalance.Inc(1)
action = closeAfterReply
return
}
// When request.RetryTime exceeds some value, srv.clusterTable.Lookup()
// will return error. Here set a limit of 20, to avoid endless loop
for i := 0; i < 20; i++ {
// get backend with cluster-name and request
clusterBackend, err = bal.Balance(request)
if err == bfe_basic.ErrBkCrossRetryBalance {
request.RetryTime += 1
continue
}
if err != nil {
// p.proxystate counter is set by bal.Balance(), only log
log.Logger.Warn("cluster [%s] select backend failed, err[%s]", cluster.Name,
err.Error())
break
}
// err == nil if and only if we choose a new backend,
// desc old backend connection num
if request.Trans.Backend != nil {
request.Trans.Backend.DecConnNum()
request.Trans.Backend = nil
}
request.SetRequestTransport(clusterBackend, clusterTransport)
log.Logger.Debug("ReverseProxy.Invoke(): before HANDLE_FORWARD backend %s:%d",
request.Trans.Backend.Addr, request.Trans.Backend.Port)
// Callback for HANDLE_FORWARD
hl := srv.CallBacks.GetHandlerList(bfe_module.HANDLE_FORWARD)
if hl != nil {
retVal := hl.FilterForward(request)
switch retVal {
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
return
}
}
log.Logger.Debug("ReverseProxy.Invoke(): after HANDLE_FORWARD backend %s:%d",
request.Trans.Backend.Addr, request.Trans.Backend.Port)
// set backend addr to out request
backend := request.Trans.Backend
backend.AddConnNum()
setBackendAddr(outreq, backend)
// invoke backend
request.Stat.BackendStart = time.Now()
if i == 0 {
// record start time of the first try
request.Stat.BackendFirst = request.Stat.BackendStart
}
transport := request.Trans.Transport
res, err = transport.RoundTrip(outreq)
request.Stat.BackendEnd = time.Now()
// record backend info to request, no matter succeed or fail
request.Backend.SubclusterName = backend.SubCluster
request.Backend.BackendName = backend.Name
request.Backend.BackendAddr = backend.Addr
request.Backend.BackendPort = uint32(backend.Port)
if err == nil {
// succeed in invoking backend
backend.OnSuccess()
// clear err msg in req.
// this step is required, if finally succeed after retry
request.ErrCode = nil
request.ErrMsg = ""
// record body size of request after forward
request.Stat.BodyLenIn = int(outreq.State.BodySize)
if bfe_debug.DebugServHTTP {
log.Logger.Debug("ReverseProxy.ServeHTTP(): get response from %s", backend.Name)
}
break
}
// fail in invoking backend
log.Logger.Info("[%s] [%s:%d] roundtrip %s", cluster.Name, backend.Addr, backend.Port, err)
p.proxyState.ErrBkRequestBackend.Inc(1)
// deal with errors here, possible error type:
// 1. connect backend error
// 2. read client request body error(POST/PUT)
// 3. write backend error
// a. haven't write any byte
// b. aleady write part of data
// 4. read backend error
// 5. other error
allowRetry := false
switch err.(type) {
case bfe_http.ConnectError:
// if error happens in dial phrase, we can retry
request.ErrCode = bfe_basic.ErrBkConnectBackend
request.ErrMsg = err.Error()
p.proxyState.ErrBkConnectBackend.Inc(1)
allowRetry = true
backend.OnFail(cluster.Name)
case bfe_http.WriteRequestError:
request.ErrCode = bfe_basic.ErrBkWriteRequest
request.ErrMsg = err.Error()
p.proxyState.ErrBkWriteRequest.Inc(1)
allowRetry = checkAllowRetry(cluster.RetryLevel(), outreq)
// if error is caused by backend server
rerr := err.(bfe_http.WriteRequestError)
if !rerr.CheckTargetError(request.RemoteAddr) {
backend.OnFail(cluster.Name)
}
case bfe_http.ReadRespHeaderError:
request.ErrCode = bfe_basic.ErrBkReadRespHeader
request.ErrMsg = err.Error()
p.proxyState.ErrBkReadRespHeader.Inc(1)
allowRetry = checkAllowRetry(cluster.RetryLevel(), outreq)
backend.OnFail(cluster.Name)
case bfe_http.RespHeaderTimeoutError:
request.ErrCode = bfe_basic.ErrBkRespHeaderTimeout
request.ErrMsg = err.Error()
p.proxyState.ErrBkRespHeaderTimeout.Inc(1)
allowRetry = checkAllowRetry(cluster.RetryLevel(), outreq)
backend.OnFail(cluster.Name)
case bfe_http.TransportBrokenError:
request.ErrCode = bfe_basic.ErrBkTransportBroken
request.ErrMsg = err.Error()
p.proxyState.ErrBkTransportBroken.Inc(1)
allowRetry = checkAllowRetry(cluster.RetryLevel(), outreq)
default:
// never go here
log.Logger.Info("roundtrip %s %s", reflect.TypeOf(err), err)
}
if !allowRetry {
log.Logger.Debug("request fail, not retry now")
p.proxyState.ClientReqFailWithNoRetry.Inc(1)
break
}
request.RetryTime += 1
}
// have retry?
if request.RetryTime > 0 {
p.proxyState.ClientReqWithRetry.Inc(1)
}
// have cross-cluster retry?
if request.Stat.IsCrossCluster {
p.proxyState.ClientReqWithCrossRetry.Inc(1)
}
log.Logger.Debug("clusterInvoke %v %v", res, err)
return
}
// sendResponse send http response to client.
func (p *ReverseProxy) sendResponse(rw bfe_http.ResponseWriter, res *bfe_http.Response,
flushInterval time.Duration, cancelOnClientClose bool) error {
// prepare SignCalculater for response
p.prepareSigner(rw, res)
bfe_http.CopyHeader(rw.Header(), res.Header)
// note: writeheader don't guarantee send header
rw.WriteHeader(res.StatusCode)
return p.copyResponse(rw, res.Body, flushInterval, cancelOnClientClose)
}
// prepareSigner prepare SignCalculater for response.
func (p *ReverseProxy) prepareSigner(rw bfe_http.ResponseWriter, res *bfe_http.Response) {
// not need to add signature for respsone
if res.Signer == nil {
return
}
// prepare Singer for signature
if resp, ok := rw.(*response); ok {
resp.SetSigner(res.Signer)
}
}
// FinishReq should be invoked after quit ServHTTP().
func (p *ReverseProxy) FinishReq(rw bfe_http.ResponseWriter, request *bfe_basic.Request) (action int) {
// get instance of BfeServer
srv := p.server
// desc connection num after request finish
defer func() {
// desc backend connection counter
if request.Trans.Backend != nil {
request.Trans.Backend.DecConnNum()
}
}()
// Callback for HANDLE_REQUEST_FINISH
hl := srv.CallBacks.GetHandlerList(bfe_module.HANDLE_REQUEST_FINISH)
if hl != nil {
retVal := hl.FilterResponse(request, request.HttpResponse)
switch retVal {
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
return
}
}
return
}
func (p *ReverseProxy) setTimeout(stage bfe_basic.OperationStage,
conn net.Conn, req *bfe_http.Request, d time.Duration) {
switch b := req.Body.(type) {
case *bfe_http2.RequestBody: // http2
if stage == bfe_basic.StageReadReqBody {
bfe_http2.SetReadStreamTimeout(b, d)
}
if stage == bfe_basic.StageWriteClient {
bfe_http2.SetWriteStreamTimeout(b, d)
}
if stage == bfe_basic.StageEndRequest {
bfe_http2.SetConnTimeout(b, d)
}
case *bfe_spdy.RequestBody: // spdy
if stage == bfe_basic.StageReadReqBody {
bfe_spdy.SetReadStreamTimeout(b, d)
}
if stage == bfe_basic.StageWriteClient {
bfe_spdy.SetWriteStreamTimeout(b, d)
}
if stage == bfe_basic.StageEndRequest {
bfe_spdy.SetConnTimeout(b, d)
}
default: // http
if stage == bfe_basic.StageReadReqBody || stage == bfe_basic.StageEndRequest {
conn.SetReadDeadline(time.Now().Add(d))
}
if stage == bfe_basic.StageWriteClient {
conn.SetWriteDeadline(time.Now().Add(d))
}
}
}
func (p *ReverseProxy) setReadClientAgainTimeout(cluster *bfe_cluster.BfeCluster, conn net.Conn) {
// for idle time + read next header time
conn.SetReadDeadline(time.Now().Add(cluster.TimeoutReadClientAgain()))
}
// ServeHTTP processes http request and send http response.
//
// Params:
// - rw : context for sending response
// - request: context for request
//
// Return:
// - action: action to do after ServeHTTP
func (p *ReverseProxy) ServeHTTP(rw bfe_http.ResponseWriter, basicReq *bfe_basic.Request) (action int) {
var err error
var res *bfe_http.Response
var hl *bfe_module.HandlerList
var retVal int
var clusterName string
var cluster *bfe_cluster.BfeCluster
var outreq *bfe_http.Request
var serverConf *bfe_route.ServerDataConf
var writeTimer *time.Timer
req := basicReq.HttpRequest
isRedirect := false
resFlushInterval := time.Duration(0)
cancelOnClientClose := false
// get instance of BfeServer
srv := p.server
// set clientip of original user for request
setClientAddr(basicReq)
// Callback for HANDLE_BEFORE_LOCATION
hl = srv.CallBacks.GetHandlerList(bfe_module.HANDLE_BEFORE_LOCATION)
if hl != nil {
retVal, res = hl.FilterRequest(basicReq)
basicReq.HttpResponse = res
switch retVal {
case bfe_module.BFE_HANDLER_CLOSE:
// close the connection directly (with no response)
action = closeDirectly
return
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
basicReq.BfeStatusCode = bfe_http.StatusInternalServerError
return
case bfe_module.BFE_HANDLER_REDIRECT:
// make redirect
Redirect(rw, req, basicReq.Redirect.Url, basicReq.Redirect.Code)
isRedirect = true
basicReq.BfeStatusCode = basicReq.Redirect.Code
goto send_response
case bfe_module.BFE_HANDLER_RESPONSE:
goto response_got
}
}
// find product
if err := srv.findProduct(basicReq); err != nil {
basicReq.ErrCode = bfe_basic.ErrBkFindProduct
basicReq.ErrMsg = err.Error()
p.proxyState.ErrBkFindProduct.Inc(1)
log.Logger.Info("FindProduct error[%s] host[%s] vip[%s] clientip[%s]", err.Error(),
basicReq.HttpRequest.Host, basicReq.Session.Vip, basicReq.ClientAddr)
// close connection
res = bfe_basic.CreateInternalSrvErrResp(basicReq)
action = closeAfterReply
goto response_got
}
// Callback for HANDLE_FOUND_PRODUCT
hl = srv.CallBacks.GetHandlerList(bfe_module.HANDLE_FOUND_PRODUCT)
if hl != nil {
retVal, res = hl.FilterRequest(basicReq)
basicReq.HttpResponse = res
switch retVal {
case bfe_module.BFE_HANDLER_CLOSE:
// close the connection directly (with no response)
action = closeDirectly
return
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
basicReq.BfeStatusCode = bfe_http.StatusInternalServerError
return
case bfe_module.BFE_HANDLER_REDIRECT:
// make redirect
Redirect(rw, req, basicReq.Redirect.Url, basicReq.Redirect.Code)
isRedirect = true
basicReq.BfeStatusCode = basicReq.Redirect.Code
goto send_response
case bfe_module.BFE_HANDLER_RESPONSE:
goto response_got
}
}
// find cluster
if err = srv.findCluster(basicReq); err != nil {
basicReq.ErrCode = bfe_basic.ErrBkFindLocation
basicReq.ErrMsg = err.Error()
p.proxyState.ErrBkFindLocation.Inc(1)
log.Logger.Info("FindLocation error[%s] host[%s]", err, basicReq.HttpRequest.Host)
// close connection
res = bfe_basic.CreateInternalSrvErrResp(basicReq)
action = closeAfterReply
goto response_got
}
clusterName = basicReq.Route.ClusterName
// look up for cluster
serverConf = basicReq.SvrDataConf.(*bfe_route.ServerDataConf)
cluster, err = serverConf.ClusterTable.Lookup(clusterName)
if err != nil {
log.Logger.Warn("no cluster for %s", clusterName)
basicReq.Stat.ResponseStart = time.Now()
basicReq.ErrCode = bfe_basic.ErrBkNoCluster
basicReq.ErrMsg = err.Error()
p.proxyState.ErrBkNoCluster.Inc(1)
res = bfe_basic.CreateInternalSrvErrResp(basicReq)
action = closeAfterReply
goto response_got
}
basicReq.Backend.ClusterName = clusterName
// set deadline to finish read client request body
p.setTimeout(bfe_basic.StageReadReqBody, basicReq.Connection, req, cluster.TimeoutReadClient())
// Callback for HANDLE_AFTER_LOCATION
hl = srv.CallBacks.GetHandlerList(bfe_module.HANDLE_AFTER_LOCATION)
if hl != nil {
retVal, res = hl.FilterRequest(basicReq)
basicReq.HttpResponse = res
switch retVal {
case bfe_module.BFE_HANDLER_CLOSE:
// close the connection directly (with no response)
action = closeDirectly
return
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
basicReq.BfeStatusCode = bfe_http.StatusInternalServerError
return
case bfe_module.BFE_HANDLER_REDIRECT:
// make redirect
Redirect(rw, req, basicReq.Redirect.Url, basicReq.Redirect.Code)
isRedirect = true
basicReq.BfeStatusCode = basicReq.Redirect.Code
goto send_response
case bfe_module.BFE_HANDLER_RESPONSE:
goto response_got
}
}
if bfe_debug.DebugServHTTP {
log.Logger.Debug("ReverseProxy.ServeHTTP(): cluster name = %s", clusterName)
}
// prepare out request to downstream RS backend
outreq = new(bfe_http.Request)
*outreq = *req // includes shallow copies of maps, but okay
basicReq.OutRequest = outreq
// set http proto for out request
httpProtoSet(outreq)
// remove hop-by-hop headers
hopByHopHeaderRemove(outreq, req)
// invoke cluster to get response
res, action, err = p.clusterInvoke(srv, cluster, basicReq, rw)
basicReq.HttpResponse = res
if err != nil {
basicReq.Stat.ResponseStart = time.Now()
basicReq.BfeStatusCode = bfe_http.StatusInternalServerError
res = bfe_basic.CreateInternalSrvErrResp(basicReq)
goto response_got
}
resFlushInterval = cluster.ResFlushInterval()
cancelOnClientClose = cluster.CancelOnClientClose()
// timeout for write response to client
// Note: we use io.Copy() to read from backend and write to client.
// For avoid from blocking on client conn or backend conn forever,
// we must timeout both conns after specified duration.
p.setTimeout(bfe_basic.StageWriteClient, basicReq.Connection, req, cluster.TimeoutWriteClient())
writeTimer = time.AfterFunc(cluster.TimeoutWriteClient(), func() {
transport := basicReq.Trans.Transport.(*bfe_http.Transport)
transport.CancelRequest(basicReq.OutRequest) // force close connection to backend
})
defer writeTimer.Stop()
// for read next request
defer p.setTimeout(bfe_basic.StageEndRequest, basicReq.Connection, req, cluster.TimeoutReadClientAgain())
defer res.Body.Close()
response_got:
// Callback for HANDLE_READ_BACKEND
hl = srv.CallBacks.GetHandlerList(bfe_module.HANDLE_READ_BACKEND)
if hl != nil {
retVal = hl.FilterResponse(basicReq, res)
switch retVal {
case bfe_module.BFE_HANDLER_FINISH:
// close the connection after response
action = closeAfterReply
basicReq.BfeStatusCode = bfe_http.StatusInternalServerError
return
case bfe_module.BFE_HANDLER_REDIRECT:
// make redirect
Redirect(rw, req, basicReq.Redirect.Url, basicReq.Redirect.Code)
isRedirect = true
basicReq.BfeStatusCode = basicReq.Redirect.Code
goto send_response
}
}
send_response:
// send http response to client
basicReq.Stat.ResponseStart = time.Now()
if !isRedirect && res != nil {
err = p.sendResponse(rw, res, resFlushInterval, cancelOnClientClose)
if err != nil {
// Note: for h2/spdy protocol, not close client conn when send
// response error. h2/spdy module will close conn/stream properly
if !CheckSupportMultiplex(basicReq.Session.Proto) {
action = closeAfterReply
}
basicReq.ErrCode = bfe_basic.ErrClientWrite
basicReq.ErrMsg = err.Error()
p.proxyState.ErrClientWrite.Inc(1)
}
}
return
}
func (p *ReverseProxy) copyResponse(dst io.Writer, src io.ReadCloser,
flushInterval time.Duration, cancelOnClientClose bool) error {
// Note: When server is blocking on read from backend (eg. io.Copy(dst, src)),
// if the client has disconnected, cancel the block operation immediately.
//
// Note: cancelOnClientClose feature must be enabled for AVS client (over http2)
if cancelOnClientClose {
if cn, ok := dst.(bfe_http.CloseNotifier); ok {
cw := bfe_http.NewCloseWatcher(cn, func() {
// Note: src is type of bfe_http.bodyEofSignal. Close() on src will
// close the underlying connection if response not ready.
// Duplicated Close() will be ignore.
src.Close()
})
go cw.WatchLoop()
defer cw.Stop()
}
}
if flushInterval < 0 {
if wf, ok := dst.(bfe_http.WriteFlusher); ok {
// Note: Flush response header immediately
if err := wf.Flush(); err != nil {
return err
}
_, err := bfe_util.CopyWithoutBuffer(wf, src)
return err
}
}
if flushInterval > 0 {
if wf, ok := dst.(bfe_http.WriteFlusher); ok {
mlw := bfe_http.NewMaxLatencyWriter(wf, flushInterval, nil)
go mlw.FlushLoop()
defer mlw.Stop()
dst = mlw
}
}
_, err := io.Copy(dst, src)
return err
}
func checkAllowRetry(retryLevel int, outreq *bfe_http.Request) bool {
if retryLevel == cluster_conf.RetryGet {
// if forward GET request error (eg. backend restart)
if outreq.Method == "GET" && checkRequestWithoutBody(outreq) {
return true
}
}
return false
}
// checkRequestWithoutBody check whether request without entity body.
func checkRequestWithoutBody(req *bfe_http.Request) bool {
// Note: RFC 2616 doesn't explicitly permit nor forbid an
// entity-body on a GET request
if req.Body == nil || req.Body == bfe_http.EofReader {
return true
}
if body, ok := req.Body.(*bfe_spdy.RequestBody); ok {
return body.Eof()
}
return false
}