query.sql.go

// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.30.0
// source: query.sql

package querytest

import (
	"context"
)

const getReferralChain = `-- name: GetReferralChain :many
WITH RECURSIVE referral_chain AS (
    SELECT user_id,
           referrer_id,
           1 AS level
    FROM user_referrals r
    WHERE r.user_id = $1

    UNION ALL

    SELECT r.user_id,
           r.referrer_id,
           rc.level + 1
    FROM user_referrals r
             JOIN referral_chain rc ON r.user_id = rc.referrer_id
    WHERE rc.level < $2
)
SELECT user_id, referrer_id, level
FROM referral_chain
`

type GetReferralChainParams struct {
	UserID int64
	Level  int32
}

type GetReferralChainRow struct {
	UserID     int64
	ReferrerID int64
	Level      int32
}

func (q *Queries) GetReferralChain(ctx context.Context, arg GetReferralChainParams) ([]GetReferralChainRow, error) {
	rows, err := q.db.Query(ctx, getReferralChain, arg.UserID, arg.Level)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []GetReferralChainRow
	for rows.Next() {
		var i GetReferralChainRow
		if err := rows.Scan(&i.UserID, &i.ReferrerID, &i.Level); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}