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

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

package repository

import (
	"context"

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

const addDocumentTextEntry = `-- name: AddDocumentTextEntry :exec
INSERT INTO documentTextExtractions (documentId, version, bucket, key) VALUES ($1, $2, $3, $4)
`

type AddDocumentTextEntryParams struct {
	Documentid pgtype.UUID `db:"documentid"`
	Version    int32       `db:"version"`
	Bucket     string      `db:"bucket"`
	Key        string      `db:"key"`
}

// AddDocumentTextEntry
//
//	INSERT INTO documentTextExtractions (documentId, version, bucket, key) VALUES ($1, $2, $3, $4)
func (q *Queries) AddDocumentTextEntry(ctx context.Context, arg *AddDocumentTextEntryParams) error {
	_, err := q.db.Exec(ctx, addDocumentTextEntry,
		arg.Documentid,
		arg.Version,
		arg.Bucket,
		arg.Key,
	)
	return err
}

const getDocumentTextEntry = `-- name: GetDocumentTextEntry :one
SELECT dc.documentId, dc.bucket, dc.key, dc.version
    FROM documentTextExtractions AS dc
    JOIN documents AS d ON d.id = dc.documentId
    LEFT JOIN collectors as c ON d.jobId = c.jobId
    LEFT JOIN collectorCodeVersions as cv ON c.id = cv.collectorId
    WHERE dc.documentId = $1 and dc.version >= coalesce(cv.minTextVersion, 1)
    ORDER BY d.id DESC LIMIT 1
`

type GetDocumentTextEntryRow struct {
	Documentid pgtype.UUID `db:"documentid"`
	Bucket     string      `db:"bucket"`
	Key        string      `db:"key"`
	Version    int32       `db:"version"`
}

// GetDocumentTextEntry
//
//	SELECT dc.documentId, dc.bucket, dc.key, dc.version
//	    FROM documentTextExtractions AS dc
//	    JOIN documents AS d ON d.id = dc.documentId
//	    LEFT JOIN collectors as c ON d.jobId = c.jobId
//	    LEFT JOIN collectorCodeVersions as cv ON c.id = cv.collectorId
//	    WHERE dc.documentId = $1 and dc.version >= coalesce(cv.minTextVersion, 1)
//	    ORDER BY d.id DESC LIMIT 1
func (q *Queries) GetDocumentTextEntry(ctx context.Context, documentid pgtype.UUID) (*GetDocumentTextEntryRow, error) {
	row := q.db.QueryRow(ctx, getDocumentTextEntry, documentid)
	var i GetDocumentTextEntryRow
	err := row.Scan(
		&i.Documentid,
		&i.Bucket,
		&i.Key,
		&i.Version,
	)
	return &i, err
}

const isDocumentTextExtracted = `-- name: IsDocumentTextExtracted :one
SELECT EXISTS(
    SELECT 1
        FROM documentTextExtractions AS dc
        JOIN documents AS d ON d.id = dc.documentId
        LEFT JOIN collectors as c ON d.jobId = c.jobId
        LEFT JOIN collectorCodeVersions as cv ON c.id = cv.collectorId
        WHERE dc.documentId = $1 and dc.version >= coalesce(cv.minTextVersion, 1)
)
`

// IsDocumentTextExtracted
//
//	SELECT EXISTS(
//	    SELECT 1
//	        FROM documentTextExtractions AS dc
//	        JOIN documents AS d ON d.id = dc.documentId
//	        LEFT JOIN collectors as c ON d.jobId = c.jobId
//	        LEFT JOIN collectorCodeVersions as cv ON c.id = cv.collectorId
//	        WHERE dc.documentId = $1 and dc.version >= coalesce(cv.minTextVersion, 1)
//	)
func (q *Queries) IsDocumentTextExtracted(ctx context.Context, documentid pgtype.UUID) (bool, error) {
	row := q.db.QueryRow(ctx, isDocumentTextExtracted, documentid)
	var exists bool
	err := row.Scan(&exists)
	return exists, err
}