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 addCollectorCodeVersion = `-- name: AddCollectorCodeVersion :exec
INSERT INTO collectorCodeVersions (collectorId, addedVersion, minCleanVersion, minTextVersion) VALUES ($1, $2, $3, $4)
`

type AddCollectorCodeVersionParams struct {
	Collectorid     pgtype.UUID `db:"collectorid"`
	Addedversion    int32       `db:"addedversion"`
	Mincleanversion int32       `db:"mincleanversion"`
	Mintextversion  int32       `db:"mintextversion"`
}

// AddCollectorCodeVersion
//
//	INSERT INTO collectorCodeVersions (collectorId, addedVersion, minCleanVersion, minTextVersion) VALUES ($1, $2, $3, $4)
func (q *Queries) AddCollectorCodeVersion(ctx context.Context, arg *AddCollectorCodeVersionParams) error {
	_, err := q.db.Exec(ctx, addCollectorCodeVersion,
		arg.Collectorid,
		arg.Addedversion,
		arg.Mincleanversion,
		arg.Mintextversion,
	)
	return err
}

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

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

// AddCollectorQuery
//
//	INSERT INTO collectorQueries (collectorId, 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.Collectorid,
		arg.Name,
		arg.Queryid,
		arg.Addedversion,
	)
	return err
}

const createCollector = `-- name: CreateCollector :one
INSERT INTO collectors (jobId) VALUES ($1) RETURNING id
`

// CreateCollector
//
//	INSERT INTO collectors (jobId) VALUES ($1) RETURNING id
func (q *Queries) CreateCollector(ctx context.Context, jobid pgtype.UUID) (pgtype.UUID, error) {
	row := q.db.QueryRow(ctx, createCollector, jobid)
	var id pgtype.UUID
	err := row.Scan(&id)
	return id, err
}

const getCollector = `-- name: GetCollector :one
SELECT id, jobid, mincleanversion, mintextversion, activeversion, latestversion, fields FROM fullActiveCollectors WHERE id = $1 LIMIT 1
`

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

const getCollectorByJobID = `-- name: GetCollectorByJobID :one
SELECT id, jobid, mincleanversion, mintextversion, activeversion, latestversion, fields FROM fullActiveCollectors WHERE jobId = $1 LIMIT 1
`

// GetCollectorByJobID
//
//	SELECT id, jobid, mincleanversion, mintextversion, activeversion, latestversion, fields FROM fullActiveCollectors WHERE jobId = $1 LIMIT 1
func (q *Queries) GetCollectorByJobID(ctx context.Context, jobid pgtype.UUID) (*Fullactivecollector, error) {
	row := q.db.QueryRow(ctx, getCollectorByJobID, jobid)
	var i Fullactivecollector
	err := row.Scan(
		&i.ID,
		&i.Jobid,
		&i.Mincleanversion,
		&i.Mintextversion,
		&i.Activeversion,
		&i.Latestversion,
		&i.Fields,
	)
	return &i, err
}

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

// ListCollectorQueries
//
//	SELECT collectorid, jobid, queryid, type, queryversion, requiredids FROM collectorQueryDependencyTree WHERE collectorId = $1
func (q *Queries) ListCollectorQueries(ctx context.Context, collectorid pgtype.UUID) ([]*Collectorquerydependencytree, error) {
	rows, err := q.db.Query(ctx, listCollectorQueries, collectorid)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	items := []*Collectorquerydependencytree{}
	for rows.Next() {
		var i Collectorquerydependencytree
		if err := rows.Scan(
			&i.Collectorid,
			&i.Jobid,
			&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 removeCollectorCodeVersion = `-- name: RemoveCollectorCodeVersion :exec
UPDATE collectorCodeVersions SET removedVersion = $1 WHERE collectorId = $2 and removedVersion is null
`

type RemoveCollectorCodeVersionParams struct {
	Removedversion *int32      `db:"removedversion"`
	Collectorid    pgtype.UUID `db:"collectorid"`
}

// RemoveCollectorCodeVersion
//
//	UPDATE collectorCodeVersions SET removedVersion = $1 WHERE collectorId = $2 and removedVersion is null
func (q *Queries) RemoveCollectorCodeVersion(ctx context.Context, arg *RemoveCollectorCodeVersionParams) error {
	_, err := q.db.Exec(ctx, removeCollectorCodeVersion, arg.Removedversion, arg.Collectorid)
	return err
}

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

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

// RemoveCollectorQuery
//
//	UPDATE collectorQueries SET removedVersion = $1 WHERE queryId = $2 and collectorId = $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.Collectorid)
	return err
}

const updateCollector = `-- name: UpdateCollector :exec
UPDATE collectors SET latestVersion = $1, activeVersion = $2 WHERE id = $3
`

type UpdateCollectorParams struct {
	Latestversion int32       `db:"latestversion"`
	Activeversion int32       `db:"activeversion"`
	ID            pgtype.UUID `db:"id"`
}

// UpdateCollector
//
//	UPDATE collectors SET latestVersion = $1, activeVersion = $2 WHERE id = $3
func (q *Queries) UpdateCollector(ctx context.Context, arg *UpdateCollectorParams) error {
	_, err := q.db.Exec(ctx, updateCollector, arg.Latestversion, arg.Activeversion, arg.ID)
	return err
}