feat Read Configuration From Ini File

Author: rosercode

README.md

@@ -0,0 +1 @@
+- 

config.ini

@@ -0,0 +1,7 @@
+[server]
+address = 0.0.0.0
+port = 18080
+
+[proxy]
+test1.example.com = http://localhost:8001
+test2.example.com = http://localhost:9000

go.mod

@@ -0,0 +1,5 @@
+module github.com/rosercode/GoHTTPProxy
+
+go 1.19
+
+require github.com/go-ini/ini v1.67.0 // indirect

main.go

@@ -6,21 +6,20 @@ import (
 	"net/http/httputil"
 	"net/url"
 	"strings"
+
+	"github.com/go-ini/ini"
 )
 
 // Define reverse proxy mapping table
-var proxyMap = map[string]*url.URL{
-	"test1.example.com": parseURL("http://localhost:8000"), // Forward test1.example.com to local HTTP server (port 8000)
-	"test2.example.com": parseURL("http://localhost:9000"), // Forward test2.example.com to local HTTP server (port 9000)
-}
+var proxyMap = make(map[string]*url.URL)
 
 // Parse URL
-func parseURL(rawurl string) *url.URL {
+func parseURL(rawurl string) (*url.URL, error) {
 	parsedURL, err := url.Parse(rawurl)
 	if err != nil {
-		log.Fatalf("Error parsing URL: %s", err)
+		return nil, err
 	}
-	return parsedURL
+	return parsedURL, nil
 }
 
 // Custom reverse proxy handler
@@ -31,6 +30,11 @@ func (rph *reverseProxyHandler) ServeHTTP(w http.ResponseWriter, req *http.Reque
 	hostParts := strings.Split(req.Host, ":")
 	domain := hostParts[0]
 
+	// Print request log
+	log.Println("Received request for domain:", domain)
+	log.Println("Request URL:", req.URL.String())
+	log.Println("Request Method:", req.Method)
+
 	// Look up the corresponding target URL in the proxy mapping table
 	targetURL, ok := proxyMap[domain]
 	if !ok {
@@ -56,24 +60,52 @@ func (rph *reverseProxyHandler) ServeHTTP(w http.ResponseWriter, req *http.Reque
 }
 
 func main() {
-	// Print the address and port that the server is listening on
-	addr := ":18080"
-	log.Printf("Reverse proxy server listening on %s...", addr)
+	// Load Configuration File
+	cfg, err := ini.Load("config.ini")
+	if err != nil {
+		log.Fatalf("Failed to load configuration file: %v", err)
+		return
+	}
 
-	// Print the reverse proxy mapping table
+	// Parse Reverse Proxy Mapping Table from Configuration File
+	section := cfg.Section("proxy")
+	if section == nil {
+		log.Fatal("Proxy section not found in configuration file")
+		return
+	}
+
+	// Iterate through Configuration Entries and Parse URLs
 	log.Println("Reverse proxy mapping table:")
-	for domain, targetURL := range proxyMap {
-		log.Printf("%s => %s", domain, targetURL.String())
+	for _, key := range section.Keys() {
+		targetURL, err := parseURL(key.String())
+		// Print the reverse proxy mapping table
+		log.Printf("%s => %s", key.Name(), targetURL.String())
+		if err != nil {
+			log.Fatalf("Failed to parse URL for key %s: %v", key, err)
+		}
+		proxyMap[key.Name()] = targetURL
 	}
 
+	// Read Server Configuration
+	serverSection := cfg.Section("server")
+	if serverSection == nil {
+		log.Fatal("Server section not found in configuration file")
+		return
+	}
+
+	address := serverSection.Key("address").String()
+	port := serverSection.Key("port").String()
+
 	// Create an instance of the custom reverse proxy handler
 	handler := &reverseProxyHandler{}
 
 	// Register the reverse proxy handler
 	http.Handle("/", handler)
 
 	// Start the server
-	err := http.ListenAndServe(addr, nil)
+	addr := address + ":" + port
+	log.Printf("Reverse proxy server listening on %s...", addr)
+	err = http.ListenAndServe(addr, nil)
 	if err != nil {
 		log.Fatal(err)
 	}