Allow for headers to be sent with HTTPRequest and provide headers back with HTTPResponse

Author: Jon Bernard

ldk/go/examples/network/loop/loop.go

@@ -3,6 +3,7 @@ package loop
 import (
 	"context"
 	"encoding/json"
+	"net/http"
 	"time"
 
 	ldk "github.com/open-olive/loop-development-kit/ldk/go"
@@ -73,15 +74,26 @@ func (c *Loop) LoopStart(sidekick ldk.Sidekick) error {
 		URL:    "https://api.fda.gov/food/enforcement.json?search=report_date:[" + now.AddDate(0, -3, 0).Format("20060102") + "+TO+" + now.Format("20060102") + "]&limit=1",
 		Method: "GET",
 		Body:   nil,
+		Headers: map[string]string{
+			"A-Header-Name": "a header value",
+		},
 	})
 	if err != nil {
 		c.logger.Error("received error from callback", err)
 		return err
 	}
 
-	var data apiResponse
-
 	if response.ResponseCode == 200 {
+		var headers http.Header
+		var data apiResponse
+
+		if err := json.Unmarshal(response.Headers, &headers); err != nil {
+			c.logger.Error("Error unmarshaling response headers", err)
+			return err
+		}
+
+		c.logger.Info("Response headers", headers)
+
 		if err := json.Unmarshal(response.Data, &data); err != nil {
 			c.logger.Error("Error unmarshaling response payload", err)
 			return err

ldk/go/networkClient.go

@@ -16,13 +16,15 @@ type NetworkClient struct {
 type HTTPResponse struct {
 	ResponseCode int
 	Data         []byte
+	Headers      []byte
 }
 
 // HTTPRequest is the structure received from HttpRequest
 type HTTPRequest struct {
-	URL    string
-	Method string
-	Body   []byte
+	URL     string
+	Method  string
+	Body    []byte
+	Headers map[string]string
 }
 
 func (n *NetworkClient) HTTPRequest(ctx context.Context, req *HTTPRequest) (*HTTPResponse, error) {
@@ -31,6 +33,7 @@ func (n *NetworkClient) HTTPRequest(ctx context.Context, req *HTTPRequest) (*HTT
 		Url:     req.URL,
 		Method:  req.Method,
 		Body:    req.Body,
+		Headers: req.Headers,
 	})
 
 	if err != nil {
@@ -40,5 +43,6 @@ func (n *NetworkClient) HTTPRequest(ctx context.Context, req *HTTPRequest) (*HTT
 	return &HTTPResponse{
 		ResponseCode: int(resp.ResponseCode),
 		Data:         resp.Data,
+		Headers:      resp.Headers,
 	}, err
 }

ldk/go/networkServer.go

@@ -19,9 +19,10 @@ func (ns *NetworkServer) HTTPRequest(ctx context.Context, req *proto.HTTPRequest
 	resp, err := ns.Impl.HTTPRequest(
 		context.WithValue(ctx, Session{}, session),
 		&HTTPRequest{
-			URL:    req.Url,
-			Method: req.Method,
-			Body:   req.Body,
+			URL:     req.Url,
+			Method:  req.Method,
+			Body:    req.Body,
+			Headers: req.Headers,
 		},
 	)
 	if err != nil {
@@ -31,6 +32,7 @@ func (ns *NetworkServer) HTTPRequest(ctx context.Context, req *proto.HTTPRequest
 	return &proto.HTTPResponseMsg{
 		ResponseCode: uint32(resp.ResponseCode),
 		Data:         resp.Data,
+		Headers:      resp.Headers,
 	}, nil
 
 }

ldk/go/proto/network.pb.go

@@ -13,9 +13,11 @@ message HTTPRequestMsg {
     string url = 2;
     string method = 3;
     bytes body = 4;
+    map <string, string> headers = 5;
 }
 
 message HTTPResponseMsg {
     uint32 responseCode = 1;
     bytes data = 2;
+    bytes headers = 3;
 }