package service import ( "beeblog/models" "github.com/astaxie/beego/orm" "errors" ) 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 }