mirror of
https://github.com/foomo/contentserver.git
synced 2025-10-16 12:25:44 +00:00
58 lines
2.2 KiB
Go
58 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func addAddressFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().String("address", "localhost:8080", "Address to bind to (host:port)")
|
|
_ = v.BindPFlag("address", cmd.Flags().Lookup("address"))
|
|
}
|
|
|
|
func addBasePathFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().String("base-path", "/contentserver", "Base path to export the webserver on")
|
|
_ = v.BindPFlag("base_path", cmd.Flags().Lookup("base_path"))
|
|
}
|
|
|
|
func addPollFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Bool("poll", false, "If true, the address arg will be used to periodically poll the content url")
|
|
_ = v.BindPFlag("poll", cmd.Flags().Lookup("poll"))
|
|
}
|
|
|
|
func addHistoryDirFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().String("history-dir", "/var/lib/contentserver", "Where to put my data")
|
|
_ = v.BindPFlag("history.dir", cmd.Flags().Lookup("history-dir"))
|
|
_ = v.BindEnv("history.dir", "CONTENT_SERVER_HISTORY_DIR")
|
|
}
|
|
|
|
func addHistoryLimitFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Int("history-limit", 2, "Number of history records to keep")
|
|
_ = v.BindPFlag("history.limit", cmd.Flags().Lookup("history-limit"))
|
|
}
|
|
|
|
func addGracefulTimeoutFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Duration("graceful-timeout", 0, "Timeout duration for graceful shutdown")
|
|
_ = v.BindPFlag("graceful.timeout", cmd.Flags().Lookup("graceful-timeout"))
|
|
}
|
|
|
|
func addShutdownTimeoutFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Duration("shutdown-timeout", 0, "Timeout duration for shutdown")
|
|
_ = v.BindPFlag("shutdown.timeout", cmd.Flags().Lookup("shutdown-timeout"))
|
|
}
|
|
|
|
func addOtelEnabledFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Bool("otel-enabled", false, "Enable otel service")
|
|
_ = v.BindPFlag("otel.enabled", cmd.Flags().Lookup("otel-enabled"))
|
|
}
|
|
|
|
func addHealthzEnabledFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Bool("healthz-enabled", false, "Enable healthz service")
|
|
_ = v.BindPFlag("healthz.enabled", cmd.Flags().Lookup("healthz-enabled"))
|
|
}
|
|
|
|
func addPrometheusEnabledFlag(cmd *cobra.Command, v *viper.Viper) {
|
|
cmd.Flags().Bool("prometheus-enabled", false, "Enable prometheus service")
|
|
_ = v.BindPFlag("prometheus.enabled", cmd.Flags().Lookup("prometheus-enabled"))
|
|
}
|