优化提示

Jrohy · Nov 18, 2020 · 97ea54a · 97ea54a
1 parent 54e2227
commit 97ea54a
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -68,6 +68,7 @@ Usage:
 
 Available Commands:
   add         添加用户
+  clean       清空指定用户流量
   completion  自动命令补全(支持bash和zsh)
   del         删除用户
   help        Help about any command

diff --git a/cmd/clean.go b/cmd/clean.go
@@ -9,7 +9,7 @@ import (
 var cleanCmd = &cobra.Command{
 	Use:   "clean",
 	Short: "清空指定用户流量",
-	Long: `传入指定用户名来清空用户流量, 多个空格隔开, 例如:
+	Long: `传入指定用户名来清空用户流量, 多个用户名空格隔开, 例如:
 trojan clean zhangsan lisi
 `,
 	Args: cobra.MinimumNArgs(1),

diff --git a/core/mysql.go b/core/mysql.go
@@ -12,10 +12,10 @@ import (
 
 	mysqlDriver "github.com/go-sql-driver/mysql"
 
-	// mysql sql驱动
 	"strconv"
 	"strings"
 
+	// mysql sql驱动
 	_ "github.com/go-sql-driver/mysql"
 )
 

diff --git a/trojan/user.go b/trojan/user.go
@@ -106,6 +106,8 @@ func CleanDataByName(usernames []string) {
 	mysql := core.GetMysql()
 	if err := mysql.CleanDataByName(usernames); err != nil {
 		fmt.Println(err.Error())
+	} else {
+		fmt.Println("清空流量成功!")
 	}
 }