cloudservices/common: General code for services

http.go

61 строка1 KB

Ivan

http & tls

26 июл 2026, 15:26

26 июл 2026, 15:260539f4e

package common

import (
	"context"
	"crypto/tls"
	"fmt"
	"net/http"
	"time"
)

type HTTPServer struct {
	raw *http.Server
}

func NewHTTPServer(addr string, tlsConfig *tls.Config, handler http.Handler) *HTTPServer {
	return &HTTPServer{
		raw: &http.Server{
			Addr:              addr,
			Handler:           handler,
			ReadTimeout:       30 * time.Second,
			ReadHeaderTimeout: 10 * time.Second,
			WriteTimeout:      30 * time.Second,
			IdleTimeout:       5 * time.Minute,
			MaxHeaderBytes:    1024 * 8,
		},
	}
}

func (server *HTTPServer) Raw() *http.Server {
	return server.raw
}

func (server *HTTPServer) Run(ctx context.Context) error {
	errCh := make(chan error, 1)
	go func() {
		defer close(errCh)

		var err error

		if server.raw.TLSConfig != nil {
			err = server.raw.ListenAndServeTLS("", "")
		} else {
			err = server.raw.ListenAndServe()
		}

		if err != nil && err != http.ErrServerClosed {
			errCh <- fmt.Errorf("http server error: %w", err)
		}
	}()

	select {

	case err := <-errCh:
		return err

	case <-ctx.Done():
		shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
		defer cancel()
		return server.raw.Shutdown(shutdownCtx)
	}
}