Refactor code for improved readability and consistency; remove unnecessary whitespace

Author: itsfuad

balancer/balancer.go

@@ -15,18 +15,18 @@ import (
 )
 
 type LoadBalancer struct {
-	Servers     []*sv.Server
-	mu          sync.RWMutex
-	Logger      *log.Logger
-	wg          sync.WaitGroup
-	shutdown    bool
-	metrics     *Metrics
-	maxRetries  int
+	Servers    []*sv.Server
+	mu         sync.RWMutex
+	Logger     *log.Logger
+	wg         sync.WaitGroup
+	shutdown   bool
+	metrics    *Metrics
+	maxRetries int
 }
 
 type Metrics struct {
-	TotalRequests uint64
-	FailedRequests uint64
+	TotalRequests     uint64
+	FailedRequests    uint64
 	ActiveConnections int64
 }
 
@@ -35,28 +35,28 @@ func NewLoadBalancer(logger *log.Logger) *LoadBalancer {
 		logger = log.New(io.Discard, "", log.LstdFlags)
 	}
 	return &LoadBalancer{
-		Logger: logger,
-		metrics: &Metrics{},
+		Logger:     logger,
+		metrics:    &Metrics{},
 		maxRetries: 3,
 	}
 }
 
 func (lb *LoadBalancer) AddServer(url string) error {
 	lb.mu.Lock()
 	defer lb.mu.Unlock()
-	
+
 	// Validate server URL
 	if url == "" {
 		return errors.New("server URL cannot be empty")
 	}
-	
+
 	// Check for duplicate servers
 	for _, server := range lb.Servers {
 		if server.URL == url {
 			return fmt.Errorf("server %s already exists", url)
 		}
 	}
-	
+
 	server := sv.NewServer(url, lb.Logger)
 	lb.Servers = append(lb.Servers, server)
 	lb.Logger.Println(utils.Colorize("Added server "+url+" to the load balancer", utils.GREEN))
@@ -66,7 +66,7 @@ func (lb *LoadBalancer) AddServer(url string) error {
 func (lb *LoadBalancer) RemoveServer(url string) error {
 	lb.mu.Lock()
 	defer lb.mu.Unlock()
-	
+
 	for i, server := range lb.Servers {
 		if server.URL == url {
 			// Wait for active connections to finish
@@ -75,21 +75,21 @@ func (lb *LoadBalancer) RemoveServer(url string) error {
 				time.Sleep(100 * time.Millisecond)
 				lb.mu.Lock()
 			}
-			
+
 			// Remove server
 			lb.Servers = append(lb.Servers[:i], lb.Servers[i+1:]...)
 			lb.Logger.Println(utils.Colorize("Removed server "+url, utils.YELLOW))
 			return nil
 		}
 	}
-	
+
 	return fmt.Errorf("server %s not found", url)
 }
 
 func (lb *LoadBalancer) GetMetrics() *Metrics {
 	return &Metrics{
-		TotalRequests: atomic.LoadUint64(&lb.metrics.TotalRequests),
-		FailedRequests: atomic.LoadUint64(&lb.metrics.FailedRequests),
+		TotalRequests:     atomic.LoadUint64(&lb.metrics.TotalRequests),
+		FailedRequests:    atomic.LoadUint64(&lb.metrics.FailedRequests),
 		ActiveConnections: atomic.LoadInt64(&lb.metrics.ActiveConnections),
 	}
 }

server/server.go

@@ -57,7 +57,7 @@ func (s *Server) HandleRequest(w http.ResponseWriter, r *http.Request) error {
 	if err != nil {
 		return fmt.Errorf("failed to read request body: %v", err)
 	}
-	
+
 	r.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
 
 	req, err := http.NewRequest(r.Method, s.URL+r.RequestURI, bytes.NewBuffer(bodyBytes))

server/server_test.go

@@ -87,4 +87,4 @@ func TestCheckHealth(t *testing.T) {
 	if !server.Healthy {
 		t.Errorf("expected server to be healthy")
 	}
-}
+}

utils/http.go

@@ -4,21 +4,21 @@ import "net/http"
 
 // CopyHeaders copies HTTP headers from src to dst
 func CopyHeaders(dst, src http.Header) {
-    for k, vv := range src {
-        for _, v := range vv {
-            dst.Add(k, v)
-        }
-    }
+	for k, vv := range src {
+		for _, v := range vv {
+			dst.Add(k, v)
+		}
+	}
 }
 
 // GetClientIP extracts the client IP from the request
 func GetClientIP(r *http.Request) string {
-    ip := r.Header.Get("X-Forwarded-For")
-    if ip == "" {
-        ip = r.Header.Get("X-Real-IP")
-    }
-    if ip == "" {
-        ip = r.RemoteAddr
-    }
-    return ip
-}
+	ip := r.Header.Get("X-Forwarded-For")
+	if ip == "" {
+		ip = r.Header.Get("X-Real-IP")
+	}
+	if ip == "" {
+		ip = r.RemoteAddr
+	}
+	return ip
+}