mirror of
https://github.com/teamhanko/hanko.git
synced 2025-10-30 08:07:28 +08:00
e963f26239
Config initialization is done in the root command but some commands do not require a config. If these commands are executed locally in the checked out repo they work fine, but they will fail if they are executed e.g. using the Docker image. I would have to provide a config file path to the container and mount the referenced file into the container even though it is not needed for the command. This commit changes this by making those commands that require a config load it themselves. It also pulls down the config flag to the individual commands that require it instead of using a persistent flag on the root level.
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package isready
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"github.com/teamhanko/hanko/backend/config"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
)
|
|
|
|
func NewIsReadyCommand() *cobra.Command {
|
|
var (
|
|
configFile string
|
|
)
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "isready",
|
|
Args: cobra.OnlyValidArgs,
|
|
ValidArgs: []string{"admin", "public"},
|
|
Short: "Health check a service",
|
|
Long: `Checks if the specified service is healthy. Possible values are "admin" and "public".
|
|
Uses the "/health/ready" endpoint to check if the service is ready to serve requests.
|
|
`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if len(args) != 1 {
|
|
log.Fatalf("Please specify a service to check")
|
|
}
|
|
service := args[0]
|
|
|
|
cfg, err := config.Load(&configFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
address := ""
|
|
if service == "admin" {
|
|
address = cfg.Server.Admin.Address
|
|
}
|
|
if service == "public" {
|
|
address = cfg.Server.Public.Address
|
|
}
|
|
host, port, err := net.SplitHostPort(address)
|
|
if err != nil {
|
|
log.Fatalf("Could not parse address %s", address)
|
|
}
|
|
if host == "" {
|
|
host = "localhost"
|
|
}
|
|
requestUrl := fmt.Sprintf("http://%s:%s/health/ready", host, port)
|
|
res, err := http.Get(requestUrl)
|
|
if err != nil {
|
|
log.Fatalf("Service %s is not ready", service)
|
|
} else {
|
|
if res.StatusCode != 200 {
|
|
log.Fatalf("Service %s is not ready", service)
|
|
} else {
|
|
log.Println(fmt.Sprintf("Service %s is ready", service))
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
cmd.Flags().StringVar(&configFile, "config", config.DefaultConfigFilePath, "config file")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func RegisterCommands(parent *cobra.Command) {
|
|
cmd := NewIsReadyCommand()
|
|
parent.AddCommand(cmd)
|
|
}
|