瀏覽代碼

增加是否开启FTP数据导入功能开关

zhou.xiaoning 2 年之前
父節點
當前提交
7061d9359f
共有 3 個文件被更改,包括 9 次插入5 次删除
  1. 1 0
      config.yaml
  2. 1 0
      config/ftp.go
  3. 7 5
      main.go

+ 1 - 0
config.yaml

@@ -33,6 +33,7 @@ system:
 
 # ftp configuration
 ftp:
+  enable: true
   address: '218.17.158.45'
   port: '23110'
   name: 'wmsuser'

+ 1 - 0
config/ftp.go

@@ -1,6 +1,7 @@
 package config
 
 type FTP struct {
+	Enable      bool   `mapstructure:"enable" json:"enable" yaml:"enable"`
 	Address     string `mapstructure:"address" json:"address" yaml:"address"`
 	Port        string `mapstructure:"port" json:"port" yaml:"port"`
 	Name        string `mapstructure:"name" json:"name" yaml:"name"`

+ 7 - 5
main.go

@@ -41,12 +41,14 @@ func main() {
 	service.StartRabbitMQReceive()
 
 	// 连接FTP
-	var err error
-	if global.M2A_FTP, err = core.FTP(); err == nil {
-		defer global.M2A_FTP.Quit()
+	if global.M2A_CONFIG.FTP.Enable {
+		var err error
+		if global.M2A_FTP, err = core.FTP(); err == nil {
+			defer global.M2A_FTP.Quit()
+		}
+		// 开始FTP Timer任务
+		service.InitTimer()
 	}
-	// 开始FTP Timer任务
-	service.InitTimer()
 
 	// 判断输入参数
 	var ftp bool