Continue on ynab import

postgres/payees.sql.go

@@ -11,13 +11,18 @@ import (
 
 const createPayee = `-- name: CreatePayee :one
 INSERT INTO payees
-(name)
-VALUES ($1)
+(name, budget_id)
+VALUES ($1, $2)
 RETURNING id, budget_id, name
 `
 
-func (q *Queries) CreatePayee(ctx context.Context, name string) (Payee, error) {
-	row := q.db.QueryRow(ctx, createPayee, name)
+type CreatePayeeParams struct {
+	Name     string
+	BudgetID uuid.UUID
+}
+
+func (q *Queries) CreatePayee(ctx context.Context, arg CreatePayeeParams) (Payee, error) {
+	row := q.db.QueryRow(ctx, createPayee, arg.Name, arg.BudgetID)
 	var i Payee
 	err := row.Scan(&i.ID, &i.BudgetID, &i.Name)
 	return i, err