From 9776bdc9b8b441ebd73c100bfe2aa4e495c233ca Mon Sep 17 00:00:00 2001 From: Joe Chen Date: Fri, 10 Jun 2022 19:54:36 +0800 Subject: db: use `context` and go-mockgen for login sources (#7041) --- internal/db/users.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'internal/db/users.go') diff --git a/internal/db/users.go b/internal/db/users.go index 096a2702..5c8f2d38 100644 --- a/internal/db/users.go +++ b/internal/db/users.go @@ -5,6 +5,7 @@ package db import ( + "context" "fmt" "strings" "time" @@ -80,6 +81,8 @@ func (err ErrLoginSourceMismatch) Error() string { } func (db *users) Authenticate(login, password string, loginSourceID int64) (*User, error) { + ctx := context.TODO() + login = strings.ToLower(login) var query *gorm.DB @@ -127,7 +130,7 @@ func (db *users) Authenticate(login, password string, loginSourceID int64) (*Use createNewUser = true } - source, err := LoginSources.GetByID(authSourceID) + source, err := LoginSources.GetByID(ctx, authSourceID) if err != nil { return nil, errors.Wrap(err, "get login source") } -- cgit v1.2.3