config add dataDir

README.md

@@ -14,5 +14,7 @@ ServerAPI = "http://127.0.0.1:18090"
 isTestNet = false
 # feeLimit, the maximum energy is 1000000000
 feeLimit = 10000000
+# Cache data file directory, default = "", current directory: ./data
+dataDir = ""
 
 ```

tron/config.go

@@ -109,6 +109,8 @@ type WalletConfig struct {
 	FeeLimit int64
 	//FeeMini 智能合约最小能量消耗起，1 Energy = 10 SUN
 	FeeMini int64
+	//数据目录
+	DataDir string
 }
 
 //NewConfig Create config instance
@@ -197,9 +199,9 @@ feeLimit = 10000000
 `
 
 	//创建目录
-	file.MkdirAll(c.dbPath)
-	file.MkdirAll(c.backupDir)
-	file.MkdirAll(c.keyDir)
+	//file.MkdirAll(c.dbPath)
+	//file.MkdirAll(c.backupDir)
+	//file.MkdirAll(c.keyDir)
 
 	return &c
 }
@@ -248,3 +250,18 @@ func (wc *WalletConfig) InitConfig() {
 	}
 
 }
+
+//创建文件夹
+func (wc *WalletConfig) makeDataDir() {
+
+	if len(wc.DataDir) == 0 {
+		//默认路径当前文件夹./data
+		wc.DataDir = "data"
+	}
+
+	//本地数据库文件路径
+	wc.dbPath = filepath.Join(wc.DataDir, strings.ToLower(wc.Symbol), "db")
+
+	//创建目录
+	file.MkdirAll(wc.dbPath)
+}

tron/config_load.go

@@ -87,5 +87,9 @@ func (wm *WalletManager) LoadAssetsConfig(c config.Configer) error {
 	//}
 
 	wm.WalletClient = NewClient(wm.Config.ServerAPI, "", false)
+	wm.Config.DataDir = c.String("dataDir")
+
+	//数据文件夹
+	wm.Config.makeDataDir()
 	return nil
 }