go_process_manager/internal/app/repository/ws_share.go

package repository

import (
	"github.com/lzh-1625/go_process_manager/internal/app/model"
	"github.com/lzh-1625/go_process_manager/internal/app/repository/query"
)

type wsShare struct{}

var WsShare = new(wsShare)

func (p *wsShare) GetWsShareDataByToken(token string) (data *model.WsShare, err error) {
	data, err = query.WsShare.Where(query.WsShare.Token.Eq(token)).First()
	return
}

func (p *wsShare) AddShareData(data model.WsShare) error {
	return db.Save(&data).Error
}

func (p *wsShare) GetWsShareList() (data []*model.WsShare) {
	ws := query.WsShare
	data, _ = ws.Find()
	return
}

func (p *wsShare) Delete(id int) error {
	ws := query.WsShare
	_, err := ws.Where(ws.Id.Eq(id)).Delete()
	return err
}

func (p *wsShare) Edit(data *model.WsShare) error {
	ws := query.WsShare
	_, err := ws.Where(ws.Id.Eq(int(data.ID))).Updates(data)
	return err
}