码农笔录博客源码
Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.
 
 
 
 
 

248 lignes
5.7 KiB

package service
import (
"beeblog/models"
"beeblog/utils"
"github.com/beego/beego/v2/client/orm"
)
type BlogService struct {
}
func (this *BlogService) count(num int, size int, cat int64) (*utils.Page, error) {
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
qs.Filter("Delflag", 0)
if cat != -1 {
qs = qs.Filter("CategoryId", cat)
}
totalCount, err := qs.Count()
if err != nil {
return nil, err
}
return utils.PageUtil(totalCount, num, size), nil
}
func (this *BlogService) countByUser(num int, size int, uid int64) (*utils.Page, error) {
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
totalCount, err := qs.Filter("Delflag", 0).Filter("UserId", uid).Count()
if err != nil {
return nil, err
}
return utils.PageUtil(totalCount, num, size), nil
}
func (this *BlogService) EditBlogBrows(id int64) {
o := orm.NewOrm()
blog := &models.Blog{Id: id}
err := o.Read(blog)
if err == nil {
blog.Browses += 1
o.Update(blog, "Browses")
}
}
func (this *BlogService) TopBlogByUser(uid int64) ([]*models.Blog, error) {
o := orm.NewOrm()
var blogs []*models.Blog
o.QueryTable(models.Blog{}).Filter("Delflag", 0).Filter("UserId", uid).Limit(12, 0).OrderBy("-Browses").All(&blogs)
return blogs, nil
}
func (this *BlogService) ReadBlog(id int64) (*models.Blog, error) {
o := orm.NewOrm()
blog := &models.Blog{Id: id}
if err := o.Read(blog); err != nil {
return nil, err
}
return blog, nil
}
func (this *BlogService) GetBlog(id int64) (*models.Blog, error) {
o := orm.NewOrm()
blog := &models.Blog{Id: id}
err := o.Read(blog)
if err != nil {
return nil, err
}
user := &models.User{Id: blog.UserId}
err = o.Read(user)
if err == nil {
blog.User = user
}
var labels []*models.NLabel
qs := o.QueryTable(&models.NLabel{})
_, err = qs.Filter("BlogId", id).All(&labels)
if err == nil {
blog.Lables = labels
}
category := &models.Category{Id: blog.CategoryId}
if err := o.Read(category); err == nil {
blog.CateName = category.Title
}
return blog, nil
}
func (this *BlogService) DelBlog(blog *models.Blog) error {
o := orm.NewOrm()
_, err := o.Update(blog, "Delflag")
return err
}
func (this *BlogService) FindBlogs(num int, size int, cat int64, flag int) (*utils.Page, error) {
page, err := this.count(num, size, cat)
if err != nil {
return nil, err
}
var blogs []*models.Blog
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
qs = qs.Filter("Delflag", 0)
if cat != -1 {
qs = qs.Filter("CategoryId", cat)
}
if flag == 0 {
qs = qs.OrderBy("-Ctime")
} else {
qs = qs.OrderBy("-Browses")
}
qs = qs.Limit(size, (page.PageNo-1)*size)
_, err = qs.All(&blogs)
if err != nil {
return nil, err
}
if len(blogs) > 0 {
for i := 0; i < len(blogs); i++ {
user := &models.User{Id: blogs[i].UserId}
err = o.Read(user)
if err == nil {
blogs[i].UserName = user.UserName
blogs[i].HeadImg = user.Headimg
}
category := &models.Category{Id: blogs[i].CategoryId}
err = o.Read(category)
if err == nil {
blogs[i].CateName = category.Title
}
}
}
page.List = blogs
return page, nil
}
func (this *BlogService) AllBlogs() ([]*models.Blog, error) {
var blogs []*models.Blog
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
qs = qs.Filter("Delflag", 0)
_, err := qs.All(&blogs)
if err != nil {
return nil, err
}
return blogs, nil
}
func (this *BlogService) SaveBlog(blog *models.Blog, strs []string) error {
o := orm.NewOrm()
tx, _ := o.Begin()
id, eror := o.Insert(blog)
if eror != nil {
tx.Rollback()
return eror
} else {
blog.Id = id
if strs != nil && len(strs) > 0 {
nlabels := make([]*models.NLabel, len(strs))
for i := 0; i < len(strs); i++ {
nlabels[i] = &models.NLabel{Title: strs[i], BlogId: id, UserId: blog.UserId}
}
if _, err := o.InsertMulti(len(nlabels), nlabels); err != nil {
tx.Rollback()
return err
}
}
tx.Commit()
}
return nil
}
func (this *BlogService) EditBlog(blog *models.Blog, strs []string) error {
o := orm.NewOrm()
tx, _ := o.Begin()
_, eror := o.Update(blog)
if eror != nil {
tx.Rollback()
return eror
} else {
o.QueryTable(models.NLabel{}).Filter("BlogId", blog.Id).Delete()
if strs != nil && len(strs) > 0 {
nlabels := make([]*models.NLabel, len(strs))
for i := 0; i < len(strs); i++ {
nlabels[i] = &models.NLabel{Title: strs[i], BlogId: blog.Id, UserId: blog.UserId}
}
if _, err := o.InsertMulti(len(nlabels), nlabels); err != nil {
tx.Rollback()
return err
}
}
tx.Commit()
}
return nil
}
func (this *BlogService) MeBlogs(num int, size int, flag int, uid int64) (*utils.Page, error) {
page, err := this.countByUser(num, size, uid)
if err != nil {
return nil, err
}
var blogs []*models.Blog
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
qs = qs.Filter("Delflag", 0)
qs = qs.Filter("UserId", uid)
if flag == 0 {
qs = qs.OrderBy("-Ctime")
} else {
qs = qs.OrderBy("-Browses")
}
qs = qs.Limit(size, (page.PageNo-1)*size)
_, err = qs.All(&blogs)
if err != nil {
return nil, err
}
page.List = blogs
return page, nil
}
func (this *BlogService) IndexBlogs(size int, flag int) ([]*models.Blog, error) {
var blogs []*models.Blog
o := orm.NewOrm()
qs := o.QueryTable(&models.Blog{})
qs = qs.Filter("Delflag", 0)
switch flag {
case 0:
qs = qs.OrderBy("-Ctime") //最新
case 1:
qs = qs.OrderBy("-Browses") //浏览量
case 2:
qs = qs.OrderBy("-Likes") // 收藏量
case 3:
qs = qs.OrderBy("-Comments") // 评论量
}
qs = qs.Limit(size, 0)
_, err := qs.All(&blogs)
return blogs, err
}
func (this *BlogService) GetNLabel(id int64) ([]*models.NLabel, error) {
var labels []*models.NLabel
o := orm.NewOrm()
_, err := o.QueryTable(&models.NLabel{}).Filter("BlogId", id).All(&labels)
if err != nil {
return labels, err
}
return labels, nil
}