Enha: remove marks
This commit is contained in:
		| @@ -8,9 +8,10 @@ import ( | ||||
| ) | ||||
|  | ||||
| type CardMarksRepo interface { | ||||
| 	CardMarksByCardID(ctx context.Context, cardID string) ([]models.CardMark, error) | ||||
| 	CardMarksByCardID(ctx context.Context, cardID uint32) ([]models.CardMark, error) | ||||
| 	CardMarksAdd(ctx context.Context, cm *models.CardMark) error | ||||
| 	CardMarksRemove(ctx context.Context, cardID, username string) error | ||||
| 	CardMarksRemove(ctx context.Context, cardID uint32, username string) error | ||||
| 	CardMarksByRoomID(ctx context.Context, roomID string) ([]models.CardMark, error) | ||||
| } | ||||
|  | ||||
| type cardMarksRepo struct { | ||||
| @@ -21,19 +22,25 @@ func NewCardMarksRepo(db *sqlx.DB) CardMarksRepo { | ||||
| 	return &cardMarksRepo{db: db} | ||||
| } | ||||
|  | ||||
| func (r *cardMarksRepo) CardMarksByCardID(ctx context.Context, cardID string) ([]models.CardMark, error) { | ||||
| 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) | ||||
| 	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, mark_type) VALUES (?, ?, ?)", cm.CardID, cm.Username, cm.MarkType) | ||||
| 	_, 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, username string) error { | ||||
| 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 | ||||
| } | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 Grail Finder
					Grail Finder