feat: implement file deletion

This commit is contained in:
2024-09-17 11:21:04 +07:00
parent e07360d2b3
commit cd9a7503b5
6 changed files with 355 additions and 137 deletions

View File

@ -243,6 +243,14 @@ func (db *mySQLdb) GetFile(fileID string) (*models.File, error) {
return &file, nil
}
func (db *mySQLdb) DeleteFile(fileID string) error {
err := db.DB.Table("files").Where("id = ?", fileID).Delete(&models.File{}).Error
if err != nil {
return err
}
return nil
}
func (db *mySQLdb) GetUserFile(name string, ownerID string) (*models.File, error) {
var file models.File
err := db.DB.Table("files").Where("name = ? AND owner_id = ?", name, ownerID).First(&file).Error
@ -403,6 +411,14 @@ func (db *postgresDB) GetFile(fileID string) (*models.File, error) {
return &file, nil
}
func (db *postgresDB) DeleteFile(fileID string) error {
err := db.DB.Table("files").Where("id = $1", fileID).Delete(&models.File{}).Error
if err != nil {
return err
}
return nil
}
func (db *postgresDB) GetUserFile(name string, ownerID string) (*models.File, error) {
var file models.File
err := db.DB.Table("files").Where("name = $1 AND owner_id = $2", name, ownerID).First(&file).Error