rustdesk-api/service/peer.go

package service

import (
	"Gwen/global"
	"Gwen/model"
	"gorm.io/gorm"
)

type PeerService struct {
}

// FindById 根据id查找
func (ps *PeerService) FindById(id string) *model.Peer {
	p := &model.Peer{}
	global.DB.Where("id = ?", id).First(p)
	return p
}
func (ps *PeerService) InfoByRowId(id uint) *model.Peer {
	p := &model.Peer{}
	global.DB.Where("row_id = ?", id).First(p)
	return p
}

//// ListByUserIds 根据用户id取列表
//func (ps *PeerService) ListByUserIds(userIds []uint, page, pageSize uint) (res *model.PeerList) {
//	res = &model.PeerList{}
//	res.Page = int64(page)
//	res.PageSize = int64(pageSize)
//	tx := global.DB.Model(&model.Peer{}).Preload("User")
//	tx.Where("user_id in (?)", userIds)
//	tx.Count(&res.Total)
//	tx.Scopes(Paginate(page, pageSize))
//	tx.Find(&res.Peers)
//	return
//}

func (ps *PeerService) List(page, pageSize uint, where func(tx *gorm.DB)) (res *model.PeerList) {
	res = &model.PeerList{}
	res.Page = int64(page)
	res.PageSize = int64(pageSize)
	tx := global.DB.Model(&model.Peer{})
	if where != nil {
		where(tx)
	}
	tx.Count(&res.Total)
	tx.Scopes(Paginate(page, pageSize))
	tx.Find(&res.Peers)
	return
}

// Create 创建
func (ps *PeerService) Create(u *model.Peer) error {
	res := global.DB.Create(u).Error
	return res
}
func (ps *PeerService) Delete(u *model.Peer) error {
	return global.DB.Delete(u).Error
}

// Update 更新
func (ps *PeerService) Update(u *model.Peer) error {
	return global.DB.Model(u).Updates(u).Error
}