From 34c941e4e9a68037c67839c1edb442074292564c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E7=A5=A5=E8=B6=85?= <iwind.liu@gmail.com>
Date: Fri, 20 Nov 2020 16:36:07 +0800
Subject: [PATCH] =?UTF-8?q?[=E6=93=8D=E4=BD=9C=E6=97=A5=E5=BF=97]=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E8=8B=A5=E5=B9=B2=E6=9F=A5=E8=AF=A2=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/db/models/log_dao.go        | 53 +++++++++++++++++++++++++---
 internal/db/models/log_model.go      |  2 ++
 internal/db/models/user_model.go     |  6 ++++
 internal/rpc/services/service_log.go |  4 +--
 4 files changed, 58 insertions(+), 7 deletions(-)

diff --git a/internal/db/models/log_dao.go b/internal/db/models/log_dao.go
index 801636c8..23e4c113 100644
--- a/internal/db/models/log_dao.go
+++ b/internal/db/models/log_dao.go
@@ -4,6 +4,9 @@ import (
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/dbs"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"regexp"
+	"strings"
 )
 
 type LogDAO dbs.DAO
@@ -32,20 +35,51 @@ func (this *LogDAO) CreateLog(adminType string, adminId int64, level string, des
 	op := NewLogOperator()
 	op.Type = adminType
 	op.AdminId, op.Level, op.Description, op.Action, op.Ip = adminId, level, description, action, ip
+	op.Day = timeutil.Format("Ymd")
 	op.Type = LogTypeAdmin
 	_, err := this.Save(op)
 	return err
 }
 
 // 计算所有日志数量
-func (this *LogDAO) CountAllLogs() (int64, error) {
-	return this.Query().
-		Count()
+func (this *LogDAO) CountLogs(dayFrom string, dayTo string, keyword string) (int64, error) {
+	dayFrom = this.formatDay(dayFrom)
+	dayTo = this.formatDay(dayTo)
+
+	query := this.Query()
+
+	if len(dayFrom) > 0 {
+		query.Gte("day", dayFrom)
+	}
+	if len(dayTo) > 0 {
+		query.Lte("day", dayTo)
+	}
+	if len(keyword) > 0 {
+		query.Where("(description LIKE :keyword OR ip LIKE :keyword OR action LIKE :keyword)").
+			Param("keyword", "%"+keyword+"%")
+	}
+
+	return query.Count()
 }
 
 // 列出单页日志
-func (this *LogDAO) ListLogs(offset int64, size int64) (result []*Log, err error) {
-	_, err = this.Query().
+func (this *LogDAO) ListLogs(offset int64, size int64, dayFrom string, dayTo string, keyword string) (result []*Log, err error) {
+	dayFrom = this.formatDay(dayFrom)
+	dayTo = this.formatDay(dayTo)
+
+	query := this.Query()
+	if len(dayFrom) > 0 {
+		query.Gte("day", dayFrom)
+	}
+	if len(dayTo) > 0 {
+		query.Lte("day", dayTo)
+	}
+	if len(keyword) > 0 {
+		query.Where("(description LIKE :keyword OR ip LIKE :keyword OR action LIKE :keyword)").
+			Param("keyword", "%"+keyword+"%")
+	}
+
+	_, err = query.
 		Offset(offset).
 		Limit(size).
 		Slice(&result).
@@ -53,3 +87,12 @@ func (this *LogDAO) ListLogs(offset int64, size int64) (result []*Log, err error
 		FindAll()
 	return
 }
+
+// 格式化日期
+func (this *LogDAO) formatDay(day string) string {
+	if !regexp.MustCompile(`^\d{4}-\d{2}-\d{2}$`).MatchString(day) {
+		return ""
+	}
+	day = strings.ReplaceAll(day, "-", "")
+	return day
+}
diff --git a/internal/db/models/log_model.go b/internal/db/models/log_model.go
index 846022a3..bb2c050e 100644
--- a/internal/db/models/log_model.go
+++ b/internal/db/models/log_model.go
@@ -12,6 +12,7 @@ type Log struct {
 	ProviderId  uint32 `field:"providerId"`  // 供应商ID
 	Ip          string `field:"ip"`          // IP地址
 	Type        string `field:"type"`        // 类型：admin, user
+	Day         string `field:"day"`         // 日期
 }
 
 type LogOperator struct {
@@ -25,6 +26,7 @@ type LogOperator struct {
 	ProviderId  interface{} // 供应商ID
 	Ip          interface{} // IP地址
 	Type        interface{} // 类型：admin, user
+	Day         interface{} // 日期
 }
 
 func NewLogOperator() *LogOperator {
diff --git a/internal/db/models/user_model.go b/internal/db/models/user_model.go
index b883ca19..be1a89ce 100644
--- a/internal/db/models/user_model.go
+++ b/internal/db/models/user_model.go
@@ -9,6 +9,9 @@ type User struct {
 	CreatedAt uint64 `field:"createdAt"` // 创建时间
 	UpdatedAt uint64 `field:"updatedAt"` // 修改时间
 	State     uint8  `field:"state"`     // 状态
+	Mobile    string `field:"mobile"`    // 手机号
+	Remark    string `field:"remark"`    // 备注
+	Email     string `field:"email"`     // 邮箱地址
 }
 
 type UserOperator struct {
@@ -19,6 +22,9 @@ type UserOperator struct {
 	CreatedAt interface{} // 创建时间
 	UpdatedAt interface{} // 修改时间
 	State     interface{} // 状态
+	Mobile    interface{} // 手机号
+	Remark    interface{} // 备注
+	Email     interface{} // 邮箱地址
 }
 
 func NewUserOperator() *UserOperator {
diff --git a/internal/rpc/services/service_log.go b/internal/rpc/services/service_log.go
index 145585ae..b0771dc6 100644
--- a/internal/rpc/services/service_log.go
+++ b/internal/rpc/services/service_log.go
@@ -34,7 +34,7 @@ func (this *LogService) CountLogs(ctx context.Context, req *pb.CountLogRequest)
 		return nil, err
 	}
 
-	count, err := models.SharedLogDAO.CountAllLogs()
+	count, err := models.SharedLogDAO.CountLogs(req.DayFrom, req.DayTo, req.Keyword)
 	if err != nil {
 		return nil, err
 	}
@@ -49,7 +49,7 @@ func (this *LogService) ListLogs(ctx context.Context, req *pb.ListLogsRequest) (
 		return nil, err
 	}
 
-	logs, err := models.SharedLogDAO.ListLogs(req.Offset, req.Size)
+	logs, err := models.SharedLogDAO.ListLogs(req.Offset, req.Size, req.DayFrom, req.DayTo, req.Keyword)
 	if err != nil {
 		return nil, err
 	}