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 (jobId, name, queryId, addedVersion) VALUES ($1, $2, $3, $4)
`

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

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

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

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

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

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

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

// GetCollectorByJobID
//
//	SELECT 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.Jobid,
		&i.Mincleanversion,
		&i.Mintextversion,
		&i.Activeversion,
		&i.Latestversion,
		&i.Fields,
	)
	return &i, err
}

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

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

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

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

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

type SetActiveCollectorVersionParams struct {
	Jobid     pgtype.UUID `db:"jobid"`
	Versionid int32       `db:"versionid"`
}

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

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

type SetCollectorCleanVersionParams struct {
	Jobid        pgtype.UUID `db:"jobid"`
	Addedversion int32       `db:"addedversion"`
	Versionid    int32       `db:"versionid"`
}

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

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

type SetCollectorTextVersionParams struct {
	Jobid        pgtype.UUID `db:"jobid"`
	Addedversion int32       `db:"addedversion"`
	Versionid    int32       `db:"versionid"`
}

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