Skip to content

Commit

Permalink
cmd: port server flag
Browse files Browse the repository at this point in the history
  • Loading branch information
labasubagia committed Sep 13, 2023
1 parent 11889bc commit b166c96
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion cmd/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ func init() {

rootCmd.AddCommand(serverCmd)
serverCmd.Flags().StringP("database", "d", repository.DefaultRepoKey, fmt.Sprintf("select database in (%s)", dbOptStr))
serverCmd.Flags().IntP("port", "p", 5000, fmt.Sprint("server port "))
}

var serverCmd = &cobra.Command{
Expand All @@ -28,8 +29,18 @@ var serverCmd = &cobra.Command{
Long: "Run gin server restful API",
Run: func(cmd *cobra.Command, args []string) {

dbType := cmd.Flag("database").Value.String()
port, err := cmd.Flags().GetInt("port")
if err != nil {
log.Fatal("failed get port flag", err)
}
config.HTTPServerAddress = fmt.Sprintf("0.0.0.0:%d", port)

dbType, err := cmd.Flags().GetString("database")
if err != nil {
log.Fatal("failed get database flag", err)
}
dbType = strings.ToLower(dbType)

newRepo, exist := repository.RepoFnMap[dbType]
if !exist {
log.Fatal("invalid database", dbType)
Expand Down

0 comments on commit b166c96

Please sign in to comment.