mw_context_vars.go

package main

import (
	"net/http"
	"strings"

	"github.com/satori/go.uuid"
)

type MiddlewareContextVars struct {
	*BaseMiddleware
}

func (m *MiddlewareContextVars) GetName() string {
	return "MiddlewareContextVars"
}

func (m *MiddlewareContextVars) IsEnabledForSpec() bool {
	return m.Spec.EnableContextVars
}

// ProcessRequest will run any checks on the request on the way through the system, return an error to have the chain fail
func (m *MiddlewareContextVars) ProcessRequest(w http.ResponseWriter, r *http.Request, _ interface{}) (error, int) {

	if !m.Spec.EnableContextVars {
		return nil, 200
	}

	copiedRequest := CopyHttpRequest(r)
	contextDataObject := make(map[string]interface{})

	copiedRequest.ParseForm()

	// Form params (map[string][]string)
	contextDataObject["request_data"] = copiedRequest.Form

	contextDataObject["headers"] = map[string][]string(copiedRequest.Header)

	for hname, vals := range copiedRequest.Header {
		n := "headers_" + strings.Replace(hname, "-", "_", -1)
		contextDataObject[n] = vals[0]
	}

	// Path parts
	segmentedPathArray := strings.Split(copiedRequest.URL.Path, "/")
	contextDataObject["path_parts"] = segmentedPathArray

	// path data
	contextDataObject["path"] = copiedRequest.URL.Path

	// IP:Port
	contextDataObject["remote_addr"] = copiedRequest.RemoteAddr

	//Correlation ID
	contextDataObject["request_id"] = uuid.NewV4().String()

	ctxSetData(r, contextDataObject)

	return nil, 200
}