From 7e12d2819e8e6512e8b2850efb0b1ac5d8c79930 Mon Sep 17 00:00:00 2001 From: Bilal Date: Fri, 8 May 2020 06:32:04 +0200 Subject: [PATCH] Encode/decode request from load balancer server to the worker client --- workerclient/workerclient.go | 5 ++++- workerserver/workerserver.go | 4 +++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/workerclient/workerclient.go b/workerclient/workerclient.go index 7459514..ac7682d 100644 --- a/workerclient/workerclient.go +++ b/workerclient/workerclient.go @@ -46,7 +46,7 @@ func startSingleConnection(connectionId int) { } for { - requestMessage, err := bufio.NewReader(conn).ReadString('\n') + encodedRequestMessage, err := bufio.NewReader(conn).ReadString('\n') if err != nil { log.Printf("(%d) Error receiving request from load balancer server : %s\n", connectionId, err) _ = conn.Close() @@ -54,6 +54,9 @@ func startSingleConnection(connectionId int) { break } + requestMessageBytes, _ := b64.StdEncoding.DecodeString(strings.TrimSpace(encodedRequestMessage)) + requestMessage := string(requestMessageBytes) + log.Printf("(%d) Received new request message : %s", connectionId, requestMessage) // TODO: Move prefix to the ENV diff --git a/workerserver/workerserver.go b/workerserver/workerserver.go index 248508b..b7f0c6f 100644 --- a/workerserver/workerserver.go +++ b/workerserver/workerserver.go @@ -61,7 +61,9 @@ func handleConnection(conn net.Conn) { for { request := <-workerDescription.Req - _, err := conn.Write([]byte("URL " + request.Url + "\n")) + requestBase64 := b64.StdEncoding.EncodeToString([]byte("URL " + request.Url)) + + _, err := conn.Write([]byte(requestBase64 + "\n")) if err != nil { log.Println("Cannot send to " + clientAddr) conn.Close()