From 451c3477bc30d8aa1e797f2a436ab09625d588be Mon Sep 17 00:00:00 2001 From: Bilal Date: Fri, 8 May 2020 10:43:55 +0200 Subject: [PATCH] Use config values instead of hardcoded values --- workerserver/workerserver.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/workerserver/workerserver.go b/workerserver/workerserver.go index b7f0c6f..c1a7cb8 100644 --- a/workerserver/workerserver.go +++ b/workerserver/workerserver.go @@ -3,6 +3,7 @@ package workerserver import ( "bufio" b64 "encoding/base64" + c "gitlab.com/saburly/kiviscraplib/config" "gitlab.com/saburly/kiviscraplib/structures" "log" "math/rand" @@ -15,13 +16,12 @@ var requests chan structures.Request var workers chan structures.WorkerDescription func ServeWorkers(queue chan structures.Request, end chan<- string) { - rand.Seed(time.Now().Unix()) - workers = make(chan structures.WorkerDescription, 50) // TODO: move to env var + workers = make(chan structures.WorkerDescription, c.WorkerServerConfig.WorkersCount) requests = queue - listener, err := net.Listen("tcp", "127.0.0.1:1338") + listener, err := net.Listen("tcp", c.WorkerServerConfig.Address) if err != nil { log.Fatal("tcp server listener error:", err) end <- "tcp server" @@ -61,7 +61,7 @@ func handleConnection(conn net.Conn) { for { request := <-workerDescription.Req - requestBase64 := b64.StdEncoding.EncodeToString([]byte("URL " + request.Url)) + requestBase64 := b64.StdEncoding.EncodeToString([]byte(c.WorkerServerConfig.RequestMessagePrefix + request.Url)) _, err := conn.Write([]byte(requestBase64 + "\n")) if err != nil {