Added health chack up function

Signed-off-by: viveksahu26 <vivekkumarsahu650@gmail.com>

main.go

@@ -3,12 +3,14 @@ package main
 import (
 	"fmt"
 	"net/http"
-	"time"
 
 	"github.com/viveksahu26/url_shortner/src"
 )
 
 func handleShortURL(writer http.ResponseWriter, req *http.Request) {
+	if req.Method != "GET" {
+		writer.WriteHeader(http.StatusMethodNotAllowed)
+	}
 	// get original URL from GET method by quering
 	originalURL := req.URL.Query().Get("longURL")
 	fmt.Println("originalURL: ", originalURL)
@@ -20,18 +22,27 @@ func handleShortURL(writer http.ResponseWriter, req *http.Request) {
 	// save short and long URL to file
 	src.SaveInFile(shortURL, originalURL)
 
-	host := "www.simplifyurl.com"
+	host := req.Host
 
 	// build Response
 	resp := src.BuildURLWithResponse(host, shortURL, originalURL)
-	fmt.Println("response: ", resp)
 
 	err := src.RespondWithJSON(writer, 200, resp)
 	if err != nil {
 		writer.Write([]byte("Failed to respond with JSON"))
 	}
 }
 
+func healthCheckUp(w http.ResponseWriter, r *http.Request) {
+	method := r.Method
+	if method == "GET" {
+		w.WriteHeader(http.StatusOK)
+		w.Write([]byte("Health of Server is UP & Running..."))
+	} else {
+		w.WriteHeader(http.StatusMethodNotAllowed)
+	}
+}
+
 const addr = "localhost:8080"
 
 func main() {
@@ -40,14 +51,15 @@ func main() {
 	// multiplexer: It provides seperate server interface for each request.
 	serveMux := http.NewServeMux()
 	srv := http.Server{
-		Handler:      serveMux,
-		Addr:         addr,
-		WriteTimeout: 30 * time.Second,
-		ReadTimeout:  30 * time.Second,
+		Handler: serveMux,
+		Addr:    addr,
 	}
 
 	// handleShortUrl function mapped to /enterLongURL
-	serveMux.HandleFunc("/enterLongURL", handleShortURL)
+	serveMux.HandleFunc("/sort-url", handleShortURL)
+
+	// healthCheckUp function mapped to /health
+	http.HandleFunc("/health", healthCheckUp)
 
 	// Server Listing on "localhost:8080"
 	srv.ListenAndServe()

src/saveInfile.go

@@ -23,8 +23,7 @@ func SaveInFile(shortURL string, longURL string) {
 
 	} else {
 		fmt.Printf("File %s already exist.", fileName)
-		_, msg, fileContainLongURL := IsLongURLPresentInFile(fileName, longURL)
-		fmt.Println("\n", msg)
+		_, _, fileContainLongURL := IsLongURLPresentInFile(fileName, longURL)
 
 		if !fileContainLongURL {
 			file, err := os.OpenFile(fileName, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0o644)

src/shortURLGenerator.go

@@ -19,7 +19,6 @@ func GenerateShortURL(longURL string) string {
 
 	if isFilePresent {
 		shortAndLongURLKeyValuePair, _, fileContainLongURL := IsLongURLPresentInFile(fileName, longURL)
-
 		if fileContainLongURL {
 			// then retrieve ShortURL from there.
 			if shorturl, ok := shortAndLongURLKeyValuePair[longURL]; ok {