collectscollector

internal/database/repository/collector.sql.go

@@ -11,35 +11,56 @@ import (
 	"github.com/jackc/pgx/v5/pgtype"
 )
 
-const getQueriesFromJobID = `-- name: GetQueriesFromJobID :many
-SELECT id, type, requiredQueryId, queryVersion, minCleanVersion, minTextVersion FROM collectorQueryDependencyTree WHERE jobId = $1
+const getCollectorFromJobID = `-- name: GetCollectorFromJobID :one
+SELECT id, jobId, minCleanVersion, minTextVersion FROM collectors WHERE jobId = $1 LIMIT 1
 `
 
-type GetQueriesFromJobIDRow struct {
+type GetCollectorFromJobIDRow struct {
 	ID              pgtype.UUID
-	Type            Querytype
-	Requiredqueryid pgtype.UUID
-	Queryversion    int32
+	Jobid           pgtype.UUID
 	Mincleanversion int32
 	Mintextversion  int32
 }
 
-func (q *Queries) GetQueriesFromJobID(ctx context.Context, jobid pgtype.UUID) ([]GetQueriesFromJobIDRow, error) {
-	rows, err := q.db.Query(ctx, getQueriesFromJobID, jobid)
+func (q *Queries) GetCollectorFromJobID(ctx context.Context, jobid pgtype.UUID) (GetCollectorFromJobIDRow, error) {
+	row := q.db.QueryRow(ctx, getCollectorFromJobID, jobid)
+	var i GetCollectorFromJobIDRow
+	err := row.Scan(
+		&i.ID,
+		&i.Jobid,
+		&i.Mincleanversion,
+		&i.Mintextversion,
+	)
+	return i, err
+}
+
+const getCollectorQueries = `-- name: GetCollectorQueries :many
+SELECT collectorId, queryId, type, requiredQueryId, queryVersion FROM collectorQueryDependencyTree WHERE collectorId = $1
+`
+
+type GetCollectorQueriesRow struct {
+	Collectorid     pgtype.UUID
+	Queryid         pgtype.UUID
+	Type            NullQuerytype
+	Requiredqueryid pgtype.UUID
+	Queryversion    pgtype.Int4
+}
+
+func (q *Queries) GetCollectorQueries(ctx context.Context, collectorid pgtype.UUID) ([]GetCollectorQueriesRow, error) {
+	rows, err := q.db.Query(ctx, getCollectorQueries, collectorid)
 	if err != nil {
 		return nil, err
 	}
 	defer rows.Close()
-	var items []GetQueriesFromJobIDRow
+	var items []GetCollectorQueriesRow
 	for rows.Next() {
-		var i GetQueriesFromJobIDRow
+		var i GetCollectorQueriesRow
 		if err := rows.Scan(
-			&i.ID,
+			&i.Collectorid,
+			&i.Queryid,
 			&i.Type,
 			&i.Requiredqueryid,
 			&i.Queryversion,
-			&i.Mincleanversion,
-			&i.Mintextversion,
 		); err != nil {
 			return nil, err
 		}