senaduka/old-kiviscraplib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Encode/decode request from load balancer server to the worker client

Browse Source
This commit is contained in:
Bilal
2020-05-08 06:32:04 +02:00
parent a77b46aad6
commit 7e12d2819e
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
workerclient/workerclient.go
View File

@@ -46,7 +46,7 @@ func startSingleConnection(connectionId int) {
} }
for { for {
requestMessage, err := bufio.NewReader(conn).ReadString('\n') encodedRequestMessage, err := bufio.NewReader(conn).ReadString('\n')
if err != nil { if err != nil {
log.Printf("(%d) Error receiving request from load balancer server : %s\n", connectionId, err) log.Printf("(%d) Error receiving request from load balancer server : %s\n", connectionId, err)
_ = conn.Close() _ = conn.Close()
@@ -54,6 +54,9 @@ func startSingleConnection(connectionId int) {
break break
} }
requestMessageBytes, _ := b64.StdEncoding.DecodeString(strings.TrimSpace(encodedRequestMessage))
requestMessage := string(requestMessageBytes)
log.Printf("(%d) Received new request message : %s", connectionId, requestMessage) log.Printf("(%d) Received new request message : %s", connectionId, requestMessage)
// TODO: Move prefix to the ENV // TODO: Move prefix to the ENV

4
workerserver/workerserver.go
View File

@@ -61,7 +61,9 @@ func handleConnection(conn net.Conn) {
for { for {
request := <-workerDescription.Req 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 { if err != nil {
log.Println("Cannot send to " + clientAddr) log.Println("Cannot send to " + clientAddr)
conn.Close() conn.Close()