官方更新内容

pull/361/head
HXY 2 years ago
parent 7f68cac0e1
commit a42f837535

@ -5,29 +5,29 @@
package jinzhu package jinzhu
import ( import (
"strings"
"time"
"github.com/rocboss/paopao-ce/internal/core" "github.com/rocboss/paopao-ce/internal/core"
"github.com/rocboss/paopao-ce/internal/core/ms"
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr" "github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
"gorm.io/gorm" "gorm.io/gorm"
"strings"
"time"
) )
var ( var (
_ core.UserManageService = (*userManageServant)(nil) _ core.UserManageService = (*userManageSrv)(nil)
) )
type userManageServant struct { type userManageSrv struct {
db *gorm.DB db *gorm.DB
} }
func newUserManageService(db *gorm.DB) core.UserManageService { func newUserManageService(db *gorm.DB) core.UserManageService {
return &userManageServant{ return &userManageSrv{
db: db, db: db,
} }
} }
func (s *userManageServant) GetUserByID(id int64) (*core.User, error) { func (s *userManageSrv) GetUserByID(id int64) (*ms.User, error) {
user := &dbr.User{ user := &dbr.User{
Model: &dbr.Model{ Model: &dbr.Model{
ID: id, ID: id,
@ -36,28 +36,28 @@ func (s *userManageServant) GetUserByID(id int64) (*core.User, error) {
return user.Get(s.db) return user.Get(s.db)
} }
func (s *userManageServant) GetUserByUsername(username string) (*core.User, error) { func (s *userManageSrv) GetUserByUsername(username string) (*ms.User, error) {
user := &dbr.User{ user := &dbr.User{
Username: username, Username: username,
} }
return user.Get(s.db) return user.Get(s.db)
} }
func (s *userManageServant) GetUserByPhone(phone string) (*core.User, error) { func (s *userManageSrv) GetUserByPhone(phone string) (*ms.User, error) {
user := &dbr.User{ user := &dbr.User{
Phone: phone, Phone: phone,
} }
return user.Get(s.db) return user.Get(s.db)
} }
func (s *userManageServant) GetUsersByIDs(ids []int64) ([]*core.User, error) { func (s *userManageSrv) GetUsersByIDs(ids []int64) ([]*ms.User, error) {
user := &dbr.User{} user := &dbr.User{}
return user.List(s.db, &dbr.ConditionsT{ return user.List(s.db, &dbr.ConditionsT{
"id IN ?": ids, "id IN ?": ids,
}, 0, 0) }, 0, 0)
} }
func (s *userManageServant) GetUsersByKeyword(keyword string) ([]*core.User, error) { func (s *userManageSrv) GetUsersByKeyword(keyword string) ([]*ms.User, error) {
user := &dbr.User{} user := &dbr.User{}
keyword = strings.Trim(keyword, " ") + "%" keyword = strings.Trim(keyword, " ") + "%"
if keyword == "%" { if keyword == "%" {
@ -71,26 +71,11 @@ func (s *userManageServant) GetUsersByKeyword(keyword string) ([]*core.User, err
} }
} }
func (s *userManageServant) GetTagsByKeyword(keyword string) ([]*core.Tag, error) { func (s *userManageSrv) CreateUser(user *dbr.User) (*ms.User, error) {
tag := &dbr.Tag{}
keyword = "%" + strings.Trim(keyword, " ") + "%"
if keyword == "%%" {
return tag.List(s.db, &dbr.ConditionsT{
"ORDER": "quote_num DESC",
}, 0, 6)
} else {
return tag.List(s.db, &dbr.ConditionsT{
"tag LIKE ?": keyword,
"ORDER": "quote_num DESC",
}, 0, 6)
}
}
func (s *userManageServant) CreateUser(user *dbr.User) (*core.User, error) {
return user.Create(s.db) return user.Create(s.db)
} }
func (s *userManageServant) UpdateUser(user *core.User) error { func (s *userManageSrv) UpdateUser(user *ms.User) error {
return s.db.Transaction(func(tx *gorm.DB) error { return s.db.Transaction(func(tx *gorm.DB) error {
//获取用户的账单,判断是否赠送过 //获取用户的账单,判断是否赠送过
bills, err := (&dbr.WalletStatement{}).List(tx, &dbr.ConditionsT{ bills, err := (&dbr.WalletStatement{}).List(tx, &dbr.ConditionsT{

@ -7,6 +7,7 @@ package jinzhu
import ( import (
"github.com/rocboss/paopao-ce/internal/conf" "github.com/rocboss/paopao-ce/internal/conf"
"github.com/rocboss/paopao-ce/internal/core" "github.com/rocboss/paopao-ce/internal/core"
"github.com/rocboss/paopao-ce/internal/core/ms"
"github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr" "github.com/rocboss/paopao-ce/internal/dao/jinzhu/dbr"
"gorm.io/gorm" "gorm.io/gorm"
"time" "time"
@ -26,7 +27,7 @@ func newWalletService(db *gorm.DB) core.WalletService {
} }
} }
func (d *walletServant) GetRechargeByID(id int64) (*core.WalletRecharge, error) { func (d *walletServant) GetRechargeByID(id int64) (*ms.WalletRecharge, error) {
recharge := &dbr.WalletRecharge{ recharge := &dbr.WalletRecharge{
Model: &dbr.Model{ Model: &dbr.Model{
ID: id, ID: id,
@ -37,7 +38,7 @@ func (d *walletServant) GetRechargeByID(id int64) (*core.WalletRecharge, error)
} }
// 点击前往支付会创建支付数据 // 点击前往支付会创建支付数据
func (d *walletServant) CreateRecharge(userId, amount int64) (*core.WalletRecharge, error) { func (d *walletServant) CreateRecharge(userId, amount int64) (*ms.WalletRecharge, error) {
recharge := &dbr.WalletRecharge{ recharge := &dbr.WalletRecharge{
UserID: userId, UserID: userId,
Amount: amount, Amount: amount,
@ -46,7 +47,7 @@ func (d *walletServant) CreateRecharge(userId, amount int64) (*core.WalletRechar
return recharge.Create(d.db) return recharge.Create(d.db)
} }
func (d *walletServant) GetUserWalletBills(userID int64, offset, limit int) ([]*core.WalletStatement, error) { func (d *walletServant) GetUserWalletBills(userID int64, offset, limit int) ([]*ms.WalletStatement, error) {
statement := &dbr.WalletStatement{ statement := &dbr.WalletStatement{
UserID: userID, UserID: userID,
} }
@ -63,7 +64,7 @@ func (d *walletServant) GetUserWalletBillCount(userID int64) (int64, error) {
return statement.Count(d.db, &dbr.ConditionsT{}) return statement.Count(d.db, &dbr.ConditionsT{})
} }
func (d *walletServant) HandleRechargeSuccess(recharge *core.WalletRecharge, tradeNo string) error { func (d *walletServant) HandleRechargeSuccess(recharge *ms.WalletRecharge, tradeNo string) error {
user, _ := (&dbr.User{ user, _ := (&dbr.User{
Model: &dbr.Model{ Model: &dbr.Model{
ID: recharge.UserID, ID: recharge.UserID,
@ -111,7 +112,7 @@ func (d *walletServant) HandleRechargeSuccess(recharge *core.WalletRecharge, tra
}) })
} }
func (d *walletServant) HandlePostAttachmentBought(post *core.Post, user *core.User) error { func (d *walletServant) HandlePostAttachmentBought(post *ms.Post, user *ms.User) error {
return d.db.Transaction(func(tx *gorm.DB) error { return d.db.Transaction(func(tx *gorm.DB) error {
// 扣除金额 // 扣除金额
if err := tx.Model(user).Update("balance", gorm.Expr("balance - ?", post.AttachmentPrice)).Error; err != nil { if err := tx.Model(user).Update("balance", gorm.Expr("balance - ?", post.AttachmentPrice)).Error; err != nil {

@ -96,4 +96,6 @@ var (
ErrUserNameEmpty = xerror.NewError(11003, "用户名为空") ErrUserNameEmpty = xerror.NewError(11003, "用户名为空")
ErrDeleteUserKeyFailed = xerror.NewError(11004, "删除用户Share Key失败") ErrDeleteUserKeyFailed = xerror.NewError(11004, "删除用户Share Key失败")
ErrGetUserShareKeyCountFailed = xerror.NewError(11005, "获取用户Share Key数量失败") ErrGetUserShareKeyCountFailed = xerror.NewError(11005, "获取用户Share Key数量失败")
ErrNotImplemented = xerror.NewError(10501, "功能未实现")
) )

Loading…
Cancel
Save