ai-gateway/model/channel.go
2023-04-22 22:02:59 +08:00

51 lines
1.2 KiB
Go

package model
import (
_ "gorm.io/driver/sqlite"
)
type Channel struct {
Id int `json:"id"`
Type int `json:"type" gorm:"default:0"`
Key string `json:"key"`
Status int `json:"status" gorm:"default:1"`
}
func GetAllChannels(startIdx int, num int) ([]*Channel, error) {
var channels []*Channel
var err error
err = DB.Order("id desc").Limit(num).Offset(startIdx).Find(&channels).Error
return channels, err
}
func SearchChannels(keyword string) (channels []*Channel, err error) {
err = DB.Select([]string{"id", "key"}, keyword, keyword).Find(&channels).Error
return channels, err
}
func GetChannelById(id int) (*Channel, error) {
channel := Channel{Id: id}
var err error = nil
err = DB.Select([]string{"id", "type"}).First(&channel, "id = ?", id).Error
return &channel, err
}
func (channel *Channel) Insert() error {
var err error
err = DB.Create(channel).Error
return err
}
func (channel *Channel) Update() error {
var err error
err = DB.Model(channel).Updates(channel).Error
return err
}
// Delete Make sure link is valid! Because we will use os.Remove to delete it!
func (channel *Channel) Delete() error {
var err error
err = DB.Delete(channel).Error
return err
}