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/google/uuid"
)

const addDocumentText = `-- name: AddDocumentText :one
INSERT INTO documentTextExtractions (bucket, key, hash) VALUES ($1, $2, $3) returning id
`

type AddDocumentTextParams struct {
	Bucket string `db:"bucket"`
	Key    string `db:"key"`
	Hash   string `db:"hash"`
}

// AddDocumentText
//
//	INSERT INTO documentTextExtractions (bucket, key, hash) VALUES ($1, $2, $3) returning id
func (q *Queries) AddDocumentText(ctx context.Context, arg *AddDocumentTextParams) (uuid.UUID, error) {
	row := q.db.QueryRow(ctx, addDocumentText, arg.Bucket, arg.Key, arg.Hash)
	var id uuid.UUID
	err := row.Scan(&id)
	return id, err
}

const addDocumentTextEntry = `-- name: AddDocumentTextEntry :exec
INSERT INTO documentTextExtractionEntries (textId, triggerId, version) VALUES ($1, $2, $3)
`

type AddDocumentTextEntryParams struct {
	Textid    uuid.UUID `db:"textid"`
	Triggerid uuid.UUID `db:"triggerid"`
	Version   int64     `db:"version"`
}

// AddDocumentTextEntry
//
//	INSERT INTO documentTextExtractionEntries (textId, triggerId, version) VALUES ($1, $2, $3)
func (q *Queries) AddDocumentTextEntry(ctx context.Context, arg *AddDocumentTextEntryParams) error {
	_, err := q.db.Exec(ctx, addDocumentTextEntry, arg.Textid, arg.Triggerid, arg.Version)
	return err
}

const addDocumentTextTrigger = `-- name: AddDocumentTextTrigger :one
INSERT INTO documentTextTriggerExtractions (cleanId, version) VALUES ($1, $2) returning id
`

type AddDocumentTextTriggerParams struct {
	Cleanid uuid.UUID `db:"cleanid"`
	Version int64     `db:"version"`
}

// AddDocumentTextTrigger
//
//	INSERT INTO documentTextTriggerExtractions (cleanId, version) VALUES ($1, $2) returning id
func (q *Queries) AddDocumentTextTrigger(ctx context.Context, arg *AddDocumentTextTriggerParams) (uuid.UUID, error) {
	row := q.db.QueryRow(ctx, addDocumentTextTrigger, arg.Cleanid, arg.Version)
	var id uuid.UUID
	err := row.Scan(&id)
	return id, err
}

const addDocumentTextTriggerJobId = `-- name: AddDocumentTextTriggerJobId :exec
UPDATE documentTextTriggerExtractions SET textractJobId = $1 WHERE id = $2
`

type AddDocumentTextTriggerJobIdParams struct {
	Textractjobid *string   `db:"textractjobid"`
	ID            uuid.UUID `db:"id"`
}

// AddDocumentTextTriggerJobId
//
//	UPDATE documentTextTriggerExtractions SET textractJobId = $1 WHERE id = $2
func (q *Queries) AddDocumentTextTriggerJobId(ctx context.Context, arg *AddDocumentTextTriggerJobIdParams) error {
	_, err := q.db.Exec(ctx, addDocumentTextTriggerJobId, arg.Textractjobid, arg.ID)
	return err
}

const getDocumentTextExtractionByHash = `-- name: GetDocumentTextExtractionByHash :one
SELECT id FROM currentTextEntries WHERE cleanId = $1 and hash = $2
`

type GetDocumentTextExtractionByHashParams struct {
	Cleanentryid uuid.UUID `db:"cleanentryid"`
	Hash         string    `db:"hash"`
}

// GetDocumentTextExtractionByHash
//
//	SELECT id FROM currentTextEntries WHERE cleanId = $1 and hash = $2
func (q *Queries) GetDocumentTextExtractionByHash(ctx context.Context, arg *GetDocumentTextExtractionByHashParams) (uuid.UUID, error) {
	row := q.db.QueryRow(ctx, getDocumentTextExtractionByHash, arg.Cleanentryid, arg.Hash)
	var id uuid.UUID
	err := row.Scan(&id)
	return id, err
}

const getDocumentTextTrigger = `-- name: GetDocumentTextTrigger :one
SELECT dt.id, dt.cleanId, dt.version, dt.textractJobId, dc.documentId
    from documentTextTriggerExtractions as dt
    join documentCleans as dc on dt.cleanId = dc.id
    where dt.id = $1
`

type GetDocumentTextTriggerRow struct {
	ID            uuid.UUID `db:"id"`
	Cleanid       uuid.UUID `db:"cleanid"`
	Version       int64     `db:"version"`
	Textractjobid *string   `db:"textractjobid"`
	Documentid    uuid.UUID `db:"documentid"`
}

// GetDocumentTextTrigger
//
//	SELECT dt.id, dt.cleanId, dt.version, dt.textractJobId, dc.documentId
//	    from documentTextTriggerExtractions as dt
//	    join documentCleans as dc on dt.cleanId = dc.id
//	    where dt.id = $1
func (q *Queries) GetDocumentTextTrigger(ctx context.Context, triggerid uuid.UUID) (*GetDocumentTextTriggerRow, error) {
	row := q.db.QueryRow(ctx, getDocumentTextTrigger, triggerid)
	var i GetDocumentTextTriggerRow
	err := row.Scan(
		&i.ID,
		&i.Cleanid,
		&i.Version,
		&i.Textractjobid,
		&i.Documentid,
	)
	return &i, err
}

const getTextEntryByDocId = `-- name: GetTextEntryByDocId :one
SELECT id, documentId, bucket, key, hash, cleanId,
    triggerId, textractJobId, triggerVersion, extractionVersion
    FROM currentTextEntries
    WHERE documentId = $1
`

// GetTextEntryByDocId
//
//	SELECT id, documentId, bucket, key, hash, cleanId,
//	    triggerId, textractJobId, triggerVersion, extractionVersion
//	    FROM currentTextEntries
//	    WHERE documentId = $1
func (q *Queries) GetTextEntryByDocId(ctx context.Context, documentid uuid.UUID) (*Currenttextentry, error) {
	row := q.db.QueryRow(ctx, getTextEntryByDocId, documentid)
	var i Currenttextentry
	err := row.Scan(
		&i.ID,
		&i.Documentid,
		&i.Bucket,
		&i.Key,
		&i.Hash,
		&i.Cleanid,
		&i.Triggerid,
		&i.Textractjobid,
		&i.Triggerversion,
		&i.Extractionversion,
	)
	return &i, err
}

const isDocumentTextExtracted = `-- name: IsDocumentTextExtracted :one
SELECT EXISTS(
    SELECT 1 FROM currentTextEntries WHERE documentId = $1
)
`

// IsDocumentTextExtracted
//
//	SELECT EXISTS(
//	    SELECT 1 FROM currentTextEntries WHERE documentId = $1
//	)
func (q *Queries) IsDocumentTextExtracted(ctx context.Context, documentid uuid.UUID) (bool, error) {
	row := q.db.QueryRow(ctx, isDocumentTextExtracted, documentid)
	var exists bool
	err := row.Scan(&exists)
	return exists, err
}