优化提示

Author: Jrohy

README.md

@@ -68,6 +68,7 @@ Usage:
 
 Available Commands:
   add         添加用户
+  clean       清空指定用户流量
   completion  自动命令补全(支持bash和zsh)
   del         删除用户
   help        Help about any command

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),

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"
 )
 

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("清空流量成功!")
 	}
 }