bagas/filekeeper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add file download count tracking

Browse Source
This commit is contained in:
Bagas Aulia Rezki
2024-09-15 17:58:13 +07:00
parent 4ee86d0fc4
commit db79d77d13
3 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
db/database.go
View File

@ -261,6 +261,20 @@ func (db *mySQLdb) GetFiles(ownerID string) ([]*models.File, error) {
return files, err
}
func (db *mySQLdb) IncrementDownloadCount(fileID string) error {
var file models.File
err := db.DB.Table("files").Where("id = ?", fileID).First(&file).Error
if err != nil {
return err
}
file.Downloaded = file.Downloaded + 1
err = db.DB.Updates(file).Error
if err != nil {
return err
}
return nil
}
func (db *mySQLdb) InitializeTotp(email string, secret string) error {
var user models.User
err := db.DB.Table("users").Where("email = ?", email).First(&user).Error
@ -398,6 +412,20 @@ func (db *postgresDB) GetFiles(ownerID string) ([]*models.File, error) {
return files, err
}
func (db *postgresDB) IncrementDownloadCount(fileID string) error {
var file models.File
err := db.DB.Table("files").Where("id = $1", fileID).First(&file).Error
if err != nil {
return err
}
file.Downloaded = file.Downloaded + 1
err = db.DB.Updates(file).Error
if err != nil {
return err
}
return nil
}
func (db *postgresDB) InitializeTotp(email string, secret string) error {
var user models.User
err := db.DB.Table("users").Where("email = $1", email).First(&user).Error