From ac352ca6b6dbe6583485d7491484240f5ee5710d Mon Sep 17 00:00:00 2001
From: Pulkit Kathuria <kevincobain2000@gmail.com>
Date: Sun, 19 May 2024 22:11:35 +0900
Subject: [PATCH] (feat) adds mutex lock for queues

---
 frontend/src/components/Extractor.astro | 30 ++++++++++++++++---------
 pkg/extract_handler.go                  |  4 ++++
 2 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/frontend/src/components/Extractor.astro b/frontend/src/components/Extractor.astro
index 3be2329..25c5642 100644
--- a/frontend/src/components/Extractor.astro
+++ b/frontend/src/components/Extractor.astro
@@ -14,7 +14,7 @@ const baseURL = import.meta.env.PUBLIC_BASE_URL;
   <p class="text-center text-slate-400 font-semibold select-none">
     <span class="text-slate-500"
       >Free <b class="text-green-500">online</b> tool to extract emails from any
-      URL.
+      URL
     </span>
     <br />
     <br />
@@ -99,6 +99,7 @@ const baseURL = import.meta.env.PUBLIC_BASE_URL;
             class="block font-mono w-full p-3 ps-4 text-sm text-slate-300 hover:text-slate-100 border border-gray-500 hover:border-gray-100 rounded-lg bg-gray-900 focus:outline-none"
             placeholder="-1"
           />
+          <p class="p-1 text-sm text-slate-500">-1 to crawl all links. 0 to only current link. 1 onwards to crawl links forward depth</p>
         </div>
         <div>
           <label
@@ -150,22 +151,29 @@ const baseURL = import.meta.env.PUBLIC_BASE_URL;
   <div class="inline-flex items-center justify-center w-full select-none">
     <hr class="w-1/2 h-px my-8 border-0 bg-gray-700" />
     <span
-      class="absolute px-8 py-1 text-sm font-bold -translate-x-1/2 left-1/2 text-green-300 bg-slate-900 rounded border-2 border-gray-700"
+      class="absolute px-8 py-2 text-sm font-bold -translate-x-1/2 left-1/2 text-green-300 bg-slate-900 rounded border-2 border-gray-700"
     >
-      Status
+      Crawling Status
       <span x-text="loading.url" class="text-blue-200 font-medium font-mono"
       ></span>
     </span>
   </div>
 
-  <div class="text-center text-sm pt-5 select-none">
-    <span class="font-semibold"
-      >Crawls <span class="text-slate-500">#</span></span
-    ><span x-text="loading.counterCrawls" class="text-green-300 font-bold"
-    ></span>
-    <span class="font-semibold pl-5"
-      >Hits <span class="text-slate-500">#</span></span
-    ><span x-text="loading.counterHits" class="text-green-500 font-bold"></span>
+  <div class="text-center text-sm pt-5 select-none flex items-center justify-center">
+    <div class="flex items-center space-x-8">
+        <div class="flex flex-col items-center">
+            <div class="flex items-center justify-center w-24 h-24 bg-blue-500 rounded-full">
+                <span class="text-2xl font-bold text-white" x-text="loading.counterCrawls">75</span>
+            </div>
+            <span class="mt-2 text-lg font-bold text-slate-400">Crawls</span>
+        </div>
+        <div class="flex flex-col items-center">
+            <div class="flex items-center justify-center w-24 h-24 bg-green-500 rounded-full">
+                <span class="text-2xl font-bold text-white" x-text="loading.counterHits">120</span>
+            </div>
+            <span class="mt-2 text-lg font-bold text-slate-400">Hits</span>
+        </div>
+    </div>
   </div>
 
   <template x-if="result.data.length > 0">
diff --git a/pkg/extract_handler.go b/pkg/extract_handler.go
index 88676a2..832a60e 100644
--- a/pkg/extract_handler.go
+++ b/pkg/extract_handler.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"net/http"
 	"net/url"
+	"sync"
 
 	"github.com/labstack/echo/v4"
 	"github.com/mcuadros/go-defaults"
@@ -12,6 +13,7 @@ import (
 type ExtractHandler struct {
 	Extractor *Extract
 	queue     int
+	mu        sync.Mutex
 }
 
 const limitQueue = 20
@@ -32,6 +34,8 @@ type ExtractorRequest struct {
 }
 
 func (h *ExtractHandler) Get(c echo.Context) error {
+	h.mu.Lock()
+	defer h.mu.Unlock()
 	h.queue++
 	if h.queue > limitQueue {
 		return echo.NewHTTPError(http.StatusUnprocessableEntity, "queue is full, please try again later")