feat(wallet): 添加可自定义的名字接口 (#52)

FloatTech · May 11, 2024 · f91b28e · f91b28e
1 parent e1a4844
commit f91b28e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/wallet/wallet.go b/wallet/wallet.go
@@ -30,6 +30,7 @@ var (
 			DBPath: "data/wallet/wallet.db",
 		},
 	}
+	walletName = "Atri币"
 )
 
 func init() {
@@ -49,6 +50,16 @@ func init() {
 	}
 }
 
+// GetWalletName 获取货币名称
+func GetWalletName() string {
+	return walletName
+}
+
+// SetWalletName 设置货币名称
+func SetWalletName(name string) {
+	walletName = name
+}
+
 // GetWalletOf 获取钱包数据
 func GetWalletOf(uid int64) (money int) {
 	return sdb.getWalletOf(uid).Money