From d98178f6f0d8d8e579183a3dd5accb5a889ddcc4 Mon Sep 17 00:00:00 2001
From: Victoria Erokhina <viktoria.erokhina@gmail.com>
Date: Thu, 29 Aug 2024 17:57:49 +0000
Subject: [PATCH] Support connections via unix sockets

---
 cmd/api/main.go      |  4 ++--
 pkg/api/server.go    | 48 +++++++++++++++++++++++++++++++++++---------
 pkg/config/config.go |  3 ++-
 3 files changed, 43 insertions(+), 12 deletions(-)

diff --git a/cmd/api/main.go b/cmd/api/main.go
index 1ceff7ce..ee1914c0 100644
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@@ -90,7 +90,7 @@ func main() {
 		storageBlockCh,
 	})
 
-	server, err := api.NewServer(log, h, fmt.Sprintf(":%d", cfg.API.Port),
+	server, err := api.NewServer(log, h,
 		api.WithTransactionSource(source),
 		api.WithBlockHeadersSource(source),
 		api.WithTraceSource(tracer),
@@ -110,5 +110,5 @@ func main() {
 	}()
 
 	log.Warn("start server", zap.Int("port", cfg.API.Port))
-	server.Run()
+	server.Run(fmt.Sprintf(":%d", cfg.API.Port), cfg.API.UnixSockets)
 }
diff --git a/pkg/api/server.go b/pkg/api/server.go
index 3c65aa14..6d10750a 100644
--- a/pkg/api/server.go
+++ b/pkg/api/server.go
@@ -2,7 +2,10 @@ package api
 
 import (
 	"errors"
+	"fmt"
+	"net"
 	"net/http"
+	"os"
 
 	"github.com/tonkeeper/tongo/config"
 	"go.uber.org/zap"
@@ -93,7 +96,7 @@ func WithMemPool(memPool sources.MemPoolSource) ServerOption {
 	}
 }
 
-func NewServer(log *zap.Logger, handler *Handler, address string, opts ...ServerOption) (*Server, error) {
+func NewServer(log *zap.Logger, handler *Handler, opts ...ServerOption) (*Server, error) {
 	options := &ServerOptions{}
 	for _, o := range opts {
 		o(options)
@@ -137,7 +140,6 @@ func NewServer(log *zap.Logger, handler *Handler, address string, opts ...Server
 		mux:              mux,
 		asyncMiddlewares: asyncMiddlewares,
 		httpServer: &http.Server{
-			Addr:    address,
 			Handler: mux,
 		},
 	}
@@ -161,11 +163,39 @@ func (s *Server) RegisterAsyncHandler(pattern string, handler AsyncHandler, conn
 	s.mux.Handle(pattern, wrapAsync(connectionType, allowTokenInQuery, chainMiddlewares(handler, s.asyncMiddlewares...)))
 }
 
-func (s *Server) Run() {
-	err := s.httpServer.ListenAndServe()
-	if errors.Is(err, http.ErrServerClosed) {
-		s.logger.Warn("opentonapi quit")
-		return
-	}
-	s.logger.Fatal("ListedAndServe() failed", zap.Error(err))
+func (s *Server) Run(address string, unixSockets []string) {
+	go func() {
+		tcpListener, err := net.Listen("tcp", address)
+		if err != nil {
+			s.logger.Fatal("Failed to listen on tcp address", zap.Error(err))
+		}
+		err = s.httpServer.Serve(tcpListener)
+
+		if errors.Is(err, http.ErrServerClosed) {
+			s.logger.Warn("opentonapi quit")
+			return
+		}
+		s.logger.Fatal("ListenAndServe() failed", zap.Error(err))
+	}()
+
+	for _, socketPath := range unixSockets {
+		go func(socketPath string) {
+			if _, err := os.Stat(socketPath); err == nil {
+				os.Remove(socketPath)
+			}
+
+			unixListener, err := net.Listen("unix", socketPath)
+			if err != nil {
+				s.logger.Fatal(fmt.Sprintf("Failed to listen on Unix socket %v", socketPath), zap.Error(err))
+			}
+
+			err = s.httpServer.Serve(unixListener)
+			if errors.Is(err, http.ErrServerClosed) {
+				s.logger.Warn("opentonapi quit")
+				return
+			}
+			s.logger.Fatal(fmt.Sprintf("ListenAndServe() failed for %v", socketPath), zap.Error(err))
+		}(socketPath)
+	}
+	<-make(chan struct{})
 }
diff --git a/pkg/config/config.go b/pkg/config/config.go
index d10c735f..30264d93 100644
--- a/pkg/config/config.go
+++ b/pkg/config/config.go
@@ -13,7 +13,8 @@ import (
 
 type Config struct {
 	API struct {
-		Port int `env:"PORT" envDefault:"8081"`
+		Port        int      `env:"PORT" envDefault:"8081"`
+		UnixSockets []string `env:"UNIX_SOCKETS" envSeparator:","`
 	}
 	App struct {
 		LogLevel           string              `env:"LOG_LEVEL" envDefault:"INFO"`