feat 结构调整
This commit is contained in:
40
Server/user/internal/dao/repository/users.go
Normal file
40
Server/user/internal/dao/repository/users.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"common/db/mysql"
|
||||
"context"
|
||||
"user/internal/dao/model"
|
||||
"user/internal/dao/query"
|
||||
)
|
||||
|
||||
type UserDao struct {
|
||||
ctx context.Context
|
||||
query *query.Query
|
||||
}
|
||||
|
||||
func NewUserDao(ctx context.Context) *UserDao {
|
||||
return &UserDao{
|
||||
ctx: ctx,
|
||||
query: query.Use(mysql.GetDB("user_db")),
|
||||
}
|
||||
}
|
||||
|
||||
func (d *UserDao) FindBySn(sn int64) (*model.User, error) {
|
||||
first, err := d.query.User.WithContext(d.ctx).
|
||||
Where(d.query.User.Sn.Eq(sn)).
|
||||
First()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return first, nil
|
||||
}
|
||||
|
||||
func (d *UserDao) FindByPhone(phone string) (*model.User, error) {
|
||||
first, err := d.query.User.WithContext(d.ctx).
|
||||
Where(d.query.User.Phone.Eq(phone)).
|
||||
First()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return first, nil
|
||||
}
|
||||
Reference in New Issue
Block a user