25개 이상의 토픽을 선택하실 수 없습니다.
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.3 KiB
76 lines
2.3 KiB
package service
|
|
|
|
import (
|
|
"beeblog/models"
|
|
"github.com/beego/beego/v2/client/orm"
|
|
)
|
|
|
|
type UserService struct {
|
|
}
|
|
|
|
func (this *UserService) GetUser(id int64) (*models.User, error) {
|
|
o := orm.NewOrm()
|
|
user := &models.User{Id: id}
|
|
err := o.Read(user)
|
|
return user, err
|
|
}
|
|
|
|
func (this *UserService) FindByUserName(username string) (*models.User, error) {
|
|
o := orm.NewOrm()
|
|
qs := o.QueryTable(&models.User{})
|
|
var users []*models.User
|
|
_, err := qs.Filter("UserName", username).All(&users)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(users) != 0 {
|
|
return users[0], nil
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (this *UserService) SaveUser(user *models.User) error {
|
|
o := orm.NewOrm()
|
|
id, eror := o.Insert(user)
|
|
if eror != nil {
|
|
return eror
|
|
} else {
|
|
user.Id = id
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (this *UserService) CountBlog(uid int64) {
|
|
o := orm.NewOrm()
|
|
browses := 0
|
|
o.Raw("UPDATE `auth_user` SET `blog_count` = (SELECT count(id) num FROM blog where delflag = 0 and user_id =? ) WHERE `id` = ? ", uid, uid).QueryRow(&browses)
|
|
return
|
|
}
|
|
func (this *UserService) CountBrows(uid int64) {
|
|
o := orm.NewOrm()
|
|
browses := 0
|
|
o.Raw("UPDATE `auth_user` SET `blog_browes` = (select SUM(browses) browses from blog where user_id = ?) WHERE `id` = ? ", uid, uid).QueryRow(&browses) //获取总条数
|
|
return
|
|
}
|
|
func (this *UserService) CountComments(uid int64, bid int64) {
|
|
o := orm.NewOrm()
|
|
browses := 0
|
|
o.Raw("UPDATE `auth_user` SET `blog_comment` = (select count(id) num from comment where cuser_id = ?) WHERE `id` = ? ", uid, uid).QueryRow(&browses) //获取总条数
|
|
o.Raw("UPDATE `blog` SET `comments` = (select count(id) num from comment where blog_id = ?) WHERE `id` = ? ", bid, bid).QueryRow(&browses) //获取总条数
|
|
return
|
|
}
|
|
func (this *UserService) CountLike(uid int64, bid int64) {
|
|
o := orm.NewOrm()
|
|
browses := 0
|
|
o.Raw("UPDATE `auth_user` SET `blog_like` = (select count(id) num from tb_like where user_id = ?) WHERE `id` = ?", uid, uid).QueryRow(&browses)
|
|
o.Raw("UPDATE `blog` SET `likes` = (select count(id) num from tb_like where blog_id = ?) WHERE `id` = ?", bid, bid).QueryRow(&browses)
|
|
return
|
|
}
|
|
|
|
func (this *UserService) EditUser(user *models.User) (int64, error) {
|
|
return orm.NewOrm().Update(user)
|
|
}
|
|
|
|
func (this *UserService) EditHeadImg(user *models.User) (int64, error) {
|
|
return orm.NewOrm().Update(user, "Headimg")
|
|
}
|
|
|