You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.2 KiB
105 lines
2.2 KiB
package service
|
|
|
|
import (
|
|
"beeblog/models"
|
|
"errors"
|
|
"github.com/beego/beego/v2/client/orm"
|
|
)
|
|
|
|
type NoteService struct {
|
|
}
|
|
|
|
func (this *NoteService) EditNote(note *models.Note) error {
|
|
o := orm.NewOrm()
|
|
id, err := o.Update(note)
|
|
if err == nil {
|
|
note.Id = id
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (this *NoteService) SaveNote(note *models.Note) error {
|
|
o := orm.NewOrm()
|
|
id, err := o.Insert(note)
|
|
if err == nil {
|
|
note.Id = id
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (this *NoteService) GetNote(note *models.Note) error {
|
|
o := orm.NewOrm()
|
|
return o.Read(note)
|
|
}
|
|
|
|
func (this *NoteService) DelNote(note *models.Note) error {
|
|
o := orm.NewOrm()
|
|
_, err := o.Delete(note)
|
|
return err
|
|
}
|
|
|
|
func (this *NoteService) GetNoteByPid(pid int64) ([]*models.Note, error) {
|
|
var notes []*models.Note
|
|
o := orm.NewOrm()
|
|
qs := o.QueryTable(models.Note{})
|
|
_, err := qs.Filter("Pid", pid).All(¬es)
|
|
return notes, err
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func (this *NoteService) SaveNoteColl(note *models.NoteColl) error {
|
|
o := orm.NewOrm()
|
|
id, err := o.Insert(note)
|
|
if err == nil {
|
|
note.Id = id
|
|
}
|
|
return err
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
func (this *NoteService) GetNoteColl(uid int64) ([]*models.NoteColl, error) {
|
|
var notes []*models.NoteColl
|
|
o := orm.NewOrm()
|
|
qs := o.QueryTable(models.NoteColl{})
|
|
_, err := qs.Filter("UserId", uid).All(¬es)
|
|
return notes, err
|
|
}
|
|
|
|
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
|
|
}
|
|
|