architectv / estate-task

import (

	"fmt"

	"github.com/architectv/estate-task/pkg/model"

	"github.com/jmoiron/sqlx"

)

type RoomPostgres struct {

	db *sqlx.DB

}

func (r *RoomPostgres) Create(room *model.Room) (int, error) {

	var id int

	query := fmt.Sprintf(

		`INSERT INTO %s (description, price) VALUES ($1, $2) RETURNING id`,

		roomsTable)

	row := r.db.QueryRow(query, room.Description, room.Price)

	if err := row.Scan(&id); err != nil {

		return 0, err

	}

	return id, nil

}

func (r *RoomPostgres) Delete(id int) error {

	query := fmt.Sprintf("DELETE FROM %s WHERE id=$1", roomsTable)

	_, err := r.db.Exec(query, id)

	return err

}

func (r *RoomPostgres) GetAll(sortField string, desc bool) ([]*model.Room, error) {

	var rooms []*model.Room

	query := fmt.Sprintf("SELECT * FROM %s ORDER BY %s", roomsTable, sortField)

	if desc {

		query += " DESC"

	}

	err := r.db.Select(&rooms, query)

	return rooms, err

}

func (r *RoomPostgres) GetById(id int) (*model.Room, error) {

	room := &model.Room{}

	query := fmt.Sprintf("SELECT * FROM %s WHERE id=$1", roomsTable)

	err := r.db.Get(room, query, id)

	return room, err

}