From e18bd075828c4902afc18d8c8633b59c9493794f Mon Sep 17 00:00:00 2001
From: Pierre-Olivier Mercier <nemunaire@nemunai.re>
Date: Thu, 19 Nov 2020 01:38:53 +0100
Subject: [PATCH] Get chunk size from env

---
 main.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index 4750ba9..70dd729 100644
--- a/main.go
+++ b/main.go
@@ -9,11 +9,13 @@ import (
 	"log"
 	"math/rand"
 	"net/http"
+	"os"
+	"strconv"
 	"strings"
 	"time"
 )
 
-var chunkSize uint = 2
+var chunkSize = 2
 var currentChunk string
 
 func init() {
@@ -120,9 +122,14 @@ func ServeScores(w http.ResponseWriter, r *http.Request) {
 }
 
 func main() {
+	if v, exists := os.LookupEnv("CHUNKSIZE"); exists {
+		if cs, err := strconv.Atoi(v); err == nil {
+			chunkSize = cs
+		}
+	}
 	var bind = flag.String("bind", "0.0.0.0:8081", "Bind port/socket")
 	var dsn = flag.String("dsn", DSNGenerator(), "DSN to connect to the MySQL server")
-	flag.UintVar(&chunkSize, "chunkSize", chunkSize, "Taille du chunk à trouver")
+	flag.IntVar(&chunkSize, "chunkSize", chunkSize, "Taille du chunk à trouver")
 	flag.Parse()
 	newChunk()