コード例 #1
0
ファイル: middleware_test.go プロジェクト: chendx79/gafka
// 764 ns/op 96 B/op 4 allocs/op
func BenchmarkBuildCommonLogLine(b *testing.B) {
	gw := &Gateway{}
	r, err := mockHttpRequest()
	if err != nil {
		b.Fatal(err)
	}

	b.ReportAllocs()
	for i := 0; i < b.N; i++ {
		buf := mpool.AccessLogLineBufferGet()[0:]
		gw.buildCommonLogLine(buf, r, 200, 100)
		mpool.AccessLogLineBufferPut(buf)
	}
}
コード例 #2
0
ファイル: middleware.go プロジェクト: funkygao/gafka
func (this *Gateway) middleware(h httprouter.Handle) httprouter.Handle {
	var (
		// GC will touch every single item of the map during mark and scan phase
		// Go 1.5 https://github.com/golang/go/issues/9477
		// TODO map[int64]int
		connections   = make(map[string]int, 500) // remoteAddr:counter
		connectionsMu sync.Mutex
	)

	return func(w http.ResponseWriter, r *http.Request, params httprouter.Params) {
		w.Header().Set("Server", "kateway")

		// kateway response is mostly json, including error reponse
		// for non-json response, handler can override this
		w.Header().Set("Content-Type", "application/json; charset=utf8")

		// CORS: cross origin resource sharing
		if origin := r.Header.Get("Origin"); origin != "" {
			w.Header().Set("Access-Control-Allow-Origin", origin)
			w.Header().Set("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS")
			w.Header().Set("Access-Control-Allow-Headers", "Origin, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token")
			w.Header().Set("Access-Control-Allow-Credentials", "true")
		}

		// max request per conn to rebalance the session sticky http conns
		if Options.MaxRequestPerConn > 1 {
			maxReqReached := false
			connectionsMu.Lock()
			if n, present := connections[r.RemoteAddr]; present && n >= Options.MaxRequestPerConn {
				maxReqReached = true
				delete(connections, r.RemoteAddr)
			} else {
				connections[r.RemoteAddr]++ // in golang, works even when present=false
			}
			connectionsMu.Unlock()

			if maxReqReached {
				log.Trace("%s max req per conn reached: %d", r.RemoteAddr, Options.MaxRequestPerConn)

				w.Header().Set("Connection", "close")
			}
		}

		if !Options.EnableAccessLog {
			h(w, r, params)

			return
		}

		// TODO latency histogram here
		// TODO slow response recording here

		ww := SniffWriter(w) // sniff the status and content size for logging
		h(ww, r, params)     // delegate request to the given handle

		if this.accessLogger != nil {
			// NCSA Common Log Format (CLF)
			// host ident authuser date request status bytes

			// TODO whitelist
			buf := mpool.AccessLogLineBufferGet()[0:]
			this.accessLogger.Log(this.buildCommonLogLine(buf, r, ww.Status(), ww.BytesWritten()))
			mpool.AccessLogLineBufferPut(buf)
		}
	}
}