mantrae/internal/store/db/udp_services.sql.go

// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.28.0
// source: udp_services.sql

package db

import (
	"context"

	"github.com/traefik/traefik/v3/pkg/config/dynamic"
)

const countUdpServices = `-- name: CountUdpServices :one
SELECT
  COUNT(*)
FROM
  udp_services
`

func (q *Queries) CountUdpServices(ctx context.Context) (int64, error) {
	row := q.queryRow(ctx, q.countUdpServicesStmt, countUdpServices)
	var count int64
	err := row.Scan(&count)
	return count, err
}

const createUdpService = `-- name: CreateUdpService :one
INSERT INTO
  udp_services (
    profile_id,
    agent_id,
    name,
    config,
    created_at,
    updated_at
  )
VALUES
  (?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP) RETURNING id, profile_id, agent_id, name, config, created_at, updated_at
`

type CreateUdpServiceParams struct {
	ProfileID int64               `json:"profileId"`
	AgentID   *string             `json:"agentId"`
	Name      string              `json:"name"`
	Config    *dynamic.UDPService `json:"config"`
}

func (q *Queries) CreateUdpService(ctx context.Context, arg CreateUdpServiceParams) (UdpService, error) {
	row := q.queryRow(ctx, q.createUdpServiceStmt, createUdpService,
		arg.ProfileID,
		arg.AgentID,
		arg.Name,
		arg.Config,
	)
	var i UdpService
	err := row.Scan(
		&i.ID,
		&i.ProfileID,
		&i.AgentID,
		&i.Name,
		&i.Config,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const deleteUdpService = `-- name: DeleteUdpService :exec
DELETE FROM udp_services
WHERE
  id = ?
`

func (q *Queries) DeleteUdpService(ctx context.Context, id int64) error {
	_, err := q.exec(ctx, q.deleteUdpServiceStmt, deleteUdpService, id)
	return err
}

const getUdpService = `-- name: GetUdpService :one
SELECT
  id, profile_id, agent_id, name, config, created_at, updated_at
FROM
  udp_services
WHERE
  id = ?
`

func (q *Queries) GetUdpService(ctx context.Context, id int64) (UdpService, error) {
	row := q.queryRow(ctx, q.getUdpServiceStmt, getUdpService, id)
	var i UdpService
	err := row.Scan(
		&i.ID,
		&i.ProfileID,
		&i.AgentID,
		&i.Name,
		&i.Config,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}

const getUdpServicesByProfile = `-- name: GetUdpServicesByProfile :many
SELECT
  id, profile_id, agent_id, name, config, created_at, updated_at
FROM
  udp_services
WHERE
  profile_id = ?
`

func (q *Queries) GetUdpServicesByProfile(ctx context.Context, profileID int64) ([]UdpService, error) {
	rows, err := q.query(ctx, q.getUdpServicesByProfileStmt, getUdpServicesByProfile, profileID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []UdpService
	for rows.Next() {
		var i UdpService
		if err := rows.Scan(
			&i.ID,
			&i.ProfileID,
			&i.AgentID,
			&i.Name,
			&i.Config,
			&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 listUdpServices = `-- name: ListUdpServices :many
SELECT
  id, profile_id, agent_id, name, config, created_at, updated_at
FROM
  udp_services
ORDER BY
  name
LIMIT
  ?
OFFSET
  ?
`

type ListUdpServicesParams struct {
	Limit  int64 `json:"limit"`
	Offset int64 `json:"offset"`
}

func (q *Queries) ListUdpServices(ctx context.Context, arg ListUdpServicesParams) ([]UdpService, error) {
	rows, err := q.query(ctx, q.listUdpServicesStmt, listUdpServices, arg.Limit, arg.Offset)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	var items []UdpService
	for rows.Next() {
		var i UdpService
		if err := rows.Scan(
			&i.ID,
			&i.ProfileID,
			&i.AgentID,
			&i.Name,
			&i.Config,
			&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 updateUdpService = `-- name: UpdateUdpService :one
UPDATE udp_services
SET
  name = ?,
  config = ?,
  updated_at = CURRENT_TIMESTAMP
WHERE
  id = ? RETURNING id, profile_id, agent_id, name, config, created_at, updated_at
`

type UpdateUdpServiceParams struct {
	Name   string              `json:"name"`
	Config *dynamic.UDPService `json:"config"`
	ID     int64               `json:"id"`
}

func (q *Queries) UpdateUdpService(ctx context.Context, arg UpdateUdpServiceParams) (UdpService, error) {
	row := q.queryRow(ctx, q.updateUdpServiceStmt, updateUdpService, arg.Name, arg.Config, arg.ID)
	var i UdpService
	err := row.Scan(
		&i.ID,
		&i.ProfileID,
		&i.AgentID,
		&i.Name,
		&i.Config,
		&i.CreatedAt,
		&i.UpdatedAt,
	)
	return i, err
}