signoz/pkg/factory/servicetest/http.go

package servicetest

import (
	"context"
	"net"
	"net/http"

	"go.signoz.io/signoz/pkg/factory"
)

var _ factory.Service = (*httpService)(nil)

type httpService struct {
	Listener net.Listener
	Server   *http.Server
	name     string
}

func NewHttpService(name string) (*httpService, error) {
	return &httpService{
		name:   name,
		Server: &http.Server{},
	}, nil
}

func (service *httpService) Name() factory.Name {
	return factory.MustNewName(service.name)
}

func (service *httpService) Start(ctx context.Context) error {
	listener, err := net.Listen("tcp", "localhost:0")
	if err != nil {
		return err
	}
	service.Listener = listener

	if err := service.Server.Serve(service.Listener); err != nil {
		if err != http.ErrServerClosed {
			return err
		}
	}
	return nil
}

func (service *httpService) Stop(ctx context.Context) error {
	if err := service.Server.Shutdown(ctx); err != nil {
		return err
	}

	return nil
}