health.go

package health

import (
	"context"
	"net/http"

	pbs "github.com/hashicorp/boundary/internal/gen/controller/ops/services"
	"github.com/hashicorp/boundary/internal/servers/controller/handlers"
)

type Service struct {
	pbs.UnimplementedHealthServiceServer

	replyWithServiceUnavailable bool
}

func NewService() *Service {
	s := Service{}
	return &s
}

func (s *Service) GetHealth(ctx context.Context, req *pbs.GetHealthRequest) (*pbs.GetHealthResponse, error) {
	if s.replyWithServiceUnavailable {
		err := handlers.SetStatusCode(ctx, http.StatusServiceUnavailable)
		if err != nil {
			return nil, err
		}
		return &pbs.GetHealthResponse{}, nil
	}

	return &pbs.GetHealthResponse{}, nil
}

// StartServiceUnavailableReplies gets returned to the caller of NewService.
// When invoked, we start responding to any health queries with a 503.
func (s *Service) StartServiceUnavailableReplies() {
	s.replyWithServiceUnavailable = true
}