Allow file payload in HTTP invoker.

Signed-off-by: Luka Simić <git@kocka.tech>

Author: KockaAdmiralac

pkg/config/parser.go

@@ -62,6 +62,7 @@ type LoaderConfiguration struct {
 	RpsMemoryMB                 int     `json:"RpsMemoryMB"`
 	RpsIterationMultiplier      int     `json:"RpsIterationMultiplier"`
 	RpsDataSizeMB               float64 `json:"RpsDataSizeMB"`
+	RpsFile                     string  `json:"RpsFile"`
 
 	TracePath          string `json:"TracePath"`
 	Granularity        string `json:"Granularity"`

pkg/driver/clients/http_client.go

@@ -5,7 +5,9 @@ import (
 	"crypto/rand"
 	"encoding/json"
 	"io"
+	"mime/multipart"
 	"net/http"
+	"os"
 	"strconv"
 	"strings"
 	"time"
@@ -36,11 +38,11 @@ func newHTTPInvoker(cfg *config.LoaderConfiguration) *httpInvoker {
 }
 
 var payload []byte = nil
+var contentType string = "application/octet-stream"
 
-func CreateRequestPayload(sizeInMB float64) *bytes.Buffer {
-	byteCount := int(sizeInMB * 1024.0 * 1024.0) // MB -> B
-
+func CreateRandomPayload(sizeInMB float64) *bytes.Buffer {
 	if payload == nil {
+		byteCount := int(sizeInMB * 1024.0 * 1024.0) // MB -> B
 		payload = make([]byte, byteCount)
 
 		n, err := rand.Read(payload)
@@ -52,6 +54,35 @@ func CreateRequestPayload(sizeInMB float64) *bytes.Buffer {
 	return bytes.NewBuffer(payload)
 }
 
+func CreateFilePayload(filePath string) *bytes.Buffer {
+	if payload == nil {
+		file, err := os.Open(filePath)
+		if err != nil {
+			log.Fatalf("Failed to open file %s: %v", filePath, err)
+		}
+
+		buffer := &bytes.Buffer{}
+		writer := multipart.NewWriter(buffer)
+		part, err := writer.CreateFormFile("images", "invitro.payload")
+		if err != nil {
+			log.Fatalf("Failed to create form file: %v", err)
+		}
+
+		if _, err = io.Copy(part, file); err != nil {
+			log.Fatalf("Failed to enter file into the form: %v", err)
+		}
+		if err = writer.Close(); err != nil {
+			log.Fatalf("Failed to close writer: %v", err)
+		}
+
+		payload = buffer.Bytes()
+		contentType = writer.FormDataContentType()
+		return buffer
+	}
+
+	return bytes.NewBuffer(payload)
+}
+
 func (i *httpInvoker) Invoke(function *common.Function, runtimeSpec *common.RuntimeSpecification) (bool, *mc.ExecutionRecord) {
 	isDandelion := strings.Contains(strings.ToLower(i.cfg.Platform), "dandelion")
 	isKnative := strings.Contains(strings.ToLower(i.cfg.Platform), "knative")
@@ -77,14 +108,20 @@ func (i *httpInvoker) Invoke(function *common.Function, runtimeSpec *common.Runt
 	}
 	if i.cfg.RpsTarget != 0 {
 		ts := time.Now()
-		requestBody = CreateRequestPayload(i.cfg.RpsDataSizeMB)
-		log.Debugf("Took %v to generate request body.", time.Since(ts))
+		if i.cfg.RpsFile != "" {
+			requestBody = CreateFilePayload(i.cfg.RpsFile)
+			log.Debugf("Took %v to create file body.", time.Since(ts))
+		} else {
+			requestBody = CreateRandomPayload(i.cfg.RpsDataSizeMB)
+			log.Debugf("Took %v to generate request body.", time.Since(ts))
+		}
 	}
 
 	start := time.Now()
 	record.StartTime = start.UnixMicro()
 
 	req, err := http.NewRequest("POST", "http://"+function.Endpoint, requestBody)
+	req.Header.Add("Content-Type", contentType)
 	if err != nil {
 		log.Errorf("Failed to create a HTTP request - %v\n", err)