set routing

Signed-off-by: Vivek Kumar Sahu <vivekkumarsahu650@gmail.com>
viveksahu26 · Feb 9, 2024 · 146a148 · 146a148
1 parent a12d62b
commit 146a148
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/cmd/url_shortner/main.go b/cmd/url_shortner/main.go
@@ -36,19 +36,17 @@ func main() {
 	case "server":
 		serverCmd.Parse(os.Args[2:])
 		fmt.Printf("Starting server on port %s...\n", *port)
-		http.HandleFunc("/health", short.HealthCheckUp)
-		http.HandleFunc("/short-url", short.HandleShortURL)
-		http.HandleFunc("/long-url", short.HandleLongURL)
 		// setupRoutes() // Function to set up web routes
+		setupRoutes()
 		log.Fatal(http.ListenAndServe(":"+*port, nil))
 	default:
 		fmt.Println("expected 'short' or 'server' subcommands")
 		os.Exit(1)
 	}
 }
 
-// func setupRoutes() {
-// 	http.HandleFunc("/health", src.HealthCheckUp)
-// 	http.HandleFunc("/short-url", src.HandleShortURL)
-// 	http.HandleFunc("/long-url", src.HandleLongURL)
-// }
+func setupRoutes() {
+	http.HandleFunc("/health", short.HealthCheckUp)
+	http.HandleFunc("/short-url", short.HandleShortURL)
+	http.HandleFunc("/long-url", short.HandleLongURL)
+}