change proxy close

multiversx · Nov 10, 2023 · 0afc5b8 · 0afc5b8
1 parent 44d0df6
commit 0afc5b8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/proxy/creator.go b/pkg/proxy/creator.go
@@ -1,6 +1,7 @@
 package proxy
 
 import (
+	"context"
 	"errors"
 	"math/big"
 	"net/http"
@@ -211,12 +212,16 @@ func CreateProxy(args ArgsProxy) (ProxyHandler, error) {
 		}
 	}()
 
-	proxyInstance.closableComponents.Add(proxyInstance.httpServer)
-
 	return proxyInstance, nil
 }
 
 // Close will close the proxy
 func (p *proxy) Close() {
 	p.closableComponents.Close()
+
+	shutdownContext, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	_ = p.httpServer.Shutdown(shutdownContext)
+	_ = p.httpServer.Close()
 }