Bumped v0.7.2

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

cube.go

@@ -111,14 +111,15 @@ func (c *Cube) dispatch() error {
 // Start starts recording an HTTP request.
 func (c *Cube) Start(r *http.Request, w http.ResponseWriter) (request *CubeRequest) {
 	request = &CubeRequest{
+		ID:        RequestID(r, w),
 		Time:      time.Now(),
 		AppID:     c.AppID,
 		AppName:   c.AppName,
 		Host:      r.Host,
 		Path:      r.URL.Path,
 		Method:    r.Method,
 		UserAgent: r.UserAgent(),
-		RemoteIP:  realIP(r),
+		RemoteIP:  RealIP(r),
 	}
 	request.ClientID = request.RemoteIP
 	atomic.AddInt64(&c.activeRequests, 1)

util.go

@@ -11,7 +11,17 @@ const (
 	rfc3339Micro = "2006-01-02T15:04:05.000000Z07:00"
 )
 
-func realIP(r *http.Request) string {
+// RequestID returns the request ID from the request or response.
+func RequestID(r *http.Request, w http.ResponseWriter) string {
+	id := r.Header.Get("X-Request-ID")
+	if id == "" {
+		id = w.Header().Get("X-Request-ID")
+	}
+	return id
+}
+
+// RealIP returns the real IP from the request.
+func RealIP(r *http.Request) string {
 	ra := r.RemoteAddr
 	if ip := r.Header.Get("X-Forwarded-For"); ip != "" {
 		ra = strings.Split(ip, ", ")[0]