Merged in feature/docinit (pull request #48)

Feature/docinit

* createunittests

* cleanup

* skip

* cleanup
This commit is contained in:
Michael McGuinness
2025-02-05 17:44:01 +00:00
parent 92334ad1dd
commit 5c253b3592
28 changed files with 536 additions and 101 deletions
+45 -14
View File
@@ -11,41 +11,72 @@ import (
"github.com/jackc/pgx/v5/pgtype"
)
const addDocumentEntry = `-- name: AddDocumentEntry :exec
INSERT INTO documentEntries (documentId, bucket, location) VALUES ($1, $2, $3)
`
type AddDocumentEntryParams struct {
Documentid pgtype.UUID `db:"documentid"`
Bucket string `db:"bucket"`
Location string `db:"location"`
}
// AddDocumentEntry
//
// INSERT INTO documentEntries (documentId, bucket, location) VALUES ($1, $2, $3)
func (q *Queries) AddDocumentEntry(ctx context.Context, arg *AddDocumentEntryParams) error {
_, err := q.db.Exec(ctx, addDocumentEntry, arg.Documentid, arg.Bucket, arg.Location)
return err
}
const createDocument = `-- name: CreateDocument :one
INSERT INTO documents (jobId, hash, location) VALUES ($1, $2, $3) RETURNING id
INSERT INTO documents (jobId, hash) VALUES ($1, $2) RETURNING id
`
type CreateDocumentParams struct {
Jobid pgtype.UUID `db:"jobid"`
Hash string `db:"hash"`
Location string `db:"location"`
Jobid pgtype.UUID `db:"jobid"`
Hash string `db:"hash"`
}
// CreateDocument
//
// INSERT INTO documents (jobId, hash, location) VALUES ($1, $2, $3) RETURNING id
// INSERT INTO documents (jobId, hash) VALUES ($1, $2) RETURNING id
func (q *Queries) CreateDocument(ctx context.Context, arg *CreateDocumentParams) (pgtype.UUID, error) {
row := q.db.QueryRow(ctx, createDocument, arg.Jobid, arg.Hash, arg.Location)
row := q.db.QueryRow(ctx, createDocument, arg.Jobid, arg.Hash)
var id pgtype.UUID
err := row.Scan(&id)
return id, err
}
const getDocument = `-- name: GetDocument :one
SELECT id, jobId, hash, location FROM documents WHERE id = $1 LIMIT 1
SELECT id, jobId, hash FROM documents WHERE id = $1 LIMIT 1
`
// GetDocument
//
// SELECT id, jobId, hash, location FROM documents WHERE id = $1 LIMIT 1
// SELECT id, jobId, hash FROM documents WHERE id = $1 LIMIT 1
func (q *Queries) GetDocument(ctx context.Context, id pgtype.UUID) (*Document, error) {
row := q.db.QueryRow(ctx, getDocument, id)
var i Document
err := row.Scan(
&i.ID,
&i.Jobid,
&i.Hash,
&i.Location,
)
err := row.Scan(&i.ID, &i.Jobid, &i.Hash)
return &i, err
}
const getDocumentIDByHash = `-- name: GetDocumentIDByHash :one
SELECT id FROM documents WHERE hash = $1 and jobId = $2
`
type GetDocumentIDByHashParams struct {
Hash string `db:"hash"`
Jobid pgtype.UUID `db:"jobid"`
}
// GetDocumentIDByHash
//
// SELECT id FROM documents WHERE hash = $1 and jobId = $2
func (q *Queries) GetDocumentIDByHash(ctx context.Context, arg *GetDocumentIDByHashParams) (pgtype.UUID, error) {
row := q.db.QueryRow(ctx, getDocumentIDByHash, arg.Hash, arg.Jobid)
var id pgtype.UUID
err := row.Scan(&id)
return id, err
}