107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
const createMenuItems = `-- name: CreateMenuItems :exec
|
|
INSERT INTO menu_items(
|
|
location_id,
|
|
name,
|
|
price,
|
|
category,
|
|
description,
|
|
is_available,
|
|
submitted_by
|
|
) values (
|
|
$1, $2, $3, $4, $5, $6, $7
|
|
)
|
|
RETURNING id
|
|
`
|
|
|
|
type CreateMenuItemsParams struct {
|
|
LocationID int32 `json:"location_id"`
|
|
Name string `json:"name"`
|
|
Price int32 `json:"price"`
|
|
Category string `json:"category"`
|
|
Description string `json:"description"`
|
|
IsAvailable bool `json:"is_available"`
|
|
SubmittedBy int32 `json:"submitted_by"`
|
|
}
|
|
|
|
const getMenuItems = `-- name: GetMenuItems :many
|
|
SELECT
|
|
m.id,
|
|
m.location_id,
|
|
m.name,
|
|
m.price,
|
|
m.category,
|
|
m.description,
|
|
m.is_available,
|
|
m.submitted_by,
|
|
ROUND(AVG(r.score::float8))::float8 AS avg_score
|
|
FROM menu_items m
|
|
LEFT JOIN menu_item_reviews r ON r.menu_item_id = m.id AND r.is_hidden = false
|
|
WHERE ($1::int4 IS NULL OR m.location_id = $1::int4)
|
|
GROUP BY m.id
|
|
ORDER BY m.id ASC
|
|
`
|
|
|
|
type GetMenuItemsRow struct {
|
|
ID int32 `json:"id"`
|
|
LocationID int32 `json:"location_id"`
|
|
Name string `json:"name"`
|
|
Price int32 `json:"price"`
|
|
Category string `json:"category"`
|
|
Description string `json:"description"`
|
|
IsAvailable bool `json:"is_available"`
|
|
SubmittedBy int32 `json:"submitted_by"`
|
|
AvgScore *float64 `json:"avg_score"`
|
|
}
|
|
|
|
func (q *Queries) GetMenuItems(ctx context.Context, locationID pgtype.Int4) ([]GetMenuItemsRow, error) {
|
|
rows, err := q.db.Query(ctx, getMenuItems, locationID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var items []GetMenuItemsRow
|
|
for rows.Next() {
|
|
var i GetMenuItemsRow
|
|
if err := rows.Scan(
|
|
&i.ID,
|
|
&i.LocationID,
|
|
&i.Name,
|
|
&i.Price,
|
|
&i.Category,
|
|
&i.Description,
|
|
&i.IsAvailable,
|
|
&i.SubmittedBy,
|
|
&i.AvgScore,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
items = append(items, i)
|
|
}
|
|
return items, rows.Err()
|
|
}
|
|
|
|
func (q *Queries) CreateMenuItems(ctx context.Context, arg CreateMenuItemsParams) (int32, error) {
|
|
row := q.db.QueryRow(ctx, createMenuItems,
|
|
arg.LocationID,
|
|
arg.Name,
|
|
arg.Price,
|
|
arg.Category,
|
|
arg.Description,
|
|
arg.IsAvailable,
|
|
arg.SubmittedBy,
|
|
)
|
|
|
|
var i int32
|
|
err := row.Scan(&i)
|
|
return i, err
|
|
}
|