40 lines
878 B
Go
40 lines
878 B
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/huoxue1/min-report/config"
|
|
"github.com/huoxue1/min-report/module"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
rootCmd = cobra.Command{
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if err := server().Run(fmt.Sprintf("%s:%d", viper.GetString("http.address"), viper.GetInt("http.port"))); err != nil {
|
|
return
|
|
}
|
|
},
|
|
PreRun: func(cmd *cobra.Command, args []string) {
|
|
fmt.Println(fmt.Sprintf("build commit: %s, build version: %s, build date: %s", Commit, Version, Date))
|
|
config.InitConfig(configFile)
|
|
module.InitDb(viper.GetString("db.driver"), viper.GetString("db.url"))
|
|
},
|
|
}
|
|
)
|
|
|
|
var (
|
|
configFile string
|
|
)
|
|
|
|
func Run() {
|
|
if err := rootCmd.Execute(); err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.PersistentFlags().StringVarP(&configFile, "config", "c", "./config.yaml", "config file")
|
|
}
|