fission-src/pkg/utils/httpserver/server.go

package httpserver

import (
	"context"
	"fmt"
	"net/http"
	"strings"

	"go.uber.org/zap"

	"github.com/fission/fission/pkg/utils/manager"
)

func StartServer(ctx context.Context, log *zap.Logger, mgr manager.Interface, svc string, port string, handler http.Handler) {
	if !strings.Contains(port, ":") {
		port = fmt.Sprintf(":%s", port)
	}
	server := http.Server{
		Addr:    port,
		Handler: handler,
	}
	l := log.With(zap.String("service", svc), zap.String("addr", server.Addr))
	l.Info("starting server")
	mgr.Add(ctx, func(ctx context.Context) {
		if err := server.ListenAndServe(); err != nil {
			if err != http.ErrServerClosed {
				l.Error("server error", zap.Error(err))
			}
		}
	})
	<-ctx.Done()
	l.Info("shutting down server")
	if err := server.Shutdown(ctx); err != nil {
		if err != context.Canceled && err != context.DeadlineExceeded {
			l.Error("server shutdown error", zap.Error(err))
		}
	}
}