47 lines
1.6 KiB
Go
47 lines
1.6 KiB
Go
package repos
|
|
|
|
import (
|
|
"context"
|
|
"gralias/models"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type CardMarksRepo interface {
|
|
CardMarksByCardID(ctx context.Context, cardID uint32) ([]models.CardMark, error)
|
|
CardMarksAdd(ctx context.Context, cm *models.CardMark) error
|
|
CardMarksRemove(ctx context.Context, cardID uint32, username string) error
|
|
CardMarksByRoomID(ctx context.Context, roomID string) ([]models.CardMark, error)
|
|
}
|
|
|
|
type cardMarksRepo struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func NewCardMarksRepo(db *sqlx.DB) CardMarksRepo {
|
|
return &cardMarksRepo{db: db}
|
|
}
|
|
|
|
func (r *cardMarksRepo) CardMarksByCardID(ctx context.Context, cardID uint32) ([]models.CardMark, error) {
|
|
var cardMarks []models.CardMark
|
|
err := sqlx.SelectContext(ctx, getDB(ctx, r.db), &cardMarks, "SELECT * FROM card_marks WHERE card_id = ?", cardID)
|
|
return cardMarks, err
|
|
}
|
|
|
|
func (r *cardMarksRepo) CardMarksAdd(ctx context.Context, cm *models.CardMark) error {
|
|
_, err := getDB(ctx, r.db).ExecContext(ctx, "INSERT INTO card_marks (card_id, username) VALUES (?, ?)", cm.CardID, cm.Username)
|
|
return err
|
|
}
|
|
|
|
func (r *cardMarksRepo) CardMarksRemove(ctx context.Context, cardID uint32, username string) error {
|
|
db := getDB(ctx, r.db)
|
|
_, err := db.ExecContext(ctx, "DELETE FROM card_marks WHERE card_id = ? AND username = ?", cardID, username)
|
|
return err
|
|
}
|
|
|
|
func (r *cardMarksRepo) CardMarksByRoomID(ctx context.Context, roomID string) ([]models.CardMark, error) {
|
|
var cardMarks []models.CardMark
|
|
err := sqlx.SelectContext(ctx, getDB(ctx, r.db), &cardMarks, "SELECT * FROM card_marks WHERE card_id IN (select id from word_cards where room_id = ?)", roomID)
|
|
return cardMarks, err
|
|
}
|