|
|
|
@@ -12,7 +12,7 @@ import (
|
|
|
|
|
const createFile = `-- name: CreateFile :one
|
|
|
|
|
INSERT INTO files (id, name, mime_type, size, storage_key, checksum, created_at, updated_at)
|
|
|
|
|
VALUES (?, ?, ?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
|
|
|
|
|
RETURNING id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at, nlp_data
|
|
|
|
|
RETURNING id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
|
type CreateFileParams struct {
|
|
|
|
@@ -44,7 +44,6 @@ func (q *Queries) CreateFile(ctx context.Context, arg CreateFileParams) (File, e
|
|
|
|
|
&i.OcrText,
|
|
|
|
|
&i.CreatedAt,
|
|
|
|
|
&i.UpdatedAt,
|
|
|
|
|
&i.NlpData,
|
|
|
|
|
)
|
|
|
|
|
return i, err
|
|
|
|
|
}
|
|
|
|
@@ -60,7 +59,7 @@ func (q *Queries) DeleteFile(ctx context.Context, id string) error {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const getFile = `-- name: GetFile :one
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at, nlp_data FROM files
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at FROM files
|
|
|
|
|
WHERE id = ? LIMIT 1
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
@@ -77,13 +76,12 @@ func (q *Queries) GetFile(ctx context.Context, id string) (File, error) {
|
|
|
|
|
&i.OcrText,
|
|
|
|
|
&i.CreatedAt,
|
|
|
|
|
&i.UpdatedAt,
|
|
|
|
|
&i.NlpData,
|
|
|
|
|
)
|
|
|
|
|
return i, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const listFiles = `-- name: ListFiles :many
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at, nlp_data FROM files
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at FROM files
|
|
|
|
|
ORDER BY created_at DESC
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
@@ -106,7 +104,6 @@ func (q *Queries) ListFiles(ctx context.Context) ([]File, error) {
|
|
|
|
|
&i.OcrText,
|
|
|
|
|
&i.CreatedAt,
|
|
|
|
|
&i.UpdatedAt,
|
|
|
|
|
&i.NlpData,
|
|
|
|
|
); err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
@@ -122,7 +119,7 @@ func (q *Queries) ListFiles(ctx context.Context) ([]File, error) {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const listFilesByID = `-- name: ListFilesByID :many
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at, nlp_data FROM files
|
|
|
|
|
SELECT id, name, mime_type, size, storage_key, checksum, ocr_text, created_at, updated_at FROM files
|
|
|
|
|
WHERE id IN (SELECT value FROM json_each(?))
|
|
|
|
|
ORDER BY created_at DESC
|
|
|
|
|
`
|
|
|
|
@@ -146,7 +143,6 @@ func (q *Queries) ListFilesByID(ctx context.Context, jsonEach interface{}) ([]Fi
|
|
|
|
|
&i.OcrText,
|
|
|
|
|
&i.CreatedAt,
|
|
|
|
|
&i.UpdatedAt,
|
|
|
|
|
&i.NlpData,
|
|
|
|
|
); err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
@@ -183,19 +179,3 @@ func (q *Queries) UpdateFile(ctx context.Context, arg UpdateFileParams) error {
|
|
|
|
|
)
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const updateNLPData = `-- name: UpdateNLPData :exec
|
|
|
|
|
UPDATE files
|
|
|
|
|
SET nlp_data = ?, updated_at = CURRENT_TIMESTAMP
|
|
|
|
|
WHERE id = ?
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
|
type UpdateNLPDataParams struct {
|
|
|
|
|
NlpData string `json:"nlp_data"`
|
|
|
|
|
ID string `json:"id"`
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (q *Queries) UpdateNLPData(ctx context.Context, arg UpdateNLPDataParams) error {
|
|
|
|
|
_, err := q.db.ExecContext(ctx, updateNLPData, arg.NlpData, arg.ID)
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|