Add request/response header processing support

enginyoyen · Apr 8, 2021 · dbfb519 · dbfb519
1 parent ca58bbd
commit dbfb519
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/pkg/mapping/mapping.go b/pkg/mapping/mapping.go
@@ -15,14 +15,16 @@ type Mapping struct {
 }
 
 type Request struct {
-	Url    string
-	Method string
+	Url     string
+	Method  string
+	Headers map[string]string
 }
 
 type Response struct {
 	Status    int
 	Body      json.RawMessage
 	WithDelay time.Duration
+	Headers   map[string]string
 }
 
 func MapRequests(files *[]string) *[]Mapping {

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -22,7 +22,11 @@ func Serve(config *cmd.Config, maps *[]mapping.Mapping) {
 
 	for _, v := range *mappings {
 		m := v
-		r.HandleFunc(m.Request.Url, func(w http.ResponseWriter, r *http.Request) {
+		route := r.HandleFunc(m.Request.Url, func(w http.ResponseWriter, r *http.Request) {
+			// Set headers in the
+			for header, value := range m.Response.Headers {
+				w.Header().Set(header, value)
+			}
 			w.WriteHeader(m.Response.Status)
 			body := string(m.Response.Body)
 			body = strings.TrimLeft(body, "\"")
@@ -32,6 +36,10 @@ func Serve(config *cmd.Config, maps *[]mapping.Mapping) {
 				time.Sleep(time.Millisecond * m.Response.WithDelay)
 			}
 		}).Methods(m.Request.Method)
+		//Set request headers
+		for header, value := range m.Request.Headers {
+			route.Headers(header, value)
+		}
 	}
 
 	addr := config.BindAddress + ":" + strconv.Itoa(config.Port)