41 lines
956 B
Go
41 lines
956 B
Go
package service
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"x-ui/database"
|
|
"x-ui/database/model"
|
|
)
|
|
|
|
type InboundService struct {
|
|
}
|
|
|
|
func (s *InboundService) GetInbounds(userId int) ([]*model.Inbound, error) {
|
|
db := database.GetDB()
|
|
var inbounds []*model.Inbound
|
|
err := db.Model(model.Inbound{}).Where("user_id = ?", userId).Find(&inbounds).Error
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
|
return nil, err
|
|
}
|
|
return inbounds, nil
|
|
}
|
|
|
|
func (s *InboundService) GetAllInbounds() ([]*model.Inbound, error) {
|
|
db := database.GetDB()
|
|
var inbounds []*model.Inbound
|
|
err := db.Model(model.Inbound{}).Find(&inbounds).Error
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
|
return nil, err
|
|
}
|
|
return inbounds, nil
|
|
}
|
|
|
|
func (s *InboundService) AddInbound(inbound *model.Inbound) error {
|
|
db := database.GetDB()
|
|
return db.Save(inbound).Error
|
|
}
|
|
|
|
func (s *InboundService) DelInbound(id int) error {
|
|
db := database.GetDB()
|
|
return db.Delete(model.Inbound{}, id).Error
|
|
}
|