mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-07-08 07:31:26 +08:00
feat: init commit
This commit is contained in:
55
dao/file.go
Normal file
55
dao/file.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package dao
|
||||
|
||||
import "github.com/krau/SaveAny-Bot/model"
|
||||
|
||||
func AddReceivedFile(receivedFile *model.ReceivedFile) error {
|
||||
return db.Create(receivedFile).Error
|
||||
}
|
||||
|
||||
func GetReceivedFileByFileID(fileID string) (*model.ReceivedFile, error) {
|
||||
var receivedFile model.ReceivedFile
|
||||
err := db.Where("file_id = ?", fileID).First(&receivedFile).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &receivedFile, nil
|
||||
}
|
||||
|
||||
func GetReceivedFileByFileUniqueID(fileUniqueID string) (*model.ReceivedFile, error) {
|
||||
var receivedFile model.ReceivedFile
|
||||
err := db.Where("file_unique_id = ?", fileUniqueID).First(&receivedFile).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &receivedFile, nil
|
||||
}
|
||||
|
||||
func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.ReceivedFile, error) {
|
||||
var receivedFile model.ReceivedFile
|
||||
err := db.Where("chat_id = ? AND message_id = ?", chatID, messageID).First(&receivedFile).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &receivedFile, nil
|
||||
}
|
||||
|
||||
func GetReceivedFilesByMediaGroupID(mediaGroupID string) ([]model.ReceivedFile, error) {
|
||||
var receivedFiles []model.ReceivedFile
|
||||
err := db.Where("media_group_id = ?", mediaGroupID).Find(&receivedFiles).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return receivedFiles, nil
|
||||
}
|
||||
|
||||
func UpdateReceivedFile(receivedFile *model.ReceivedFile) error {
|
||||
return db.Save(receivedFile).Error
|
||||
}
|
||||
|
||||
func DeleteReceivedFileByFileID(fileID string) error {
|
||||
return db.Where("file_id = ?", fileID).Delete(&model.ReceivedFile{}).Error
|
||||
}
|
||||
|
||||
func DeleteReceivedFileByFileUniqueID(fileUniqueID string) error {
|
||||
return db.Where("file_unique_id = ?", fileUniqueID).Delete(&model.ReceivedFile{}).Error
|
||||
}
|
||||
Reference in New Issue
Block a user