query-orchestration/internal/database/repository/collector.sql.go

// Code generated by sqlc. DO NOT EDIT.
// versions:
//   sqlc v1.27.0
// source: collector.sql

package repository

import (
	"context"

	"github.com/jackc/pgx/v5/pgtype"
)

const addCollectorQuery = `-- name: AddCollectorQuery :exec
INSERT INTO collectorQueries (clientId, name, queryId, addedVersion) VALUES ($1, $2, $3, $4)
`

type AddCollectorQueryParams struct {
	Clientid     pgtype.UUID `db:"clientid"`
	Name         string      `db:"name"`
	Queryid      pgtype.UUID `db:"queryid"`
	Addedversion int32       `db:"addedversion"`
}

// AddCollectorQuery
//
//	INSERT INTO collectorQueries (clientId, name, queryId, addedVersion) VALUES ($1, $2, $3, $4)
func (q *Queries) AddCollectorQuery(ctx context.Context, arg *AddCollectorQueryParams) error {
	_, err := q.db.Exec(ctx, addCollectorQuery,
		arg.Clientid,
		arg.Name,
		arg.Queryid,
		arg.Addedversion,
	)
	return err
}

const addLatestCollectorVersion = `-- name: AddLatestCollectorVersion :one
INSERT INTO collectorVersions (clientId) VALUES ($1) RETURNING id
`

// AddLatestCollectorVersion
//
//	INSERT INTO collectorVersions (clientId) VALUES ($1) RETURNING id
func (q *Queries) AddLatestCollectorVersion(ctx context.Context, clientid pgtype.UUID) (int32, error) {
	row := q.db.QueryRow(ctx, addLatestCollectorVersion, clientid)
	var id int32
	err := row.Scan(&id)
	return id, err
}

const getCollectorByClientExternalID = `-- name: GetCollectorByClientExternalID :one
WITH client as (
    SELECT id from clients where externalId = $1
)
SELECT fc.clientid, fc.mincleanversion, fc.mintextversion, fc.activeversion, fc.latestversion, fc.fields
    FROM fullActiveCollectors as fc
    JOIN client as c ON fc.clientId = c.id LIMIT 1
`

// GetCollectorByClientExternalID
//
//	WITH client as (
//	    SELECT id from clients where externalId = $1
//	)
//	SELECT fc.clientid, fc.mincleanversion, fc.mintextversion, fc.activeversion, fc.latestversion, fc.fields
//	    FROM fullActiveCollectors as fc
//	    JOIN client as c ON fc.clientId = c.id LIMIT 1
func (q *Queries) GetCollectorByClientExternalID(ctx context.Context, clientid string) (*Fullactivecollector, error) {
	row := q.db.QueryRow(ctx, getCollectorByClientExternalID, clientid)
	var i Fullactivecollector
	err := row.Scan(
		&i.Clientid,
		&i.Mincleanversion,
		&i.Mintextversion,
		&i.Activeversion,
		&i.Latestversion,
		&i.Fields,
	)
	return &i, err
}

const getCollectorByClientID = `-- name: GetCollectorByClientID :one
SELECT clientid, mincleanversion, mintextversion, activeversion, latestversion, fields FROM fullActiveCollectors WHERE clientId = $1 LIMIT 1
`

// GetCollectorByClientID
//
//	SELECT clientid, mincleanversion, mintextversion, activeversion, latestversion, fields FROM fullActiveCollectors WHERE clientId = $1 LIMIT 1
func (q *Queries) GetCollectorByClientID(ctx context.Context, clientid pgtype.UUID) (*Fullactivecollector, error) {
	row := q.db.QueryRow(ctx, getCollectorByClientID, clientid)
	var i Fullactivecollector
	err := row.Scan(
		&i.Clientid,
		&i.Mincleanversion,
		&i.Mintextversion,
		&i.Activeversion,
		&i.Latestversion,
		&i.Fields,
	)
	return &i, err
}

const listCollectorQueries = `-- name: ListCollectorQueries :many
SELECT clientid, queryid, type, queryversion, requiredids FROM collectorQueryDependencyTree WHERE clientId = $1
`

// ListCollectorQueries
//
//	SELECT clientid, queryid, type, queryversion, requiredids FROM collectorQueryDependencyTree WHERE clientId = $1
func (q *Queries) ListCollectorQueries(ctx context.Context, clientid pgtype.UUID) ([]*Collectorquerydependencytree, error) {
	rows, err := q.db.Query(ctx, listCollectorQueries, clientid)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []*Collectorquerydependencytree{}
	for rows.Next() {
		var i Collectorquerydependencytree
		if err := rows.Scan(
			&i.Clientid,
			&i.Queryid,
			&i.Type,
			&i.Queryversion,
			&i.Requiredids,
		); err != nil {
			return nil, err
		}
		items = append(items, &i)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return items, nil
}

const removeCollectorQuery = `-- name: RemoveCollectorQuery :exec
UPDATE collectorQueries SET removedVersion = $1 WHERE queryId = $2 and clientId = $3 and removedVersion is null
`

type RemoveCollectorQueryParams struct {
	Removedversion *int32      `db:"removedversion"`
	Queryid        pgtype.UUID `db:"queryid"`
	Clientid       pgtype.UUID `db:"clientid"`
}

// RemoveCollectorQuery
//
//	UPDATE collectorQueries SET removedVersion = $1 WHERE queryId = $2 and clientId = $3 and removedVersion is null
func (q *Queries) RemoveCollectorQuery(ctx context.Context, arg *RemoveCollectorQueryParams) error {
	_, err := q.db.Exec(ctx, removeCollectorQuery, arg.Removedversion, arg.Queryid, arg.Clientid)
	return err
}

const setActiveCollectorVersion = `-- name: SetActiveCollectorVersion :exec
INSERT INTO collectorActiveVersions (clientId, versionId) VALUES ($1, $2)
`

type SetActiveCollectorVersionParams struct {
	Clientid  pgtype.UUID `db:"clientid"`
	Versionid int32       `db:"versionid"`
}

// SetActiveCollectorVersion
//
//	INSERT INTO collectorActiveVersions (clientId, versionId) VALUES ($1, $2)
func (q *Queries) SetActiveCollectorVersion(ctx context.Context, arg *SetActiveCollectorVersionParams) error {
	_, err := q.db.Exec(ctx, setActiveCollectorVersion, arg.Clientid, arg.Versionid)
	return err
}

const setCollectorCleanVersion = `-- name: SetCollectorCleanVersion :exec
INSERT INTO collectorMinCleanVersions (clientId, addedVersion, versionId) VALUES ($1, $2, $3)
`

type SetCollectorCleanVersionParams struct {
	Clientid     pgtype.UUID `db:"clientid"`
	Addedversion int32       `db:"addedversion"`
	Versionid    int64       `db:"versionid"`
}

// SetCollectorCleanVersion
//
//	INSERT INTO collectorMinCleanVersions (clientId, addedVersion, versionId) VALUES ($1, $2, $3)
func (q *Queries) SetCollectorCleanVersion(ctx context.Context, arg *SetCollectorCleanVersionParams) error {
	_, err := q.db.Exec(ctx, setCollectorCleanVersion, arg.Clientid, arg.Addedversion, arg.Versionid)
	return err
}

const setCollectorTextVersion = `-- name: SetCollectorTextVersion :exec
INSERT INTO collectorMinTextVersions (clientId, addedVersion, versionId) VALUES ($1, $2, $3)
`

type SetCollectorTextVersionParams struct {
	Clientid     pgtype.UUID `db:"clientid"`
	Addedversion int32       `db:"addedversion"`
	Versionid    int64       `db:"versionid"`
}

// SetCollectorTextVersion
//
//	INSERT INTO collectorMinTextVersions (clientId, addedVersion, versionId) VALUES ($1, $2, $3)
func (q *Queries) SetCollectorTextVersion(ctx context.Context, arg *SetCollectorTextVersionParams) error {
	_, err := q.db.Exec(ctx, setCollectorTextVersion, arg.Clientid, arg.Addedversion, arg.Versionid)
	return err
}