Merged in feature/docinit (pull request #48)
Feature/docinit * createunittests * cleanup * skip * cleanup
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user