From 21f24899a25c139140f047622cd9016effd5a7fc Mon Sep 17 00:00:00 2001
From: Daniel Legt <contact@danlegt.com>
Date: Sat, 20 Jan 2024 02:09:10 +0200
Subject: [PATCH] * Custom listen

---
 lib/config/config.go | 7 +++++++
 main.go              | 3 ++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/lib/config/config.go b/lib/config/config.go
index 79009ff..bb5face 100644
--- a/lib/config/config.go
+++ b/lib/config/config.go
@@ -12,6 +12,7 @@ type DHConfig struct {
 	DiskFetchFrequency  int `json:"diskFetchFrequency"`
 	MemoryDumpFrequency int `json:"memoryDumpFrequency"`
 	MaxHistoryAge       int `json:"maxHistoryAge"`
+	Listen              string
 }
 
 func GetConfiguration() DHConfig {
@@ -24,6 +25,8 @@ func GetConfiguration() DHConfig {
 		DiskFetchFrequency:  5,       // default value
 		MemoryDumpFrequency: 60,      // default value
 		MaxHistoryAge:       2592000, // default value
+
+		Listen: ":8080",
 	}
 
 	if val, exists := os.LookupEnv("DISK_FETCH_FREQUENCY"); exists {
@@ -44,5 +47,9 @@ func GetConfiguration() DHConfig {
 		}
 	}
 
+	if val, exists := os.LookupEnv("LISTEN"); exists {
+		config.Listen = val
+	}
+
 	return config
 }
diff --git a/main.go b/main.go
index 3079f48..40fdfcd 100644
--- a/main.go
+++ b/main.go
@@ -9,6 +9,7 @@ import (
 	"syscall"
 	"time"
 
+	"tea.chunkbyte.com/kato/drive-health/lib/config"
 	"tea.chunkbyte.com/kato/drive-health/lib/svc"
 	"tea.chunkbyte.com/kato/drive-health/lib/web"
 )
@@ -20,7 +21,7 @@ func main() {
 	router := web.SetupRouter()
 
 	srv := &http.Server{
-		Addr:    ":8080",
+		Addr:    config.GetConfiguration().Listen,
 		Handler: router,
 	}