531a0184f7
Replace old used-to-be-GORM datastructures (#104305) with sqlc-generated structs. This also makes it possible to use more specific structs that are more taylored to the specific queries, increasing efficiency. This commit deals with the remaining areas, like the job deleter, task timeout checker, and task state machine. And anything else to get things running again. Functional changes are kept to a minimum, as the API still serves the same data. Because this work covers so much of Flamenco's code, it's been split up into different commits. Each commit brings Flamenco to a state where it compiles and unit tests pass. Only the result of the final commit has actually been tested properly. Ref: #104343
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package persistence
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
|
|
"projects.blender.org/studio/flamenco/internal/manager/persistence/sqlc"
|
|
)
|
|
|
|
// LastRendered only has one entry in its database table, to indicate the job
|
|
// that was the last to receive a "last rendered image" from a Worker.
|
|
// This is used to show the global last-rendered image in the web interface.
|
|
|
|
// SetLastRendered sets this job as the one with the most recent rendered image.
|
|
func (db *DB) SetLastRendered(ctx context.Context, jobUUID string) error {
|
|
queries := db.queries()
|
|
|
|
jobID, err := queries.FetchJobIDFromUUID(ctx, jobUUID)
|
|
if err != nil {
|
|
return jobError(err, "finding job with UUID %q", jobUUID)
|
|
}
|
|
|
|
now := db.nowNullable()
|
|
return queries.SetLastRendered(ctx, sqlc.SetLastRenderedParams{
|
|
CreatedAt: now.Time,
|
|
UpdatedAt: now,
|
|
JobID: jobID,
|
|
})
|
|
}
|
|
|
|
// GetLastRendered returns the UUID of the job with the most recent rendered image.
|
|
func (db *DB) GetLastRenderedJobUUID(ctx context.Context) (string, error) {
|
|
queries := db.queries()
|
|
|
|
jobUUID, err := queries.GetLastRenderedJobUUID(ctx)
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return "", nil
|
|
}
|
|
if err != nil {
|
|
return "", jobError(err, "finding job with most rencent render")
|
|
}
|
|
return jobUUID, nil
|
|
}
|