2025-01-27 22:00:51 +01:00
|
|
|
package database
|
|
|
|
|
|
|
|
import (
|
2025-01-28 00:30:30 +01:00
|
|
|
"database/sql"
|
|
|
|
"errors"
|
2025-01-27 22:00:51 +01:00
|
|
|
"fmt"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"git.adyxax.org/adyxax/tfstated/pkg/model"
|
|
|
|
)
|
|
|
|
|
2025-01-28 00:30:30 +01:00
|
|
|
func (db *DB) LoadVersionById(id int) (*model.Version, error) {
|
|
|
|
version := model.Version{
|
|
|
|
Id: id,
|
|
|
|
}
|
|
|
|
var (
|
|
|
|
created int64
|
|
|
|
encryptedData []byte
|
|
|
|
)
|
|
|
|
err := db.QueryRow(
|
|
|
|
`SELECT account_id, state_id, data, lock, created FROM versions WHERE id = ?;`,
|
|
|
|
id).Scan(
|
|
|
|
&version.AccountId,
|
|
|
|
&version.StateId,
|
|
|
|
&encryptedData,
|
|
|
|
&version.Lock,
|
|
|
|
&created)
|
|
|
|
if err != nil {
|
|
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
return nil, fmt.Errorf("failed to load version id %d from database: %w", id, err)
|
|
|
|
}
|
|
|
|
version.Created = time.Unix(created, 0)
|
|
|
|
version.Data, err = db.dataEncryptionKey.DecryptAES256(encryptedData)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("failed to decrypt version %d data: %w", id, err)
|
|
|
|
}
|
|
|
|
return &version, nil
|
|
|
|
}
|
|
|
|
|
2025-01-27 22:00:51 +01:00
|
|
|
func (db *DB) LoadVersionsByState(state *model.State) ([]model.Version, error) {
|
|
|
|
rows, err := db.Query(
|
|
|
|
`SELECT account_id, created, data, id, lock
|
|
|
|
FROM versions
|
|
|
|
WHERE state_id = ?
|
|
|
|
ORDER BY id DESC;`, state.Id)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("failed to load versions from database: %w", err)
|
|
|
|
}
|
|
|
|
defer rows.Close()
|
|
|
|
versions := make([]model.Version, 0)
|
|
|
|
for rows.Next() {
|
|
|
|
var version model.Version
|
|
|
|
var created int64
|
|
|
|
err = rows.Scan(&version.AccountId, &created, &version.Data, &version.Id, &version.Lock)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("failed to load version from row: %w", err)
|
|
|
|
}
|
|
|
|
version.Created = time.Unix(created, 0)
|
|
|
|
versions = append(versions, version)
|
|
|
|
}
|
|
|
|
if err := rows.Err(); err != nil {
|
|
|
|
return nil, fmt.Errorf("failed to load versions from rows: %w", err)
|
|
|
|
}
|
|
|
|
return versions, nil
|
|
|
|
}
|