zitadel/cmd/ready/ready.go

package ready

import (
	"crypto/tls"
	"net"
	"net/http"
	"os"
	"strconv"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"github.com/zitadel/logging"
)

func New() *cobra.Command {
	return &cobra.Command{
		Use:   "ready",
		Short: "Checks if zitadel is ready",
		Long:  "Checks if zitadel is ready",
		Run: func(cmd *cobra.Command, args []string) {
			config := MustNewConfig(viper.GetViper())
			if !ready(config) {
				os.Exit(1)
			}
		},
	}
}

func ready(config *Config) bool {
	scheme := "https"
	if !config.TLS.Enabled {
		scheme = "http"
	}
	// Checking the TLS cert is not in the scope of the readiness check
	httpClient := http.Client{Transport: &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}}}
	res, err := httpClient.Get(scheme + "://" + net.JoinHostPort("localhost", strconv.Itoa(int(config.Port))) + "/debug/ready")
	if err != nil {
		logging.WithError(err).Warn("ready check failed")
		return false
	}
	defer res.Body.Close()
	if res.StatusCode != 200 {
		logging.WithFields("status", res.StatusCode).Warn("ready check failed")
		return false
	}
	return true
}