From a1f5fa6446dc1c6444483b1c3e1fed81a93f3e53 Mon Sep 17 00:00:00 2001
From: zijiren233 <pyh1670605849@gmail.com>
Date: Wed, 25 Oct 2023 22:03:53 +0800
Subject: [PATCH] Opt: db log color

---
 internal/bootstrap/db.go |  4 ++--
 internal/op/users.go     | 16 +---------------
 2 files changed, 3 insertions(+), 17 deletions(-)

diff --git a/internal/bootstrap/db.go b/internal/bootstrap/db.go
index 0a6febb..4855f82 100644
--- a/internal/bootstrap/db.go
+++ b/internal/bootstrap/db.go
@@ -135,8 +135,8 @@ func newDBLogger() logger.Interface {
 			SlowThreshold:             time.Second,
 			LogLevel:                  logLevel,
 			IgnoreRecordNotFoundError: true,
-			ParameterizedQueries:      true,
-			Colorful:                  false,
+			ParameterizedQueries:      flags.Dev,
+			Colorful:                  true,
 		},
 	)
 }
diff --git a/internal/op/users.go b/internal/op/users.go
index 01eb239..814217e 100644
--- a/internal/op/users.go
+++ b/internal/op/users.go
@@ -30,20 +30,6 @@ func GetUserById(id uint) (*User, error) {
 	return u2, userCache.SetWithExpire(id, u2, time.Hour)
 }
 
-// slow
-func GetUserByUsername(username string) (*User, error) {
-	u, err := db.GetUserByUsername(username)
-	if err != nil {
-		return nil, err
-	}
-
-	u2 := &User{
-		User: *u,
-	}
-
-	return u2, userCache.SetWithExpire(u.ID, u2, time.Hour)
-}
-
 func CreateUser(username string, p provider.OAuth2Provider, pid uint, conf ...db.CreateUserConfig) (*User, error) {
 	if username == "" {
 		return nil, errors.New("username cannot be empty")
@@ -104,7 +90,7 @@ func DeleteUserByID(userID uint) error {
 }
 
 func SaveUser(u *model.User) error {
-	userCache.Remove(u.ID)
+	defer userCache.Remove(u.ID)
 	return db.SaveUser(u)
 }