From 97ea54acbf506551a63582d8d7ad1d7a425be55c Mon Sep 17 00:00:00 2001
From: Jrohy <euvkzx@gmail.com>
Date: Wed, 18 Nov 2020 15:11:32 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=8F=90=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md      | 1 +
 cmd/clean.go   | 2 +-
 core/mysql.go  | 2 +-
 trojan/user.go | 2 ++
 4 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 837578db..6df4e806 100644
--- 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
index c88d0604..a67ea517 100644
--- 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
index 0e2116d4..9f00004a 100644
--- 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
index 48b2a5b5..9701cf1c 100644
--- 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("清空流量成功!")
 	}
 }