use gin instead of standard http lib in main

llama-swap.go

@@ -3,9 +3,9 @@ package main
 import (
 	"flag"
 	"fmt"
-	"net/http"
 	"os"
 
+	"github.com/gin-gonic/gin"
 	"github.com/mostlygeek/llama-swap/proxy"
 )
 
@@ -22,12 +22,16 @@ func main() {
 		os.Exit(1)
 	}
 
-	proxyManager := proxy.New(config)
+	if mode := os.Getenv("GIN_MODE"); mode != "" {
-	http.HandleFunc("/", proxyManager.HandlerFunc)
+		gin.SetMode(mode)
+	} else {
+		gin.SetMode(gin.ReleaseMode)
+	}
 
+	proxyManager := proxy.New(config)
 	fmt.Println("llama-swap listening on " + *listenStr)
-	if err := http.ListenAndServe(*listenStr, nil); err != nil {
+	if err := proxyManager.Run(*listenStr); err != nil {
-		fmt.Printf("Error starting server: %v\n", err)
+		fmt.Printf("Server error: %v\n", err)
 		os.Exit(1)
 	}
 }

proxy/proxymanager.go

@@ -46,6 +46,10 @@ func New(config *Config) *ProxyManager {
 	return pm
 }
 
+func (pm *ProxyManager) Run(addr ...string) error {
+	return pm.ginEngine.Run(addr...)
+}
+
 func (pm *ProxyManager) HandlerFunc(w http.ResponseWriter, r *http.Request) {
 	pm.ginEngine.ServeHTTP(w, r)
 }