How to Use Embed with UPDATE query?

[billy-le]

I'm fairly new to Postgres and sqlc.

I understand the example in the Docs for a SELECT query but
I have a query where I want to UPDATE my "bookings" table and then JOIN my "invoices" table

I've gotten close but this is my current query:

-- name: UpdateBooking :one
WITH booking_with_invoice AS (
    SELECT sqlc.embed(bookings), sqlc.embed(invoices)
    FROM bookings
    JOIN invoices ON invoices.booking_id = bookings.id
)
UPDATE booking_with_invoice
SET
    start_date = COALESCE(sqlc.narg(start_date), start_date),
    end_date = COALESCE(sqlc.narg(end_date), end_date),
    status = COALESCE(sqlc.narg(status), status)
WHERE bookings.id = @id
RETURNING *;

And my generated code:

type UpdateBookingRow struct {
    Bookings interface{} `json:"bookings"`
    Invoices interface{} `json:"invoices"`
}

func (q *Queries) UpdateBooking(ctx context.Context, arg UpdateBookingParams) (UpdateBookingRow, error) {
    row := q.db.QueryRow(ctx, updateBooking,
        arg.StartDate,
        arg.EndDate,
        arg.Status,
        arg.ID,
    )
    var i UpdateBookingRow
    err := row.Scan(&i.Bookings, &i.Invoices)
    return i, err
}

As you can see, the UpdateBookingRow struct shows interface{} for both fields but I want the Booking and Invoice struct.