update:test

Author: matteriot

config/config.go

@@ -19,26 +19,7 @@ func LoadConfig() (err error) {
 	}
 	_, err = os.Stat(DefaultConfigFilePath)
 	if err != nil {
-		log.Println("没有找到配置文件：", DefaultConfigFilePath)
-		log.Println("开始生成默认的空白配置文件")
-		ConfigMode.Common.BindAddr = DefaultBindAddr
-		ConfigMode.Common.KcpPort = DefaultKcpPort
-		ConfigMode.Common.TcpPort = DefaultTcpPort
-		ConfigMode.Common.TlsPort = DefaultTlsPort
-		ConfigMode.Common.GrpcPort = DefaultGrpcPort
-		ConfigMode.Common.HttpPort = DefaultHttpPort
-		ConfigMode.Common.HttpsPort = DefaultHttpsPort
-		ConfigMode.Common.UdpApiPort = DefaultUdpApiPort
-		ConfigMode.Common.KcpApiPort = DefaultKcpApiPort
-		ConfigMode.Security.LoginKey = DefaultLoginKey
-		//	生成配置文件模板
-		err = writeConfigFile(ConfigMode, DefaultConfigFilePath)
-		if err != nil {
-			log.Printf("写入默认的配置文件失败：%s\n", err.Error())
-			return
-		}
-		log.Println("配置文件写入成功,路径为：", DefaultConfigFilePath)
-		log.Println("你也可以修改上述配置文件后在运行")
+		InitConfigFile()
 	}
 	log.Println("使用配置文件：", DefaultConfigFilePath)
 	ConfigMode, err = GetConfig(DefaultConfigFilePath)
@@ -48,6 +29,30 @@ func LoadConfig() (err error) {
 	return
 }
 
+func InitConfigFile() {
+	var err error
+	log.Println("没有找到配置文件：", DefaultConfigFilePath)
+	log.Println("开始生成默认的空白配置文件")
+	ConfigMode.Common.BindAddr = DefaultBindAddr
+	ConfigMode.Common.KcpPort = DefaultKcpPort
+	ConfigMode.Common.TcpPort = DefaultTcpPort
+	ConfigMode.Common.TlsPort = DefaultTlsPort
+	ConfigMode.Common.GrpcPort = DefaultGrpcPort
+	ConfigMode.Common.HttpPort = DefaultHttpPort
+	ConfigMode.Common.HttpsPort = DefaultHttpsPort
+	ConfigMode.Common.UdpApiPort = DefaultUdpApiPort
+	ConfigMode.Common.KcpApiPort = DefaultKcpApiPort
+	ConfigMode.Security.LoginKey = DefaultLoginKey
+	//	生成配置文件模板
+	err = writeConfigFile(ConfigMode, DefaultConfigFilePath)
+	if err != nil {
+		log.Printf("写入默认的配置文件失败：%s\n", err.Error())
+		return
+	}
+	log.Println("配置文件写入成功,路径为：", DefaultConfigFilePath)
+	log.Println("你也可以修改上述配置文件后在运行")
+}
+
 //从配置文件路径解析配置文件的内容
 func GetConfig(configFilePath string) (configMode models.ServerConfig, err error) {
 	content, err := ioutil.ReadFile(configFilePath)

main.go

@@ -40,6 +40,16 @@ func main() {
 			Name:    "generate",
 			Aliases: []string{"g"},
 			Usage:   "generate one token for gateway and one token for OpenIoTHub",
+			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:        "config",
+					Aliases:     []string{"c"},
+					Value:       config.DefaultConfigFilePath,
+					Usage:       "config file path",
+					EnvVars:     []string{"GatewayConfigFilePath"},
+					Destination: &config.DefaultConfigFilePath,
+				},
+			},
 			Action: func(c *cli.Context) error {
 				err := config.LoadConfig()
 				if err != nil {
@@ -60,6 +70,25 @@ func main() {
 				return nil
 			},
 		},
+		{
+			Name:    "init",
+			Aliases: []string{"i"},
+			Usage:   "init config file",
+			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:        "config",
+					Aliases:     []string{"c"},
+					Value:       config.DefaultConfigFilePath,
+					Usage:       "config file path",
+					EnvVars:     []string{"GatewayConfigFilePath"},
+					Destination: &config.DefaultConfigFilePath,
+				},
+			},
+			Action: func(c *cli.Context) error {
+				config.InitConfigFile()
+				return nil
+			},
+		},
 	}
 	myApp.Action = func(c *cli.Context) error {
 		err := run()

test/main.go

@@ -0,0 +1,16 @@
+package main
+
+import (
+	"fmt"
+	"github.com/OpenIoTHub/utils/models"
+	"log"
+)
+
+func main() {
+	t, err := models.DecodeToken("123abc", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJSdW5JZCI6IjBiYmM1NWRkLWNhNzAtNGRhNC1iNDAzLTRmODQ3MThkNWEyNCIsIkhvc3QiOiIzNi42My4zOS4xODUiLCJUY3BQb3J0IjozNDMyMCwiS2NwUG9ydCI6MzQzMjAsIlRsc1BvcnQiOjM0MzIxLCJHcnBjUG9ydCI6MzQzMjIsIlVEUEFwaVBvcnQiOjM0MzIxLCJLQ1BBcGlQb3J0IjozNDMyMiwiUGVybWlzc2lvbiI6MSwiZXhwIjoyMDE1OTU3NzU5OTIsIm5iZiI6MTU5NTc0NzE5Mn0.COUJwk3x6RoHZ-ajeOGqTsek9BinwLxjAlgRDriI_Wc")
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	fmt.Println(t)
+}