// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.29.0
// source: numeric_entitlements.sql

package entitlements

import (
	"context"
)

const createNumericEntitlement = `-- name: CreateNumericEntitlement :one
INSERT INTO entitlements.numeric_entitlements (pool_id, resource_key, entitlement_type, resource_limit)
VALUES ($1, $2, $3, $4)
RETURNING entitlement_id, pool_id, resource_key, entitlement_type, resource_limit, reset_period, created_at, updated_at
`

type CreateNumericEntitlementParams struct {
	PoolID          string `json:"pool_id"`
	ResourceKey     string `json:"resource_key"`
	EntitlementType string `json:"entitlement_type"`
	ResourceLimit   int64  `json:"resource_limit"`
}

func (q *Queries) CreateNumericEntitlement(ctx context.Context, arg CreateNumericEntitlementParams) (NumericEntitlement, error) {
	row := q.db.QueryRowContext(ctx, createNumericEntitlement,
		arg.PoolID,
		arg.ResourceKey,
		arg.EntitlementType,
		arg.ResourceLimit,
	)
	var i NumericEntitlement
	err := row.Scan(
		&i.EntitlementID,
		&i.PoolID,
		&i.ResourceKey,
		&i.EntitlementType,
		&i.ResourceLimit,
		&i.ResetPeriod,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const getNumericEntitlementByPoolAndResource = `-- name: GetNumericEntitlementByPoolAndResource :one
SELECT entitlement_id, pool_id, resource_key, entitlement_type, resource_limit, reset_period, created_at, updated_at FROM entitlements.numeric_entitlements
WHERE pool_id = $1 AND resource_key = $2
`

type GetNumericEntitlementByPoolAndResourceParams struct {
	PoolID      string `json:"pool_id"`
	ResourceKey string `json:"resource_key"`
}

func (q *Queries) GetNumericEntitlementByPoolAndResource(ctx context.Context, arg GetNumericEntitlementByPoolAndResourceParams) (NumericEntitlement, error) {
	row := q.db.QueryRowContext(ctx, getNumericEntitlementByPoolAndResource, arg.PoolID, arg.ResourceKey)
	var i NumericEntitlement
	err := row.Scan(
		&i.EntitlementID,
		&i.PoolID,
		&i.ResourceKey,
		&i.EntitlementType,
		&i.ResourceLimit,
		&i.ResetPeriod,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const listNumericEntitlementsByPoolID = `-- name: ListNumericEntitlementsByPoolID :many
SELECT entitlement_id, pool_id, resource_key, entitlement_type, resource_limit, reset_period, created_at, updated_at FROM entitlements.numeric_entitlements
WHERE pool_id = $1
ORDER BY resource_key ASC
`

func (q *Queries) ListNumericEntitlementsByPoolID(ctx context.Context, poolID string) ([]NumericEntitlement, error) {
	rows, err := q.db.QueryContext(ctx, listNumericEntitlementsByPoolID, poolID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []NumericEntitlement{}
	for rows.Next() {
		var i NumericEntitlement
		if err := rows.Scan(
			&i.EntitlementID,
			&i.PoolID,
			&i.ResourceKey,
			&i.EntitlementType,
			&i.ResourceLimit,
			&i.ResetPeriod,
			&i.CreatedAt,
			&i.UpdatedAt,
		); err != nil {
			return nil, err
		}
		items = append(items, i)
	}
	if err := rows.Close(); err != nil {
		return nil, err
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const updateNumericEntitlementLimit = `-- name: UpdateNumericEntitlementLimit :one
UPDATE entitlements.numeric_entitlements
SET resource_limit = $2
WHERE entitlement_id = $1
RETURNING entitlement_id, pool_id, resource_key, entitlement_type, resource_limit, reset_period, created_at, updated_at
`

type UpdateNumericEntitlementLimitParams struct {
	EntitlementID string `json:"entitlement_id"`
	ResourceLimit int64  `json:"resource_limit"`
}

func (q *Queries) UpdateNumericEntitlementLimit(ctx context.Context, arg UpdateNumericEntitlementLimitParams) (NumericEntitlement, error) {
	row := q.db.QueryRowContext(ctx, updateNumericEntitlementLimit, arg.EntitlementID, arg.ResourceLimit)
	var i NumericEntitlement
	err := row.Scan(
		&i.EntitlementID,
		&i.PoolID,
		&i.ResourceKey,
		&i.EntitlementType,
		&i.ResourceLimit,
		&i.ResetPeriod,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}