码农笔录博客源码
Du kan inte välja fler än 25 ämnen Ämnen måste starta med en bokstav eller siffra, kan innehålla bindestreck ('-') och vara max 35 tecken långa.

106 rader
2.2 KiB

6 år sedan
package service
import (
"beeblog/models"
"github.com/astaxie/beego/orm"
"errors"
6 år sedan
)
type NoteService struct {
}
6 år sedan
func (this *NoteService) EditNote(note *models.Note) error {
o := orm.NewOrm()
id, err := o.Update(note)
if err == nil {
note.Id = id
}
return err
}
6 år sedan
func (this *NoteService) SaveNote(note *models.Note) error {
6 år sedan
o := orm.NewOrm()
id, err := o.Insert(note)
if err == nil {
note.Id = id
}
return err
}
6 år sedan
func (this *NoteService) GetNote(note *models.Note) error {
6 år sedan
o := orm.NewOrm()
return o.Read(note)
}
6 år sedan
func (this *NoteService) DelNote(note *models.Note) error {
o := orm.NewOrm()
_, err := o.Delete(note)
return err
}
6 år sedan
func (this *NoteService) GetNoteByPid(pid int64) ([]*models.Note, error) {
6 år sedan
var notes []*models.Note
o := orm.NewOrm()
qs := o.QueryTable(models.Note{})
6 år sedan
_, err := qs.Filter("Pid", pid).All(&notes)
return notes, err
6 år sedan
}
6 år sedan
func (this *NoteService) CountNote(pid int64) (int64, error) {
o := orm.NewOrm()
totalCount, err := o.QueryTable(&models.Note{}).Filter("Pid", pid).Count()
if err != nil {
return 0, err
}
return totalCount, nil
}
6 år sedan
func (this *NoteService) SaveNoteColl(note *models.NoteColl) error {
6 år sedan
o := orm.NewOrm()
id, err := o.Insert(note)
if err == nil {
note.Id = id
}
return err
}
6 år sedan
func (this *NoteService) EditNoteColl(title string, id int64, uid int64) error {
o := orm.NewOrm()
noteColl := &models.NoteColl{Id: id}
if err := o.Read(noteColl); err != nil {
return err
}
if noteColl.UserId != uid {
return errors.New("403")
}
noteColl.Title = title
_, err := o.Update(noteColl, "Title")
return err
}
6 år sedan
func (this *NoteService) GetNoteColl(uid int64) ([]*models.NoteColl, error) {
6 år sedan
var notes []*models.NoteColl
o := orm.NewOrm()
qs := o.QueryTable(models.NoteColl{})
6 år sedan
_, err := qs.Filter("UserId", uid).All(&notes)
return notes, err
}
6 år sedan
func (this *NoteService) DelNoteColl(id int64, uid int64) error {
o := orm.NewOrm()
noteColl := &models.NoteColl{Id: id}
if err := o.Read(noteColl); err != nil {
return err
}
if uid != noteColl.UserId {
return errors.New("403")
}
_, err := o.QueryTable(models.Note{}).Filter("Pid", id).Delete()
if err == nil {
_, err = o.Delete(noteColl)
}
return err
}