Replace the entity slugs on organizations, workspaces, resource pools, and plan ladders with nullable `key` columns and add keys to products, prices, and entitlement sets. Rename `providers.slug` to `provider` and add partial unique indexes for system and org role names. Assign invoice numbers per billing account from a gapless transactional counter; Stripe's number moves to the invoice mapping as an external reference. Seeds, fixtures, and the operator lookup address rows by key, and the returning-login resync no longer blanks a display name when the IdP sends no `name` claim.
234 lines
6.4 KiB
Go
234 lines
6.4 KiB
Go
// Code generated by sqlc. DO NOT EDIT.
|
|
// versions:
|
|
// sqlc v1.29.0
|
|
// source: resource_pools.sql
|
|
|
|
package entitlements
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
)
|
|
|
|
const createResourcePool = `-- name: CreateResourcePool :one
|
|
INSERT INTO core.resource_pools (org_id, name, pool_type, is_auto_managed, key)
|
|
VALUES ($1, $2, $3, $4, $5)
|
|
RETURNING pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at
|
|
`
|
|
|
|
type CreateResourcePoolParams struct {
|
|
OrgID string `json:"org_id"`
|
|
Name string `json:"name"`
|
|
PoolType string `json:"pool_type"`
|
|
IsAutoManaged bool `json:"is_auto_managed"`
|
|
Key sql.NullString `json:"key"`
|
|
}
|
|
|
|
// `key` is the optional declarative address (entity-keys §3), unique within
|
|
// the organization because a pool is a child of one. The console writes it
|
|
// for exactly one row it creates by design: every organization's default
|
|
// pool, keyed `default` (§4). Any other pool leaves it NULL unless a seed or
|
|
// an API client names one.
|
|
func (q *Queries) CreateResourcePool(ctx context.Context, arg CreateResourcePoolParams) (ResourcePool, error) {
|
|
row := q.db.QueryRowContext(ctx, createResourcePool,
|
|
arg.OrgID,
|
|
arg.Name,
|
|
arg.PoolType,
|
|
arg.IsAutoManaged,
|
|
arg.Key,
|
|
)
|
|
var i ResourcePool
|
|
err := row.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&i.CreatedAt,
|
|
&i.UpdatedAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const getDefaultPoolByOrgID = `-- name: GetDefaultPoolByOrgID :one
|
|
SELECT pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at FROM core.resource_pools
|
|
WHERE org_id = $1 AND pool_type = 'default' AND status = 'active'
|
|
LIMIT 1
|
|
`
|
|
|
|
func (q *Queries) GetDefaultPoolByOrgID(ctx context.Context, orgID string) (ResourcePool, error) {
|
|
row := q.db.QueryRowContext(ctx, getDefaultPoolByOrgID, orgID)
|
|
var i ResourcePool
|
|
err := row.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&i.CreatedAt,
|
|
&i.UpdatedAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const getResourcePoolByID = `-- name: GetResourcePoolByID :one
|
|
SELECT pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at FROM core.resource_pools
|
|
WHERE pool_id = $1
|
|
`
|
|
|
|
func (q *Queries) GetResourcePoolByID(ctx context.Context, poolID string) (ResourcePool, error) {
|
|
row := q.db.QueryRowContext(ctx, getResourcePoolByID, poolID)
|
|
var i ResourcePool
|
|
err := row.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&i.CreatedAt,
|
|
&i.UpdatedAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const getResourcePoolByOrgAndKey = `-- name: GetResourcePoolByOrgAndKey :one
|
|
SELECT pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at FROM core.resource_pools
|
|
WHERE org_id = $1 AND key = $2
|
|
`
|
|
|
|
type GetResourcePoolByOrgAndKeyParams struct {
|
|
OrgID string `json:"org_id"`
|
|
Key sql.NullString `json:"key"`
|
|
}
|
|
|
|
// Child key resolver (entity-keys §5). uq_resource_pools_org_id_key
|
|
// guarantees at most one row matches; `default` resolves the pool the
|
|
// console creates for every organization.
|
|
func (q *Queries) GetResourcePoolByOrgAndKey(ctx context.Context, arg GetResourcePoolByOrgAndKeyParams) (ResourcePool, error) {
|
|
row := q.db.QueryRowContext(ctx, getResourcePoolByOrgAndKey, arg.OrgID, arg.Key)
|
|
var i ResourcePool
|
|
err := row.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&i.CreatedAt,
|
|
&i.UpdatedAt,
|
|
)
|
|
return i, err
|
|
}
|
|
|
|
const getResourcePoolsByOrgID = `-- name: GetResourcePoolsByOrgID :many
|
|
SELECT pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at FROM core.resource_pools
|
|
WHERE org_id = $1 AND status = 'active'
|
|
ORDER BY pool_type ASC, name ASC
|
|
`
|
|
|
|
func (q *Queries) GetResourcePoolsByOrgID(ctx context.Context, orgID string) ([]ResourcePool, error) {
|
|
rows, err := q.db.QueryContext(ctx, getResourcePoolsByOrgID, orgID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
items := []ResourcePool{}
|
|
for rows.Next() {
|
|
var i ResourcePool
|
|
if err := rows.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&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 listResourcePoolsByOrgIDAnyStatus = `-- name: ListResourcePoolsByOrgIDAnyStatus :many
|
|
SELECT pool_id, org_id, name, key, pool_type, is_auto_managed, description, status, suspended_at, archived_at, created_at, updated_at FROM core.resource_pools
|
|
WHERE org_id = $1
|
|
ORDER BY pool_type ASC, name ASC
|
|
`
|
|
|
|
// Every pool row for the organization regardless of status -- unlike
|
|
// GetResourcePoolsByOrgID's active-only filter, this does not hide a
|
|
// suspended or archived pool from view. The org-detail composite uses
|
|
// this for its pools panel so an operator sees a pool's actual status
|
|
// (ux-honest-surfaces requirement: "pool status and usage are visible on
|
|
// the organization view") instead of the pool silently vanishing from the
|
|
// list, and so "this org has zero pool rows" (the pool-less breakage
|
|
// case) is distinguishable from "this org's only pool exists but is not
|
|
// active".
|
|
func (q *Queries) ListResourcePoolsByOrgIDAnyStatus(ctx context.Context, orgID string) ([]ResourcePool, error) {
|
|
rows, err := q.db.QueryContext(ctx, listResourcePoolsByOrgIDAnyStatus, orgID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
items := []ResourcePool{}
|
|
for rows.Next() {
|
|
var i ResourcePool
|
|
if err := rows.Scan(
|
|
&i.PoolID,
|
|
&i.OrgID,
|
|
&i.Name,
|
|
&i.Key,
|
|
&i.PoolType,
|
|
&i.IsAutoManaged,
|
|
&i.Description,
|
|
&i.Status,
|
|
&i.SuspendedAt,
|
|
&i.ArchivedAt,
|
|
&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
|
|
}
|