beeblog/service/NoteService.go

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(&notes)
	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(&notes)
	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
}