35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package storagebonus
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/ente-io/museum/ente/storagebonus"
|
|
)
|
|
|
|
func (r *Repository) InsertBFBonus(ctx context.Context, userID int64, validTill int64, storage int64) error {
|
|
bonusID := fmt.Sprintf("%s-%d", storagebonus.AddOnBf2023, userID)
|
|
_, err := r.DB.ExecContext(ctx, "INSERT INTO storage_bonus (bonus_id, user_id, storage, type, valid_till) VALUES ($1, $2, $3, $4, $5)", bonusID, userID, storage, storagebonus.AddOnBf2023, validTill)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *Repository) RemoveBFBonus(ctx context.Context, userID int64) (int64, error) {
|
|
bonusID := fmt.Sprintf("%s-%d", storagebonus.AddOnBf2023, userID)
|
|
res, err := r.DB.ExecContext(ctx, "DELETE FROM storage_bonus WHERE bonus_id = $1", bonusID)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return res.RowsAffected()
|
|
}
|
|
|
|
func (r *Repository) UpdateBFBonus(ctx context.Context, userID int64, validTill int64, storage int64) error {
|
|
bonusID := fmt.Sprintf("%s-%d", storagebonus.AddOnBf2023, userID)
|
|
_, err := r.DB.ExecContext(ctx, "UPDATE storage_bonus SET storage = $1, valid_till = $2 WHERE bonus_id = $3", storage, validTill, bonusID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|