mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-11-04 18:26:25 +00:00
59 lines
1.7 KiB
Go
59 lines
1.7 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var cmd = &cobra.Command{
|
|
Short: "A gRPC micro service",
|
|
SilenceUsage: true,
|
|
FParseErrWhitelist: cobra.FParseErrWhitelist{UnknownFlags: true},
|
|
}
|
|
|
|
const (
|
|
serverAddress = "server_address"
|
|
|
|
secure = "secure"
|
|
reflection = "reflection"
|
|
|
|
caCert = "ca_cert"
|
|
serverCert = "server_cert"
|
|
serverKey = "server_key"
|
|
)
|
|
|
|
func init() {
|
|
viper.AutomaticEnv()
|
|
// server_address
|
|
cmd.Flags().String(serverAddress, "0.0.0.0:0", "Bind address for the server. 127.0.0.1:9090 [$SERVER_ADDRESS]")
|
|
viper.BindPFlag(serverAddress, cmd.Flags().Lookup(serverAddress))
|
|
|
|
// secure
|
|
cmd.Flags().Bool(secure, true, "Generate self signed certificate if none provided [$SECURE]")
|
|
viper.BindPFlag(secure, cmd.Flags().Lookup(secure))
|
|
|
|
// reflection
|
|
cmd.Flags().Bool(reflection, false, "Enable gRPC reflection server [$REFLECT]")
|
|
viper.BindPFlag(reflection, cmd.Flags().Lookup(reflection))
|
|
|
|
// ca_cert
|
|
cmd.Flags().String(caCert, "", "Path to Root CA certificate [$CA_CERT]")
|
|
viper.BindPFlag(caCert, cmd.Flags().Lookup(caCert))
|
|
// server_cert
|
|
cmd.Flags().String(serverCert, "", "Path to Server certificate [$SERVER_CERT]")
|
|
viper.BindPFlag(serverCert, cmd.Flags().Lookup(serverCert))
|
|
// server_key
|
|
cmd.Flags().String(serverKey, "", "Path to Server key [$SERVER_KEY]")
|
|
viper.BindPFlag(serverKey, cmd.Flags().Lookup(serverKey))
|
|
}
|
|
|
|
func parseFlags(o *options) *options {
|
|
o.address = viper.GetString(serverAddress)
|
|
o.secure = viper.GetBool(secure)
|
|
o.reflection = viper.GetBool(reflection)
|
|
o.caCert = viper.GetString(caCert)
|
|
o.cert = viper.GetString(serverCert)
|
|
o.key = viper.GetString(serverKey)
|
|
return o
|
|
}
|